diff --git a/modules/test/test_imdb.py b/modules/test/test_imdb.py new file mode 100644 index 0000000..d84a0e9 --- /dev/null +++ b/modules/test/test_imdb.py @@ -0,0 +1,33 @@ +""" +test_imdb.py - tests for the imdb module module +author: mutantmonkey +""" + +# 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)