1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
import unittest
from mock import patch
from urllib2 import URLError, urlopen
import yaml
from test_mic import Mic
import modules
import brain
def activeInternet():
try:
urlopen('http://www.google.com', timeout=1)
return True
except URLError:
return False
class TestModules(unittest.TestCase):
def setUp(self):
self.profile = yaml.safe_load(open("profile.yml", "r"))
self.send = False
def runConversation(self, query, inputs, module):
"""Generic method for spoofing conversation.
Arguments:
query -- The initial input to the server.
inputs -- Additional input, if conversation is extended.
Returns:
The server's responses, in a list.
"""
self.assertTrue(module.isValid(query))
mic = Mic(inputs)
module.handle(query, mic, self.profile)
return mic.outputs
def testLife(self):
query = "What is the meaning of life?"
inputs = []
outputs = self.runConversation(query, inputs, modules.Life)
self.assertEqual(len(outputs), 1)
self.assertTrue("42" in outputs[0])
def testJoke(self):
query = "Tell me a joke."
inputs = ["Who's there?", "Random response"]
outputs = self.runConversation(query, inputs, modules.Joke)
self.assertEqual(len(outputs), 3)
allJokes = open("JOKES.txt", "r").read()
self.assertTrue(outputs[2] in allJokes)
def testTime(self):
query = "What time is it?"
inputs = []
self.runConversation(query, inputs, modules.Time)
@unittest.skipIf(not activeInternet(), "No internet connection")
def testGmail(self):
key = 'gmail_password'
if not key in self.profile or not self.profile[key]:
return
query = "Check my email"
inputs = []
self.runConversation(query, inputs, modules.Gmail)
@unittest.skipIf(not activeInternet(), "No internet connection")
def testHN(self):
query = "find me some of the top hacker news stories"
if self.send:
inputs = ["the first and third"]
else:
inputs = ["no"]
outputs = self.runConversation(query, inputs, modules.HN)
self.assertTrue("front-page articles" in outputs[1])
@unittest.skipIf(not activeInternet(), "No internet connection")
def testNews(self):
query = "find me some of the top news stories"
if self.send:
inputs = ["the first"]
else:
inputs = ["no"]
outputs = self.runConversation(query, inputs, modules.News)
self.assertTrue("top headlines" in outputs[1])
@unittest.skipIf(not activeInternet(), "No internet connection")
def testWeather(self):
query = "what's the weather like tomorrow"
inputs = []
outputs = self.runConversation(query, inputs, modules.Weather)
self.assertTrue(
"can't see that far ahead" in outputs[0]
or "Tomorrow" in outputs[0])
class TestBrain(unittest.TestCase):
@staticmethod
def _emptyBrain():
mic = Mic([])
profile = yaml.safe_load(open("profile.yml", "r"))
return brain.Brain(mic, profile)
@patch.object(brain, 'logError')
def testLog(self, logError):
"""Does Brain correctly log errors when raised by modules?"""
my_brain = TestBrain._emptyBrain()
unclear = my_brain.modules[-1]
with patch.object(unclear, 'handle') as mocked_handle:
mocked_handle.side_effect = KeyError('foo')
my_brain.query("zzz gibberish zzz")
logError.assert_called_with()
def testSortByPriority(self):
"""Does Brain sort modules by priority?"""
my_brain = TestBrain._emptyBrain()
priorities = filter(lambda m: hasattr(m, 'PRIORITY'), my_brain.modules)
target = sorted(priorities, key=lambda m: m.PRIORITY, reverse=True)
self.assertEqual(target, priorities)
def testPriority(self):
"""Does Brain correctly send query to higher-priority module?"""
my_brain = TestBrain._emptyBrain()
hn_module = 'modules.HN'
hn = filter(lambda m: m.__name__ == hn_module, my_brain.modules)[0]
with patch.object(hn, 'handle') as mocked_handle:
my_brain.query("hacker news")
self.assertTrue(mocked_handle.called)
if __name__ == '__main__':
unittest.main()
|
