diff --git a/datacenterlight/tests.py b/datacenterlight/tests.py index e9505beb..9ec2df96 100644 --- a/datacenterlight/tests.py +++ b/datacenterlight/tests.py @@ -10,7 +10,7 @@ from django.test import TestCase, override_settings from model_mommy import mommy from unittest import skipIf -from datacenterlight.models import VMTemplate +from datacenterlight.models import VMTemplate, VMPricing from datacenterlight.tasks import create_vm_task from hosting.models import HostingOrder from membership.models import StripeCustomer @@ -101,12 +101,14 @@ class CeleryTaskTestCase(TestCase): cpu = specs.get('cpu') memory = specs.get('memory') disk_size = specs.get('disk_size') + VMPricing.objects.create( + name="default", + cores_unit_price=5, + ram_unit_price=2, + ssd_unit_price=0.6 + ) amount_to_be_charged = get_vm_price(cpu=cpu, memory=memory, disk_size=disk_size) - # because we don't have a VMPricing by default - # get_vm_price should return None - self.assertEquals(amount_to_be_charged, None) - amount_to_be_charged = specs.get('price') plan_name = StripeUtils.get_stripe_plan_name(cpu=cpu, memory=memory, disk_size=disk_size,