fix testing error

This commit is contained in:
Arvind Tiwari 2018-05-07 09:14:31 +05:30
parent 0fdb88b8aa
commit 55cbe3244a

View file

@ -654,7 +654,7 @@ class PaymentVMView(LoginRequiredMixin, FormView):
context.update({
'stripe_key': settings.STRIPE_API_PUBLIC_KEY,
'vm_pricing': VMPricing.get_vm_pricing_by_name(
self.request.session['specs']['pricing_name']
self.request.session.get('specs', {}).get('pricing_name')
),
})
@ -1010,7 +1010,6 @@ class CreateVirtualMachinesView(LoginRequiredMixin, View):
@method_decorator(decorators)
def get(self, request, *args, **kwargs):
print(get_cms_integration('default'))
context = {
'templates': VMTemplate.objects.all(),
'cms_integration': get_cms_integration('default'),