From 9a8af1499ca425366d934487469d9977fae7fe5f Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Thu, 7 Jan 2016 17:57:24 -0800 Subject: Fix KafkaClient->SimpleClient references --- test/test_producer.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'test/test_producer.py') diff --git a/test/test_producer.py b/test/test_producer.py index 227d4ad..aa4f0be 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -8,7 +8,7 @@ import time from mock import MagicMock, patch from . import unittest -from kafka import KafkaClient, SimpleProducer, KeyedProducer +from kafka import SimpleClient, SimpleProducer, KeyedProducer from kafka.common import ( AsyncProducerQueueFull, FailedPayloadsError, NotLeaderForPartitionError, ProduceResponsePayload, RetryOptions, TopicPartition @@ -89,11 +89,11 @@ class TestKafkaProducer(unittest.TestCase): def test_producer_sync_fail_on_error(self): error = FailedPayloadsError('failure') - with patch.object(KafkaClient, 'load_metadata_for_topics'): - with patch.object(KafkaClient, 'get_partition_ids_for_topic', return_value=[0, 1]): - with patch.object(KafkaClient, '_send_broker_aware_request', return_value = [error]): + with patch.object(SimpleClient, 'load_metadata_for_topics'): + with patch.object(SimpleClient, 'get_partition_ids_for_topic', return_value=[0, 1]): + with patch.object(SimpleClient, '_send_broker_aware_request', return_value = [error]): - client = KafkaClient(MagicMock()) + client = SimpleClient(MagicMock()) producer = SimpleProducer(client, async=False, sync_fail_on_error=False) # This should not raise -- cgit v1.2.1