Revision 184a2a8c snf-cyclades-app/synnefo/logic/tests.py
b/snf-cyclades-app/synnefo/logic/tests.py | ||
---|---|---|
43 | 43 |
from synnefo.api.util import allocate_resource |
44 | 44 |
from synnefo.logic.callbacks import (update_db, update_net, update_network, |
45 | 45 |
update_build_progress) |
46 |
from snf_django.utils.testing import mocked_quotaholder |
|
46 | 47 |
|
47 | 48 |
now = datetime.now |
48 | 49 |
from time import time |
... | ... | |
135 | 136 |
mfactory.NetworkInterfaceFactory(machine=vm) |
136 | 137 |
msg = self.create_msg(operation='OP_INSTANCE_REMOVE', |
137 | 138 |
instance=vm.backend_vm_id) |
138 |
update_db(client, msg) |
|
139 |
with mocked_quotaholder(): |
|
140 |
update_db(client, msg) |
|
139 | 141 |
client.basic_ack.assert_called_once() |
140 | 142 |
db_vm = VirtualMachine.objects.get(id=vm.id) |
141 | 143 |
self.assertEqual(db_vm.operstate, 'DESTROYED') |
... | ... | |
170 | 172 |
mfactory.NetworkInterfaceFactory(machine=vm) |
171 | 173 |
msg = self.create_msg(operation='OP_INSTANCE_REMOVE', |
172 | 174 |
instance=vm.backend_vm_id) |
173 |
update_db(client, msg) |
|
175 |
with mocked_quotaholder(): |
|
176 |
update_db(client, msg) |
|
174 | 177 |
client.basic_ack.assert_called_once() |
175 | 178 |
db_vm = VirtualMachine.objects.get(id=vm.id) |
176 | 179 |
self.assertEqual(db_vm.operstate, 'DESTROYED') |
... | ... | |
395 | 398 |
msg = self.create_msg(operation='OP_NETWORK_REMOVE', |
396 | 399 |
network=net.backend_id, |
397 | 400 |
cluster=bn.backend.clustername) |
398 |
update_network(client, msg) |
|
401 |
with mocked_quotaholder(): |
|
402 |
update_network(client, msg) |
|
399 | 403 |
client.basic_ack.assert_called_once() |
400 | 404 |
db_bnet = BackendNetwork.objects.get(id=bn.id) |
401 | 405 |
self.assertEqual(db_bnet.operstate, |
... | ... | |
422 | 426 |
msg = self.create_msg(operation='OP_NETWORK_REMOVE', |
423 | 427 |
network=net.backend_id, |
424 | 428 |
cluster=bn1.backend.clustername) |
425 |
update_network(client, msg) |
|
429 |
with mocked_quotaholder(): |
|
430 |
update_network(client, msg) |
|
426 | 431 |
client.basic_ack.assert_called_once() |
427 | 432 |
new_net = Network.objects.get(id=net.id) |
428 | 433 |
self.assertEqual(new_net.state, 'DELETED') |
Also available in: Unified diff