add imdb tests

modules/test/test_imdb.py

@@ -0,0 +1,33 @@
+"""
+test_imdb.py - tests for the imdb module module
+author: mutantmonkey <mutantmonkey@mutantmonkey.in>
+"""
+
+# add current working directory to path
+import sys
+sys.path.append('.')
+
+import re
+import unittest
+from mock import MagicMock, Mock
+from modules.imdb import imdb_search, imdb
+
+class TestImdb(unittest.TestCase):
+    def setUp(self):
+        self.phenny = MagicMock()
+
+    def test_imdb_seach(self):
+        data = imdb_search('Hackers')
+
+        assert 'Plot' in data
+        assert 'Title' in data
+        assert 'Year' in data
+        assert 'imdbID' in data
+
+    def test_imdb(self):
+        input = Mock(group=lambda x: 'Antitrust')
+        imdb(self.phenny, input)
+
+        out = self.phenny.reply.call_args[0][0]
+        m = re.match('^.* \(.*\): .* http://imdb.com/title/[a-z\d]+$', out, flags=re.UNICODE)
+        self.assertTrue(m)