def test_collect(self):
channels = self.conn.channels = {
0: self.conn, 1: Mock(name='c1'), 2: Mock(name='c2'),
}
transport = self.conn.transport
self.conn.collect()
transport.close.assert_called_with()
for i, channel in items(channels):
if i:
channel.collect.assert_called_with()
def test_collect(self):
channels = self.conn.channels = {
0: self.conn, 1: Mock(name='c1'), 2: Mock(name='c2'),
}
transport = self.conn.transport
self.conn.collect()
transport.close.assert_called_with()
for i, channel in items(channels):
if i:
channel.collect.assert_called_with()
