def setup_conn(self):
self.conn = MagicMock(name='connection')
self.conn.is_closing = False
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name='send_method')
def test_init_explicit_channel(self):
Channel(self.conn, 3)
self.conn._claim_channel_id.assert_called_with(3)
def test_init_auto_channel(self):
c = Channel(self.conn, None)
self.conn._get_free_channel_id.assert_called_with()
assert c.channel_id is self.conn._get_free_channel_id()
def test_init_confirm_disabled(self):
self.conn.confirm_publish = False
c = Channel(self.conn, 2)
assert c.basic_publish == c._basic_publish
def test_init_confirm_enabled(self):
self.conn.confirm_publish = True
c = Channel(self.conn, 2)
assert c.basic_publish == c.basic_publish_confirm
def test_init_confirm_disabled(self):
self.conn.confirm_publish = False
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c._basic_publish)
def test_init_confirm_enabled(self):
self.conn.confirm_publish = True
c = Channel(self.conn, 2)
self.assertEqual(c.basic_publish, c.basic_publish_confirm)
def setup(self):
self.conn = Mock(name='connection')
self.conn.channels = {}
self.conn._get_free_channel_id.return_value = 2
self.c = Channel(self.conn, 1)
self.c.send_method = Mock(name='send_method')
