def test_client_lost_connection(self):
"Tests to make sure that the deferred is called upon a lost connection"
d = FakeDeferred()
client = UrlClientFactory('www.example.com', ['example'], d)
proto = client.buildProtocol(('localhost', 0))
proto.connectionLost('reason')
self.assertEquals(d.callback, "called back")
def test_client_receive(self):
"Tests to make sure that the client is receiving data properly"
d = FakeDeferred()
client = UrlClientFactory('www.example.com', ['example'], d)
proto = client.buildProtocol(('localhost', 0))
transport = proto_helpers.StringTransport()
proto.makeConnection(transport)
test_dict = json.dumps({'url': 'www.example.com', 'kw': ['example']})
proto.dataReceived(test_dict)
self.assertEqual(proto.data, test_dict)
def test_client_send(self):
"Tests to make sure that the client sends the correct data to the server"
d = FakeDeferred()
client = UrlClientFactory('www.example.com', ['example'], d)
proto = client.buildProtocol(('localhost', 0))
transport = proto_helpers.StringTransport()
proto.makeConnection(transport)
print "transport", transport.value()
test_dict = json.dumps({'url': 'www.example.com', 'kw': ['example']})
self.assertEqual(transport.value(), test_dict)
