Commit 3f4f880d by Dudás Ádám Committed by Kálmán Viktor

one: fix set_quota test

parent daf265d4
...@@ -9,7 +9,6 @@ from ..models import (Disk, Instance, InstanceType, Network, Share, ...@@ -9,7 +9,6 @@ from ..models import (Disk, Instance, InstanceType, Network, Share,
Template, set_quota, reset_keys, OpenSshKeyValidator) Template, set_quota, reset_keys, OpenSshKeyValidator)
from ..util import keygen from ..util import keygen
from school.models import Course, Group, Semester from school.models import Course, Group, Semester
from store.api import StoreApi
class UserCloudDetailsTestCase(TestCase): class UserCloudDetailsTestCase(TestCase):
...@@ -35,8 +34,8 @@ class UserCloudDetailsTestCase(TestCase): ...@@ -35,8 +34,8 @@ class UserCloudDetailsTestCase(TestCase):
self.userdetails = user.cloud_details self.userdetails = user.cloud_details
date = datetime.now().date() date = datetime.now().date()
delta = timedelta(weeks=7) delta = timedelta(weeks=7)
sem = Semester.objects.create(name="testsem", start=date-delta, sem = Semester.objects.create(name="testsem", start=date - delta,
end=date+delta) end=date + delta)
course1 = Course.objects.create(code='tccode1', name='testcourse1', course1 = Course.objects.create(code='tccode1', name='testcourse1',
short_name='tc1') short_name='tc1')
grp1 = Group.objects.create(name="testgroup1", semester=sem, grp1 = Group.objects.create(name="testgroup1", semester=sem,
...@@ -119,12 +118,16 @@ def delete_user(): ...@@ -119,12 +118,16 @@ def delete_user():
@with_setup(create_user, delete_user) @with_setup(create_user, delete_user)
def test_set_quota(): @patch('one.models.StoreApi')
def test_set_quota(MockStoreApi):
MockStoreApi.userexist = Mock(return_value=True)
MockStoreApi.set_quota = Mock()
user = User.objects.get(username="testuser") user = User.objects.get(username="testuser")
details = user.cloud_details details = user.cloud_details
set_quota(None, details, None) set_quota(None, details, None)
assert StoreApi.userexist(user.username) MockStoreApi.userexist.assert_called_once_with(user.username)
# TODO check quota value MockStoreApi.set_quota.assert_called_once_with(user.username,
details.disk_quota * 1024)
@with_setup(create_user, delete_user) @with_setup(create_user, delete_user)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment