mravi/dynamicweb
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Fix case where None is returned by get_vm_price call

Browse source
This commit is contained in:
PCoder 2019-04-22 17:48:30 +02:00
commit 215047aee5
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
datacenterlight/tests.py
View file

@ -103,6 +103,10 @@ class CeleryTaskTestCase(TestCase):
disk_size = specs.get('disk_size') disk_size = specs.get('disk_size')
amount_to_be_charged = get_vm_price(cpu=cpu, memory=memory, amount_to_be_charged = get_vm_price(cpu=cpu, memory=memory,
disk_size=disk_size) 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, plan_name = StripeUtils.get_stripe_plan_name(cpu=cpu,
memory=memory, memory=memory,
disk_size=disk_size, disk_size=disk_size,