def test_server_send(self):
"sendData method should send the appropriate data"
server = UrlFactory()
server_proto = server.buildProtocol(('localhost', 0))
server_proto.transport = proto_helpers.StringTransport()
server_proto.sendData('data')
self.assertEqual(server_proto.transport.value(), 'data')
def test_server_factory_search_for(self):
"factory.search_for should return a json dict of found keywords"
server = UrlFactory()
test_page = '<html><body><h1>Test Page!</h1></body></html>'
self.assertEquals(server.search_for(test_page, ['Test']), '{"kw": ["Test"]}')
self.assertEquals(server.search_for(test_page, ['Test Page']), '{"kw": ["Test Page"]}')
self.assertEquals(server.search_for(test_page, ['A different page!']), '{"kw": []}')
def test_server_receive(self):
"The server should receive sent data"
server = UrlFactory()
server_proto = server.buildProtocol(('localhost', 0))
test_dict = json.dumps({'url': 'www.example.com', 'kw': ['example']})
server_proto.transport = proto_helpers.StringTransport()
server_proto.dataReceived(test_dict)
self.assertEqual(server_proto.url, 'www.example.com')
self.assertEqual(server_proto.kw, ['example'])
