Improve create_vm_task test
This commit is contained in:
parent
a3a8227007
commit
80c3ac5346
|
@ -12,9 +12,11 @@ from unittest import skipIf
|
||||||
|
|
||||||
from datacenterlight.models import VMTemplate
|
from datacenterlight.models import VMTemplate
|
||||||
from datacenterlight.tasks import create_vm_task
|
from datacenterlight.tasks import create_vm_task
|
||||||
|
from hosting.models import HostingOrder
|
||||||
from membership.models import StripeCustomer
|
from membership.models import StripeCustomer
|
||||||
from opennebula_api.serializers import VMTemplateSerializer
|
from opennebula_api.serializers import VMTemplateSerializer
|
||||||
from utils.hosting_utils import get_vm_price
|
from utils.hosting_utils import get_vm_price
|
||||||
|
from utils.models import BillingAddress
|
||||||
from utils.stripe_utils import StripeUtils
|
from utils.stripe_utils import StripeUtils
|
||||||
|
|
||||||
|
|
||||||
|
@ -81,11 +83,14 @@ class CeleryTaskTestCase(TestCase):
|
||||||
|
|
||||||
stripe_customer = StripeCustomer.get_or_create(
|
stripe_customer = StripeCustomer.get_or_create(
|
||||||
email=self.customer_email,
|
email=self.customer_email,
|
||||||
token=self.token)
|
token=self.token
|
||||||
|
)
|
||||||
card_details = self.stripe_utils.get_card_details(
|
card_details = self.stripe_utils.get_card_details(
|
||||||
stripe_customer.stripe_id,
|
stripe_customer.stripe_id,
|
||||||
self.token)
|
self.token
|
||||||
card_details_dict = card_details.get('response_object')
|
)
|
||||||
|
card_details_dict = card_details.get('error')
|
||||||
|
self.assertEquals(card_details_dict, None)
|
||||||
billing_address_data = {'cardholder_name': self.customer_name,
|
billing_address_data = {'cardholder_name': self.customer_name,
|
||||||
'postal_code': '1231',
|
'postal_code': '1231',
|
||||||
'country': 'CH',
|
'country': 'CH',
|
||||||
|
@ -122,10 +127,19 @@ class CeleryTaskTestCase(TestCase):
|
||||||
msg = subscription_result.get('error')
|
msg = subscription_result.get('error')
|
||||||
raise Exception("Creating subscription failed: {}".format(msg))
|
raise Exception("Creating subscription failed: {}".format(msg))
|
||||||
|
|
||||||
|
billing_address = BillingAddress(
|
||||||
|
cardholder_name=billing_address_data['cardholder_name'],
|
||||||
|
street_address=billing_address_data['street_address'],
|
||||||
|
city=billing_address_data['city'],
|
||||||
|
postal_code=billing_address_data['postal_code'],
|
||||||
|
country=billing_address_data['country']
|
||||||
|
)
|
||||||
|
billing_address.save()
|
||||||
|
|
||||||
order = HostingOrder.create(
|
order = HostingOrder.create(
|
||||||
price=specs['price'],
|
price=specs['price'],
|
||||||
vm_id=0,
|
vm_id=0,
|
||||||
customer=customer,
|
customer=stripe_customer,
|
||||||
billing_address=billing_address
|
billing_address=billing_address
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue