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)
