87 lines
2.3 KiB
Python
87 lines
2.3 KiB
Python
|
"""
|
||
|
|
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)
|