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