*** test/test_conn.py	Fri Sep 24 08:12:01 2021
--- test/test_conn.py	Fri Sep 24 08:12:13 2021
***************
*** 4,10 ****
  from errno import EALREADY, EINPROGRESS, EISCONN, ECONNRESET
  import socket
  
! import mock
  import pytest
  
  from kafka.conn import BrokerConnection, ConnectionStates, collect_hosts
--- 4,14 ----
  from errno import EALREADY, EINPROGRESS, EISCONN, ECONNRESET
  import socket
  
! try:
!     from unittest import mock
! except ImportError:
!     import mock
! 
  import pytest
  
  from kafka.conn import BrokerConnection, ConnectionStates, collect_hosts