Revision 3846dfd0 db/tests.py
b/db/tests.py | ||
---|---|---|
174 | 174 |
self.assertEquals(user.credit, 0, 'Error in charging process (%d!=0, running)' % ( user.credit, )) |
175 | 175 |
|
176 | 176 |
|
177 |
class VirtualMachineTestCase(unittest.TestCase): |
|
178 |
def setUp(self): |
|
179 |
"""Setup the test""" |
|
180 |
userdj = User.objects.create(username='testuser') |
|
181 |
userdj.save() |
|
182 |
|
|
183 |
# Add a user |
|
184 |
user = SynnefoUser(pk=1, name='Test User', credit=100, quota=100, monthly_rate=10) |
|
185 |
user.created = datetime.datetime.now() |
|
186 |
user.user = User.objects.get(username='testuser') |
|
187 |
user.save() |
|
188 |
|
|
189 |
# add a Flavor |
|
190 |
flavor = Flavor(pk=1, cpu=10, ram=10, disk=10) |
|
191 |
flavor.save() |
|
192 |
|
|
193 |
# Now, add a VM |
|
194 |
vm = VirtualMachine(pk=1) |
|
195 |
vm.created = datetime.datetime.now() |
|
196 |
vm.charged = datetime.datetime.now() |
|
197 |
vm.imageid = 1 |
|
198 |
vm.hostid = 'testhostid' |
|
199 |
vm.server_label = 'agreatserver' |
|
200 |
vm.image_version = '1.0.0' |
|
201 |
vm.ipfour = '127.0.0.1' |
|
202 |
vm.ipsix = '2001:0db8:85a3:0000:0000:8a2e:0370:7334' |
|
203 |
vm.owner = user |
|
204 |
vm.flavor = flavor |
|
205 |
|
|
206 |
vm.save() |
|
207 |
|
|
208 |
|
|
209 |
def tearDown(self): |
|
210 |
"""Cleaning up the data""" |
|
211 |
user = User.objects.get(username="testname") |
|
212 |
user.delete() |
|
213 |
|
|
214 |
flavor = Flavor.objects.get(pk=1) |
|
215 |
flavor.delete() |
|
216 |
|
|
217 |
|
|
218 |
def test_virtual_machine(self): |
|
219 |
"""VirtualMachine (model) unit test""" |
|
220 |
|
|
221 |
|
Also available in: Unified diff