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