def test_401(self): self.client_mock.get_queues.side_effect = HTTPError('', status=401) with self.assertRaises(RabbitCritical) as excinfo: get_queues(self.client_mock, self.vhost_mock) self.assertEqual(excinfo.exception.errors, {'all': ['Unauthorized.']}) self.client_mock.get_queues.assert_called_once_with(self.vhost_mock)
def test_unknown_error(self): self.client_mock.get_queues.side_effect = HTTPError('', status=500) with self.assertRaises(RabbitCritical) as excinfo: get_queues(self.client_mock, self.vhost_mock) self.assertEqual(excinfo.exception.errors, {'all': ['Unhandled HTTP error, status: 500']}) self.client_mock.get_queues.assert_called_once_with(self.vhost_mock)
def test_network_error(self): self.client_mock.get_queues.side_effect = NetworkError() with self.assertRaises(RabbitCritical) as excinfo: get_queues(self.client_mock, self.vhost_mock) self.assertEqual(excinfo.exception.errors, {'all': ['Can not communicate with RabbitMQ.']}) self.client_mock.get_queues.assert_called_once_with(self.vhost_mock)
def test_ok(self): res = get_queues(self.client_mock, self.vhost_mock) self.assertEqual(res, self.client_mock.get_queues.return_value) self.client_mock.get_queues.assert_called_once_with(self.vhost_mock)