def test_empty_batch_not_sent(self):
sender = RequestCapturingSender()
client = Client(sender, None)
client.send_batch(Batch())
self.assertIsNone(sender.request)
def test_Successfully_Sends_Batch(self):
expected_payload = "Hello, World!"
sender = RequestCapturingSender()
serializer = FakeSerializer(expected_payload)
client = Client(sender, serializer)
batch = Batch()
batch.add(Lookup())
batch.add(Lookup())
client.send_batch(batch)
self.assertEqual(expected_payload, sender.request.payload)
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)