def test_check(self, conn_class_mock):
conn_class_mock.return_value = conn = MagicMock()
ctx = _PoolableAmqp(one=1)
self.assertTrue(ctx.check())
conn_class_mock.assert_called_with(one=1)
conn.channel.assert_called_with()
conn.send_heartbeat.assert_called_with()
def test_close(self, conn_class_mock):
conn_class_mock.return_value = conn = MagicMock()
conn.channel.return_value = channel = MagicMock()
ctx = _PoolableAmqp(one=1)
ctx.close()
conn_class_mock.assert_called_with(one=1)
channel.close.assert_called_with()
conn.close.assert_called_with()
def test_check_fail(self, conn_class_mock):
conn_class_mock.return_value = conn = MagicMock()
conn.send_heartbeat.side_effect = ValueError
ctx = _PoolableAmqp(one=1)
self.assertFalse(ctx.check())
conn_class_mock.assert_called_with(one=1)
conn.channel.assert_called_with()
conn.send_heartbeat.assert_called_with()
def test_close_fd(self, conn_class_mock):
conn_class_mock.return_value = conn = MagicMock()
conn.channel.return_value = channel = MagicMock()
channel.close.side_effect = ValueError
ctx = _PoolableAmqp(one=1)
self.assertRaises(ValueError, ctx.close)
conn_class_mock.assert_called_with(one=1)
channel.close.assert_called_with()
conn.sock.close.assert_called_with()
