def test_get_service_http_error(self):
client = MagicMock()
organisation_id = 'organisation-id'
name = 'service-name'
client.accounts.services.get.side_effect = HTTPError(400, message='Error Message')
with pytest.raises(HTTPError) as exc:
_get_service(client, organisation_id, name)
assert exc.value.message == 'HTTP 400: Error Message'
def test_get_service_404_error(self):
client = MagicMock()
organisation_id = 'organisation-id'
name = 'service-name'
client.accounts.services.get.side_effect = HTTPError(404, message='Not Found')
with pytest.raises(click.ClickException) as exc:
_get_service(client, organisation_id, name)
assert exc.value.message == ('\x1b[31mOrganisation organisation-id cannot be found. '
'Please check organisation_id.\x1b[0m')
def test_get_service_no_results(self):
client = MagicMock()
organisation_id = 'organisation-id'
name = 'service-name'
client.accounts.services.get.return_value = {'status': 200, 'data': []}
result = _get_service(client, organisation_id, name)
assert result is None
def test_get_service_multiple_names(self):
client = MagicMock()
organisation_id = 'organisation-id'
name = 'service-name'
client.accounts.services.get.return_value = {'status': 200, 'data': [
{
'id': 'service1',
'name': 'service-name'
}, {
'id': 'service2',
'name': 'service-name'
}
]}
result = _get_service(client, organisation_id, name)
client.accounts.services.get.assert_called_once_with(organisation_id='organisation-id')
assert result == 'service1'