From bce590584204b681453fc19d737f1a5d1159a413 Mon Sep 17 00:00:00 2001 From: PCoder Date: Mon, 22 Apr 2019 18:00:04 +0200 Subject: [PATCH] Create a default VMPricing --- datacenterlight/tests.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) 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,