Revision d08a5f6f db/tests.py
b/db/tests.py | ||
---|---|---|
24 | 24 |
djuser.is_staff = True |
25 | 25 |
djuser.save() |
26 | 26 |
|
27 |
user = OceanUser(pk=1, name='Test User', credit=0, quota=100, monthly_rate=10)
|
|
27 |
user = SynnefoUser(pk=1, name='Test User', credit=0, quota=100, monthly_rate=10)
|
|
28 | 28 |
user.created = datetime.datetime.now() |
29 | 29 |
user.user = djuser |
30 | 30 |
user.save() |
... | ... | |
39 | 39 |
# test the allocator |
40 | 40 |
credit_allocator.allocate_credit() |
41 | 41 |
|
42 |
user = OceanUser.objects.get(pk=1)
|
|
42 |
user = SynnefoUser.objects.get(pk=1)
|
|
43 | 43 |
self.assertEquals(user.credit, 10, 'Allocation of credits failed, credit: %d (should be 10)' % ( user.credit, ) ) |
44 | 44 |
|
45 | 45 |
# test if the quota policy is endorced |
46 | 46 |
for i in range(1, 10): |
47 | 47 |
credit_allocator.allocate_credit() |
48 | 48 |
|
49 |
user = OceanUser.objects.get(pk=1)
|
|
49 |
user = SynnefoUser.objects.get(pk=1)
|
|
50 | 50 |
self.assertEquals(user.credit, user.quota, 'User exceeded quota! (cr:%d, qu:%d)' % ( user.credit, user.quota) ) |
51 | 51 |
|
52 | 52 |
|
... | ... | |
93 | 93 |
djuser.save() |
94 | 94 |
|
95 | 95 |
# add a user |
96 |
user = OceanUser(pk=1, name='Test User', credit=100, quota=100, monthly_rate=10)
|
|
96 |
user = SynnefoUser(pk=1, name='Test User', credit=100, quota=100, monthly_rate=10)
|
|
97 | 97 |
user.created = datetime.datetime.now() |
98 | 98 |
user.user = djuser |
99 | 99 |
user.save() |
... | ... | |
137 | 137 |
# charge when the vm is running |
138 | 138 |
charger.charge() |
139 | 139 |
|
140 |
user = OceanUser.objects.get(pk=1)
|
|
140 |
user = SynnefoUser.objects.get(pk=1)
|
|
141 | 141 |
self.assertEquals(user.credit, 90, 'Error in charging process (%d!=90, running)' % ( user.credit, )) |
142 | 142 |
|
143 | 143 |
# charge when the vm is stopped |
... | ... | |
148 | 148 |
|
149 | 149 |
charger.charge() |
150 | 150 |
|
151 |
user = OceanUser.objects.get(pk=1)
|
|
151 |
user = SynnefoUser.objects.get(pk=1)
|
|
152 | 152 |
self.assertEquals(user.credit, 85, 'Error in charging process (%d!=85, stopped)' % ( user.credit, )) |
153 | 153 |
|
154 | 154 |
# try charge until the user spends all his credits, see if the charger |
... | ... | |
164 | 164 |
vm.save() |
165 | 165 |
charger.charge() |
166 | 166 |
|
167 |
user = OceanUser.objects.get(pk=1)
|
|
167 |
user = SynnefoUser.objects.get(pk=1)
|
|
168 | 168 |
self.assertEquals(user.credit, 0, 'Error in charging process (%d!=0, running)' % ( user.credit, )) |
169 | 169 |
|
170 | 170 |
|
... | ... | |
176 | 176 |
djuser.save() |
177 | 177 |
|
178 | 178 |
# Add a user |
179 |
user = OceanUser(pk=1, name='Test User', credit=100, quota=100, monthly_rate=10)
|
|
179 |
user = SynnefoUser(pk=1, name='Test User', credit=100, quota=100, monthly_rate=10)
|
|
180 | 180 |
user.created = datetime.datetime.now() |
181 | 181 |
user.user = djuser |
182 | 182 |
user.save() |
... | ... | |
214 | 214 |
def test_virtual_machine(self): |
215 | 215 |
"""VirtualMachine (model) unit test""" |
216 | 216 |
|
217 |
|
|
217 |
|
Also available in: Unified diff