def __init__(self, ctx):
super(Quotas, self).__init__(ctx)
self.clients = osclients.Clients(self.context["admin"]["credential"])
self.manager = {
"nova": nova_quotas.NovaQuotas(self.clients),
"cinder": cinder_quotas.CinderQuotas(self.clients),
"manila": manila_quotas.ManilaQuotas(self.clients),
"designate": designate_quotas.DesignateQuotas(self.clients),
"neutron": neutron_quotas.NeutronQuotas(self.clients)
}
self.original_quotas = []
def test_update(self):
mock_clients = mock.MagicMock()
cinder_quo = cinder_quotas.CinderQuotas(mock_clients)
tenant_id = mock.MagicMock()
quotas_values = {
"volumes": 10,
"snapshots": 50,
"backups": 20,
"backup_gigabytes": 1000,
"gigabytes": 1000
}
cinder_quo.update(tenant_id, **quotas_values)
mock_clients.cinder().quotas.update.assert_called_once_with(
tenant_id, **quotas_values)
def test_get(self):
tenant_id = "tenant_id"
quotas = {
"gigabytes": "gb",
"snapshots": "ss",
"volumes": "v",
"backups": "b",
"backup_gigabytes": "b_g"
}
quota_set = mock.MagicMock(**quotas)
clients = mock.MagicMock()
clients.cinder.return_value.quotas.get.return_value = quota_set
cinder_quo = cinder_quotas.CinderQuotas(clients)
self.assertEqual(quotas, cinder_quo.get(tenant_id))
clients.cinder().quotas.get.assert_called_once_with(tenant_id)
def test_delete(self):
mock_clients = mock.MagicMock()
cinder_quo = cinder_quotas.CinderQuotas(mock_clients)
tenant_id = mock.MagicMock()
cinder_quo.delete(tenant_id)
mock_clients.cinder().quotas.delete.assert_called_once_with(tenant_id)