def test_close_fd(self, conn_class_mock): conn_class_mock.return_value = conn = MagicMock() conn.close.side_effect = ValueError ctx = _PoolableAmqp(one=1) self.assertRaises(ValueError, ctx.close) conn_class_mock.assert_called_with(one=1) conn.sock.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.send_heartbeat.assert_called_with()
def test_close(self, conn_class_mock): conn_class_mock.return_value = conn = MagicMock() ctx = _PoolableAmqp(one=1) ctx.close() conn_class_mock.assert_called_with(one=1) conn.close.assert_called_with()
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.send_heartbeat.assert_called_with()