def test_connection(self): conn = BrokerConnection('localhost', 9092, socket.AF_INET) assert conn.connected() is False conn.state = ConnectionStates.CONNECTED assert conn.connected() is True assert conn.connecting() is False conn.state = ConnectionStates.CONNECTING assert conn.connecting() is True conn.state = ConnectionStates.CONNECTED assert conn.connecting() is False conn.state = ConnectionStates.DISCONNECTED assert conn.connected() is False