root / db / tests.py @ 9cedbaf2
History | View | Annotate | Download (4.7 kB)
1 | 7bd50624 | Vassilios Karakoidas | #
|
---|---|---|---|
2 | 7bd50624 | Vassilios Karakoidas | # Unit Tests for db
|
3 | 7bd50624 | Vassilios Karakoidas | #
|
4 | 7bd50624 | Vassilios Karakoidas | # Provides automated tests for db module
|
5 | 7bd50624 | Vassilios Karakoidas | #
|
6 | d8e682ed | Vassilios Karakoidas | # Copyright 2010 Greek Research and Technology Network
|
7 | d8e682ed | Vassilios Karakoidas | #
|
8 | d8e682ed | Vassilios Karakoidas | |
9 | 9cedbaf2 | Georgios Gousios | from synnefo.db.models import * |
10 | 04a6d900 | Vassilios Karakoidas | |
11 | d39ea372 | Vassilios Karakoidas | from django.test import TestCase |
12 | d8e682ed | Vassilios Karakoidas | |
13 | e8fa4283 | Vassilios Karakoidas | |
14 | 8bda0440 | Vassilios Karakoidas | class FlavorTestCase(TestCase): |
15 | e48404ca | Vassilios Karakoidas | fixtures = [ 'db_test_data' ]
|
16 | 452d2391 | Vassilios Karakoidas | |
17 | 8bda0440 | Vassilios Karakoidas | def test_flavor(self): |
18 | 8bda0440 | Vassilios Karakoidas | """Test a flavor object, its internal cost calculation and naming methods"""
|
19 | 966be609 | Vassilios Karakoidas | flavor = Flavor.objects.get(pk=30000)
|
20 | 8bda0440 | Vassilios Karakoidas | |
21 | 452d2391 | Vassilios Karakoidas | # test current active/inactive costs
|
22 | 452d2391 | Vassilios Karakoidas | c_active = flavor.current_cost_active |
23 | 452d2391 | Vassilios Karakoidas | c_inactive = flavor.current_cost_inactive |
24 | 8bda0440 | Vassilios Karakoidas | |
25 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(c_active, 10, 'flavor.cost_active should be 10! (%d)' % (c_active,)) |
26 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(c_inactive, 5, 'flavor.cost_inactive should be 5! (%d)' % (c_inactive,)) |
27 | 8bda0440 | Vassilios Karakoidas | |
28 | 452d2391 | Vassilios Karakoidas | # test name property, should be C1R1024D10
|
29 | 452d2391 | Vassilios Karakoidas | f_name = flavor.name |
30 | 8bda0440 | Vassilios Karakoidas | |
31 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(f_name, 'C1R1024D10', 'flavor.name is not generated correctly, C1R1024D10! (%s)' % (f_name,)) |
32 | 6d6b83a3 | Vassilios Karakoidas | |
33 | 452d2391 | Vassilios Karakoidas | def test_flavor_get_costs(self): |
34 | a0f83bd2 | Vassilios Karakoidas | """Test the Flavor _get_costs() method"""
|
35 | a0f83bd2 | Vassilios Karakoidas | # first an easy test, a Flavor with only one FlavorCost entry
|
36 | a0f83bd2 | Vassilios Karakoidas | flavor = Flavor.objects.get(pk=30001)
|
37 | a0f83bd2 | Vassilios Karakoidas | |
38 | a0f83bd2 | Vassilios Karakoidas | start_date = datetime.datetime(year=2010, month=1, day=1) |
39 | a0f83bd2 | Vassilios Karakoidas | end_date = datetime.datetime(year=2010, month=1, day=2) |
40 | a0f83bd2 | Vassilios Karakoidas | |
41 | a0f83bd2 | Vassilios Karakoidas | # we now that the cost should be 5*24 (inactive) and 10*24 (active)
|
42 | a0f83bd2 | Vassilios Karakoidas | r_active = flavor.get_cost_active(start_date, end_date) |
43 | a0f83bd2 | Vassilios Karakoidas | r_inactive = flavor.get_cost_inactive(start_date, end_date) |
44 | a0f83bd2 | Vassilios Karakoidas | |
45 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(len(r_active), 1, 'flavor.get_cost_active() should have returned 1 entry (%d)' %(len(r_active),)) |
46 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(len(r_inactive), 1, 'flavor.get_cost_inactive() should have returned 1 entry (%d)'% (len(r_inactive),)) |
47 | a0f83bd2 | Vassilios Karakoidas | |
48 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(10*24, r_active[0][1], 'flavor.get_cost_active() is not working properly (%d!=%d)' % (r_active[0][1], 10*24)) |
49 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(5*24, r_inactive[0][1], 'flavor.get_cost_inactive() is not working properly (%d!=%d)' % (r_inactive[0][1], 5*24)) |
50 | a0f83bd2 | Vassilios Karakoidas | |
51 | a0f83bd2 | Vassilios Karakoidas | # The second test, will involve a more complex cost example
|
52 | a0f83bd2 | Vassilios Karakoidas | # The overall cost will be calculated by two FlavorCost entries
|
53 | a0f83bd2 | Vassilios Karakoidas | |
54 | 452d2391 | Vassilios Karakoidas | flavor = Flavor.objects.get(pk=30000)
|
55 | a0f83bd2 | Vassilios Karakoidas | |
56 | a0f83bd2 | Vassilios Karakoidas | start_date = datetime.datetime(year=2010, month=12, day=31) |
57 | a0f83bd2 | Vassilios Karakoidas | end_date = datetime.datetime(year=2011, month=01, day=2) |
58 | a0f83bd2 | Vassilios Karakoidas | |
59 | a0f83bd2 | Vassilios Karakoidas | # this is more complicated, active costs are 5*24 + 10*24 = 360
|
60 | a0f83bd2 | Vassilios Karakoidas | # and inactive costs are 2*24 + 5*24 = 168
|
61 | b86e97ed | Vassilios Karakoidas | |
62 | a0f83bd2 | Vassilios Karakoidas | r_active = flavor.get_cost_active(start_date, end_date) |
63 | a0f83bd2 | Vassilios Karakoidas | r_inactive = flavor.get_cost_inactive(start_date, end_date) |
64 | a0f83bd2 | Vassilios Karakoidas | |
65 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(len(r_active), 2, 'flavor.get_cost_active() should have returned 2 entries (%d)' %(len(r_active),)) |
66 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(len(r_inactive), 2, 'flavor.get_cost_inactive() should have returned 2 entries (%d)'% (len(r_inactive),)) |
67 | a0f83bd2 | Vassilios Karakoidas | |
68 | a0f83bd2 | Vassilios Karakoidas | ta_cost = sum([x[1] for x in r_active]) |
69 | a0f83bd2 | Vassilios Karakoidas | tia_cost = sum([x[1] for x in r_inactive]) |
70 | a0f83bd2 | Vassilios Karakoidas | |
71 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(360, ta_cost, 'flavor.get_cost_active() is not working properly (%d!=%d)' % (ta_cost, 360)) |
72 | a0f83bd2 | Vassilios Karakoidas | self.assertEqual(168, tia_cost, 'flavor.get_cost_inactive() is not working properly (%d!=%d)' % (tia_cost, 168)) |
73 | a0f83bd2 | Vassilios Karakoidas | |
74 | a0f83bd2 | Vassilios Karakoidas | |
75 | a0f83bd2 | Vassilios Karakoidas | class VirtualMachineTestCase(TestCase): |
76 | a0f83bd2 | Vassilios Karakoidas | fixtures = [ 'db_test_data' ]
|
77 | a0f83bd2 | Vassilios Karakoidas | |
78 | a0f83bd2 | Vassilios Karakoidas | def test_charge_method(self): |
79 | a0f83bd2 | Vassilios Karakoidas | """Test VirtualMachine.charge() method"""
|
80 | a0f83bd2 | Vassilios Karakoidas | |
81 | a0f83bd2 | Vassilios Karakoidas | # Since we have tested the costs, with this test
|
82 | a0f83bd2 | Vassilios Karakoidas | # we must ensure the following:
|
83 | a0f83bd2 | Vassilios Karakoidas | # 1. The vm.charged is updated
|
84 | a0f83bd2 | Vassilios Karakoidas | # 2. Users credits are decreased
|
85 | a0f83bd2 | Vassilios Karakoidas | |
86 | a0f83bd2 | Vassilios Karakoidas | vm_started = VirtualMachine.objects.get(pk=30000)
|
87 | a0f83bd2 | Vassilios Karakoidas | |
88 | a0f83bd2 | Vassilios Karakoidas | initial_date = vm_started.charged |
89 | a0f83bd2 | Vassilios Karakoidas | initial_credits = vm_started.owner.credit |
90 | a0f83bd2 | Vassilios Karakoidas | |
91 | a0f83bd2 | Vassilios Karakoidas | vm_started.charge() |
92 | a0f83bd2 | Vassilios Karakoidas | |
93 | a0f83bd2 | Vassilios Karakoidas | self.assertTrue(vm_started.charged > initial_date, 'Initial charged date should not be greater') |
94 | a0f83bd2 | Vassilios Karakoidas | self.assertTrue(initial_credits > vm_started.owner.credit, 'The user should have less credits now! (%d>%d)' % (initial_credits, vm_started.owner.credit)) |
95 | b86e97ed | Vassilios Karakoidas | |
96 | 9373e677 | Vassilios Karakoidas | |
97 | 452d2391 | Vassilios Karakoidas | class SynnefoUserTestCase(TestCase): |
98 | 62c6d260 | Vassilios Karakoidas | fixtures = [ 'db_test_data' ]
|
99 | 62c6d260 | Vassilios Karakoidas | |
100 | 452d2391 | Vassilios Karakoidas | def test_synnefo_user(self): |
101 | 452d2391 | Vassilios Karakoidas | """Test a SynnefoUser object"""
|
102 | 452d2391 | Vassilios Karakoidas | s_user = SynnefoUser.objects.get(pk=30000)
|
103 | 452d2391 | Vassilios Karakoidas | v_machine = VirtualMachine.objects.get(pk=30000)
|
104 | 8bda0440 | Vassilios Karakoidas | |
105 | 452d2391 | Vassilios Karakoidas | # charge the user
|
106 | 452d2391 | Vassilios Karakoidas | s_user.debit_account(10, v_machine, "This should be a structured debit message!") |
107 | 13b3c5ff | Vassilios Karakoidas | |
108 | 452d2391 | Vassilios Karakoidas | # should have only one debit object
|
109 | 452d2391 | Vassilios Karakoidas | d_list = Debit.objects.all() |
110 | 13b3c5ff | Vassilios Karakoidas | |
111 | 452d2391 | Vassilios Karakoidas | self.assertEqual(len(d_list), 1, 'SynnefoUser.debit_account() writes more than one debit entries!') |
112 | 452d2391 | Vassilios Karakoidas | |
113 | 452d2391 | Vassilios Karakoidas | # retrieve the user, now he/she should have zero credits
|
114 | 452d2391 | Vassilios Karakoidas | s_user = SynnefoUser.objects.get(pk=30000)
|
115 | 452d2391 | Vassilios Karakoidas | |
116 | 452d2391 | Vassilios Karakoidas | self.assertEqual(0, s_user.credit, 'SynnefoUser (pk=30000) should have zero credits (%d)' % ( s_user.credit, )) |