Add calculate price function to template

This commit is contained in:
Modulos 2017-05-10 02:36:11 +02:00
commit c1a3f7281f
2 changed files with 19 additions and 0 deletions

View file

@ -63,6 +63,13 @@ class VirtualMachineTemplateTestCase(TestCase):
new_count = VirtualMachineTemplate.objects.count()
self.assertNotEqual(old_count, new_count)
def test_model_can_calculate_price(self):
price = self.cores * self.core_price
price += self.memory * self.memory_price
price += self.disk_size * self.disk_size_price
self.assertEqual(price, self.template.calculate_price())
class VirtualMachineTestCase(TestCase):
def setUp(self):