def test_connection_close_channels(self): connection = Connection('127.0.0.1', 'guest', 'guest', timeout=0.1, lazy=True) connection._channels[0] = FakeChannel() connection._channels[1] = FakeChannel() connection._channels[2] = FakeChannel(FakeChannel.CLOSED) self.assertTrue(connection._channels[0].is_open) self.assertTrue(connection._channels[1].is_open) self.assertTrue(connection._channels[2].is_closed) connection._close_remaining_channels() self.assertTrue(connection._channels[0].is_closed) self.assertTrue(connection._channels[1].is_closed) self.assertTrue(connection._channels[2].is_closed)
def test_connection_close_channels(self): connection = Connection('127.0.0.1', 'guest', 'guest', lazy=True) channel_1 = Channel(1, connection, 360) channel_2 = Channel(2, connection, 360) channel_3 = Channel(3, connection, 360) channel_1.set_state(Channel.CLOSED) channel_2.set_state(Channel.CLOSED) channel_2.set_state(Channel.OPEN) connection._channels[1] = channel_1 connection._channels[2] = channel_2 connection._channels[3] = channel_3 connection._close_remaining_channels() self.assertTrue(channel_1.is_closed) self.assertTrue(channel_2.is_closed) self.assertTrue(channel_3.is_closed) self.assertFalse(connection._channels)