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