diff options
Diffstat (limited to 'test/testutil.py')
| -rw-r--r-- | test/testutil.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/test/testutil.py b/test/testutil.py index 1f1a1df..b9827a8 100644 --- a/test/testutil.py +++ b/test/testutil.py @@ -79,9 +79,15 @@ class KafkaIntegrationTestCase(unittest.TestCase): self.client.close() def current_offset(self, topic, partition): - offsets, = self.client.send_offset_request([ OffsetRequest(kafka_bytestring(topic), - partition, -1, 1) ]) - return offsets.offsets[0] + try: + offsets, = self.client.send_offset_request([ OffsetRequest(kafka_bytestring(topic), partition, -1, 1) ]) + except: + # XXX: We've seen some UnknownErrors here and cant debug w/o server logs + self.zk.child.dump_logs() + self.server.child.dump_logs() + raise + else: + return offsets.offsets[0] def msgs(self, iterable): return [ self.msg(x) for x in iterable ] |