diff options
Diffstat (limited to 'test/test_failover_integration.py')
| -rw-r--r-- | test/test_failover_integration.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_failover_integration.py b/test/test_failover_integration.py index 8c5efe2..04c9e2b 100644 --- a/test/test_failover_integration.py +++ b/test/test_failover_integration.py @@ -3,7 +3,7 @@ import os import time from kafka import KafkaClient, SimpleConsumer, KeyedProducer -from kafka.common import TopicAndPartition, FailedPayloadsError, ConnectionError +from kafka.common import TopicPartition, FailedPayloadsError, ConnectionError from kafka.producer.base import Producer from test.fixtures import ZookeeperFixture, KafkaFixture @@ -202,7 +202,7 @@ class TestFailover(KafkaIntegrationTestCase): break def _kill_leader(self, topic, partition): - leader = self.client.topics_to_brokers[TopicAndPartition(topic, partition)] + leader = self.client.topics_to_brokers[TopicPartition(topic, partition)] broker = self.brokers[leader.nodeId] broker.close() return broker |