add search tests

modules/test/test_search.py

@@ -0,0 +1,86 @@
+"""
+test_search.py - tests for the search module
+author: mutantmonkey <mutantmonkey@mutantmonkey.in>
+"""
+
+import re
+import unittest
+from mock import MagicMock, Mock
+from modules.search import google_ajax, google_search, google_count, \
+        formatnumber, g, gc, gcs, bing_search, bing, duck_search, duck, \
+        search, suggest
+
+
+class TestSearch(unittest.TestCase):
+    def setUp(self):
+        self.phenny = MagicMock()
+
+    def test_google_ajax(self):
+        data = google_ajax('phenny')
+
+        assert 'responseData' in data
+        assert data['responseStatus'] == 200
+
+    def test_google_search(self):
+        out = google_search('phenny')
+
+        m = re.match('^https?://.*$', out, flags=re.UNICODE)
+        self.assertTrue(m)
+
+    def test_g(self):
+        input = Mock(group=lambda x: 'swhack')
+        g(self.phenny, input)
+
+        self.phenny.reply.assert_not_called_with(
+                "Problem getting data from Google.")
+
+    def test_gc(self):
+        query = 'extrapolate'
+        input = Mock(group=lambda x: query)
+        gc(self.phenny, input)
+
+        out = self.phenny.say.call_args[0][0]
+        m = re.match('^{0}: [0-9,\.]+$'.format(query), out, flags=re.UNICODE)
+        self.assertTrue(m)
+
+    def test_gcs(self):
+        input = Mock(group=lambda x: 'vtluug virginia phenny')
+        gcs(self.phenny, input)
+
+        assert self.phenny.say.called == 1
+
+    def test_bing_search(self):
+        out = bing_search('phenny')
+
+        m = re.match('^https?://.*$', out, flags=re.UNICODE)
+        self.assertTrue(m)
+
+    def test_bing(self):
+        input = Mock(group=lambda x: 'swhack')
+        bing(self.phenny, input)
+
+        assert self.phenny.reply.called == 1
+
+    def test_duck_search(self):
+        out = duck_search('phenny')
+
+        m = re.match('^https?://.*$', out, flags=re.UNICODE)
+        self.assertTrue(m)
+
+    def test_duck(self):
+        input = Mock(group=lambda x: 'swhack')
+        duck(self.phenny, input)
+
+        assert self.phenny.reply.called == 1
+
+    def test_search(self):
+        input = Mock(group=lambda x: 'vtluug')
+        duck(self.phenny, input)
+
+        assert self.phenny.reply.called == 1
+
+    def test_suggest(self):
+        input = Mock(group=lambda x: 'vtluug')
+        suggest(self.phenny, input)
+
+        assert (self.phenny.reply.called == 1 or self.phenny.say.called == 1)