def test_deserialize_called_with_response_body(self):
response = Response("Hello, World!", 0)
sender = MockSender(response)
deserializer = FakeDeserializer(None)
client = Client(sender, deserializer)
client.send_lookup(Lookup())
self.assertEqual(response.payload, deserializer.input)
def test_deserialize_called_with_response_body(self):
response = Response('Hello, World!', 0)
sender = MockSender(response)
deserializer = FakeDeserializer({})
client = Client(sender, deserializer)
client.send(Lookup('1', '2'))
self.assertEqual(response.payload, deserializer.input)
def test_result_correctly_assigned_to_corresponding_lookup(self):
lookup = Lookup('1')
expected_result = {"candidates": [{"street": "2"}]}
sender = MockSender(Response('{[]}', 0))
deserializer = FakeDeserializer(expected_result)
client = Client(sender, deserializer)
client.send(lookup)
self.assertEqual('2', lookup.result[0].street)
def test_result_correctly_assigned_to_corresponding_lookup(self):
raw_result = {"meta": {}, "addresses": [{"text": "Hello, World!"}]}
expected_result = Result(raw_result)
lookup = Lookup('Hello, World!')
sender = MockSender(Response('[]', 0))
deserializer = FakeDeserializer(raw_result)
client = Client(sender, deserializer)
client.send(lookup)
self.assertEqual(expected_result.addresses[0].text, lookup.result.addresses[0].text)
def test_results_correctly_assigned_to_corresponding_lookup(self):
raw_results = [{"input_index": 0}, {"input_index": 1}]
expected_results = [Result(raw_results[0]), Result(raw_results[1])]
batch = Batch()
batch.add(Lookup())
batch.add(Lookup())
sender = MockSender(Response("[]", 0))
deserializer = FakeDeserializer(raw_results)
client = Client(sender, deserializer)
client.send_batch(batch)
self.assertEqual(expected_results[0].input_index,
batch[0].result.input_index)
self.assertEqual(expected_results[1].input_index,
batch[1].result.input_index)
def test_candidates_correctly_assigned_to_lookup(self):
raw_candidates = [{'address1': 'street 1'}, {'address1': 'street 2'}]
expected_candidates = [
Candidate(raw_candidates[0]),
Candidate(raw_candidates[1])
]
lookup = Lookup('1', '2')
sender = MockSender(Response('[]', 0))
deserializer = FakeDeserializer(raw_candidates)
client = Client(sender, deserializer)
client.send(lookup)
self.assertEqual(expected_candidates[0].address1,
lookup.result[0].address1)
self.assertEqual(expected_candidates[1].address1,
lookup.result[1].address1)
def test_candidates_correctly_assigned_to_corresponding_lookup(self):
candidate0 = {
'input_index': 0,
'candidate_index': 0,
'addressee': 'Mister 0'
}
candidate1 = {
'input_index': 1,
'candidate_index': 0,
'addressee': 'Mister 1'
}
candidate2 = {
'input_index': 1,
'candidate_index': 1,
'addressee': 'Mister 2'
}
raw_candidates = [candidate0, candidate1, candidate2]
expected_candidates = [
Candidate(candidate0),
Candidate(candidate1),
Candidate(candidate2)
]
batch = Batch()
batch.add(Lookup())
batch.add(Lookup())
sender = MockSender(Response("[]", 0))
deserializer = FakeDeserializer(raw_candidates)
client = Client(sender, deserializer)
client.send_batch(batch)
self.assertEqual(expected_candidates[0].addressee,
batch[0].result[0].addressee)
self.assertEqual(expected_candidates[1].addressee,
batch[1].result[0].addressee)
self.assertEqual(expected_candidates[2].addressee,
batch[1].result[1].addressee)
def send(self, request):
response = Response(None,
self.status_codes[self.current_status_code_index])
self.current_status_code_index += 1
return response
def send(self, request):
if not self.exception:
return None
else:
return Response(None, None, self.exception)
def send(self, request):
self.request = request
return Response("[]", 200)
