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'])