def test_connection_wait_for_connection(self):
connection = Connection('127.0.0.1', 'guest', 'guest', lazy=True)
connection.set_state(connection.OPENING)
io = IO(connection.parameters, [])
io.socket = Mock(name='socket', spec=socket.socket)
connection._io = io
self.assertFalse(connection.is_open)
def set_state_to_open(conn):
conn.set_state(conn.OPEN)
threading.Timer(function=set_state_to_open,
interval=0.1, args=(connection,)).start()
connection._wait_for_connection_state(connection.OPEN)
self.assertTrue(connection.is_open)
