diff options
Diffstat (limited to 'test/test_conn_legacy.py')
| -rw-r--r-- | test/test_conn_legacy.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/test_conn_legacy.py b/test/test_conn_legacy.py index f0ef8fb..347588e 100644 --- a/test/test_conn_legacy.py +++ b/test/test_conn_legacy.py @@ -5,9 +5,10 @@ from threading import Thread import mock from . import unittest -from kafka.common import ConnectionError +from kafka.errors import ConnectionError from kafka.conn import KafkaConnection, collect_hosts, DEFAULT_SOCKET_TIMEOUT_SECONDS + class ConnTest(unittest.TestCase): def setUp(self): |