def test_quotas_sets_defaults_get_when_service_is_enabled(self):
filters = {'user': {'tenant_id': 'tenant'}}
request = self.mock_rest_request(**{'GET': dict(filters)})
self.mock_is_service_enabled.return_value = True
self.mock_tenant_quota_get.return_value = [
base.Quota("network", 100),
base.Quota("q2", 101)
]
response = neutron.DefaultQuotaSets().get(request)
self.assertStatusCode(response, 200)
self.assertItemsCollectionEqual(response, [{
'limit': 100,
'display_name': 'Networks',
'name': 'network'
}, {
'limit': 101,
'display_name': 'Q2',
'name': 'q2'
}])
self.mock_is_service_enabled.assert_called_once_with(
request, 'network')
self.mock_tenant_quota_get.assert_called_once_with(
request, request.user.tenant_id)
def test_quota_sets_defaults_get_when_service_is_disabled(
self, mock_is_service_enabled):
filters = {'user': {'tenant_id': 'tenant'}}
request = self.mock_rest_request(**{'GET': dict(filters)})
mock_is_service_enabled.return_value = False
response = neutron.DefaultQuotaSets().get(request)
self.assertStatusCode(response, 501)
self.assertEqual(response.content.decode('utf-8'),
'"Service Neutron is disabled."')
mock_is_service_enabled.assert_called_once_with(request, 'network')
def test_quota_sets_defaults_get_when_service_is_disabled(self, client):
filters = {'user': {'tenant_id': 'tenant'}}
request = self.mock_rest_request(**{'GET': dict(filters)})
base.is_service_enabled(request, 'network').AndReturn(False)
self.mox.ReplayAll()
response = neutron.DefaultQuotaSets().get(request)
self.assertStatusCode(response, 501)
self.assertEqual(response.content.decode('utf-8'),
'"Service Neutron is disabled."')
client.tenant_quota_get.assert_not_called()