seen: add test

plugins/seen.py

@@ -1,6 +1,7 @@
 " seen.py: written by sklnd in about two beers July 2009"
 
 import time
+import unittest
 
 from util import hook, timesince
 
@@ -42,8 +43,6 @@ def seen(inp, nick='', chan='', db=None, input=None):
 
     if last_seen:
         reltime = timesince.timesince(last_seen[1])
-        if last_seen[0] != inp.lower():  # for glob matching
-            inp = last_seen[0]
         if last_seen[2][0:1] == "\x01":
             return '%s was last seen %s ago: *%s %s*' % \
                 (inp, reltime, inp, last_seen[2][8:-1])
@@ -52,3 +51,43 @@ def seen(inp, nick='', chan='', db=None, input=None):
                 (inp, reltime, last_seen[2])
     else:
         return "I've never seen %s" % inp
+
+
+class SeenTest(unittest.TestCase):
+    class Mock(object):
+        def __init__(self, **kwargs):
+            self.__dict__.update(kwargs)
+
+    def setUp(self):
+        import sqlite3
+        self.db = sqlite3.connect(':memory:')
+
+    def seeninput(self, nick, msg, chan='#test'):
+        seeninput(None, db=self.db,
+                  input=self.Mock(nick=nick, msg=msg, chan=chan))
+
+    def seen(self, inp, nick='bob', chan='#test', bot_nick='skybot'):
+        return seen(inp, nick=nick, chan=chan, db=self.db,
+                    input=self.Mock(conn=self.Mock(nick=bot_nick)))
+
+    def test_missing(self):
+        assert "I've never seen nemo" in self.seen('NEMO')
+
+    def test_seen(self):
+        self.seeninput('nemo', 'witty banter')
+        assert 'nemo was last seen' in self.seen('nemo')
+        assert 'witty banter' in self.seen('nemo')
+
+    def test_seen_missing_channel(self):
+        self.seeninput('nemo', 'msg', chan='#secret')
+        assert 'never seen' in self.seen('nemo')
+
+    def test_seen_ctcp(self):
+        self.seeninput('nemo', '\x01ACTION test lol\x01')
+        assert self.seen('nemo').endswith('ago: *nemo test lol*')
+
+    def test_snark_eyes(self):
+        assert 'eyes checked' in self.seen('skybot', bot_nick='skybot')
+
+    def test_snark_mirror(self):
+        assert 'mirror' in self.seen('bob', nick='bob')