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