diff --git a/modules/rule34.py b/modules/rule34.py index 94bca2f..690cb5d 100644 --- a/modules/rule34.py +++ b/modules/rule34.py @@ -1,6 +1,6 @@ #!/usr/bin/python3 """ -rule34.py - urban dictionary module +rule34.py - rule 34 module author: mutantmonkey """ diff --git a/modules/test/test_rule34.py b/modules/test/test_rule34.py new file mode 100644 index 0000000..3d28795 --- /dev/null +++ b/modules/test/test_rule34.py @@ -0,0 +1,29 @@ +""" +test_rule34.py - tests for the rule 34 module +author: mutantmonkey +""" + +import re +import unittest +from mock import MagicMock, Mock +from modules.rule34 import rule34 + + +class TestRule34(unittest.TestCase): + def setUp(self): + self.phenny = MagicMock() + + def test_result(self): + input = Mock(group=lambda x: 'python') + rule34(self.phenny, input) + + out = self.phenny.reply.call_args[0][0] + m = re.match('^!!NSFW!! -> http://rule34\.xxx/.* <- !!NSFW!!$', out, + flags=re.UNICODE) + self.assertTrue(m) + def test_none(self): + input = Mock(group=lambda x: '__no_results_for_this__') + rule34(self.phenny, input) + + self.phenny.reply.assert_called_once_with( + "You just broke Rule 34! Better start uploading...")