def handle_read(self):
data = self.recv(8192).strip()
if data == encode_utf8('I know kung fu.'):
reply = encode_utf8('Show me.')
else:
reply = encode_utf8('Blue Pill.')
self.send(reply + encode_utf8('\r\n'))
self.close()
def test_empty_getresponse(self):
entry = MocketEntry(('localhost', 8080), [])
self.assertEqual(entry.get_response(), encode_utf8(''))
def test_getresponse(self):
entry = MocketEntry(('localhost', 8080), ['Show me.\r\n'])
self.assertEqual(entry.get_response(), encode_utf8('Show me.\r\n'))
def iknow(self):
self._fp = self.sock.makefile('rb')
self.sock.sendall(encode_utf8('I know kung fu.\r\n'))
return self._fp.read().strip() == encode_utf8('Show me.')
