diff options
| author | David Arthur <mumrah@gmail.com> | 2013-04-01 22:06:20 -0400 |
|---|---|---|
| committer | David Arthur <mumrah@gmail.com> | 2013-04-02 20:20:32 -0400 |
| commit | 1c5081e6b3237067308b5819a121e420e45d56b0 (patch) | |
| tree | ceafa894a7fe6768840f5330e3140d456473769d /test/integration.py | |
| parent | fa13fa88d8a32943535c63e69ed111007ac1a3d3 (diff) | |
| download | kafka-python-1c5081e6b3237067308b5819a121e420e45d56b0.tar.gz | |
Update kafka-src to latest 0.8
Fix a broken test (100k was too much to send in one batch)
Diffstat (limited to 'test/integration.py')
| -rw-r--r-- | test/integration.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/test/integration.py b/test/integration.py index 04d5979..a1fcce7 100644 --- a/test/integration.py +++ b/test/integration.py @@ -225,15 +225,26 @@ class TestKafkaClient(unittest.TestCase): def test_produce_100k_gzipped(self): - produce = ProduceRequest("test_produce_100k_gzipped", 0, messages=[ - create_gzip_message(["Gzipped %d" % i for i in range(100000)]) + req1 = ProduceRequest("test_produce_100k_gzipped", 0, messages=[ + create_gzip_message(["Gzipped batch 1, message %d" % i for i in range(50000)]) ]) - for resp in self.client.send_produce_request([produce]): + for resp in self.client.send_produce_request([req1]): self.assertEquals(resp.error, 0) self.assertEquals(resp.offset, 0) (offset, ) = self.client.send_offset_request([OffsetRequest("test_produce_100k_gzipped", 0, -1, 1)]) + self.assertEquals(offset.offsets[0], 50000) + + req2 = ProduceRequest("test_produce_100k_gzipped", 0, messages=[ + create_gzip_message(["Gzipped batch 2, message %d" % i for i in range(50000)]) + ]) + + for resp in self.client.send_produce_request([req2]): + self.assertEquals(resp.error, 0) + self.assertEquals(resp.offset, 50000) + + (offset, ) = self.client.send_offset_request([OffsetRequest("test_produce_100k_gzipped", 0, -1, 1)]) self.assertEquals(offset.offsets[0], 100000) ##################### |