From 13cd098c60f159efd206db6a39f282f3c4569945 Mon Sep 17 00:00:00 2001
From: PCoder <purple.coder@yahoo.co.uk>
Date: Mon, 22 Apr 2019 18:05:32 +0200
Subject: [PATCH] Set VMPricing when creating HostingOrder

---
 datacenterlight/tests.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/datacenterlight/tests.py b/datacenterlight/tests.py
index 9ec2df96..dcbb727f 100644
--- a/datacenterlight/tests.py
+++ b/datacenterlight/tests.py
@@ -101,7 +101,7 @@ class CeleryTaskTestCase(TestCase):
         cpu = specs.get('cpu')
         memory = specs.get('memory')
         disk_size = specs.get('disk_size')
-        VMPricing.objects.create(
+        vm_pricing = VMPricing.objects.create(
             name="default",
             cores_unit_price=5,
             ram_unit_price=2,
@@ -146,7 +146,8 @@ class CeleryTaskTestCase(TestCase):
             price=specs['price'],
             vm_id=0,
             customer=stripe_customer,
-            billing_address=billing_address
+            billing_address=billing_address,
+            vm_pricing=vm_pricing
         )
 
         async_task = create_vm_task.delay(