Revision 8c7f5877

b/snf-cyclades-app/synnefo/helpdesk/tests.py
37 37
from django.test import TestCase, Client
38 38
from django.conf import settings
39 39
from django.core.urlresolvers import reverse
40
from snf_django.utils.testing import mocked_quotaholder
40 41

  
41 42

  
42 43
USER1 = "5edcb5aa-1111-4146-a8ed-2b6287824353"
......
322 323
        shutdown.return_value = 1
323 324
        self.vm1.operstate = 'STARTED'
324 325
        self.vm1.save()
325
        r = self.client.post(reverse('helpdesk-vm-shutdown', args=(pk,)),
326
                             data={'token': '0001'}, user_token='0001')
326
        with mocked_quotaholder():
327
            r = self.client.post(reverse('helpdesk-vm-shutdown', args=(pk,)),
328
                                 data={'token': '0001'}, user_token='0001')
327 329
        self.assertEqual(r.status_code, 302)
328 330
        self.assertTrue(shutdown.called)
329 331
        self.assertEqual(len(shutdown.mock_calls), 1)
......
332 334
        startup.return_value = 2
333 335
        self.vm1.operstate = 'STOPPED'
334 336
        self.vm1.save()
335
        r = self.client.post(reverse('helpdesk-vm-start', args=(pk,)),
336
                             data={'token': '0001'}, user_token='0001')
337
        with mocked_quotaholder():
338
            r = self.client.post(reverse('helpdesk-vm-start', args=(pk,)),
339
                                 data={'token': '0001'}, user_token='0001')
337 340
        self.assertEqual(r.status_code, 302)
338 341
        self.assertTrue(startup.called)
339 342
        self.assertEqual(len(startup.mock_calls), 1)

Also available in: Unified diff