def test_wizard_cart_checkout(self):
"""
Test that wizard methot done call cart checkout method
"""
request = RequestFactory().get('/')
user = model_factory(User, save=True)
cart = model_factory(Cart, contact=user.contact, is_active=True,
save=True)
model_factory(Order, cart=cart, save=True)
request.user = user
request.cart = cart
wizard = PrePaymentWizard()
wizard.request = request
with patch('salest.cart.models.Cart.checkout') as cart_checkout:
wizard.done([Mock(data={'payment_method': 'dummy'}),
Mock(data=[])])
cart_checkout.assert_called_once_with()
def test_wizard_form_save(self):
"""
Test that wizard mwthod done save forms
"""
request = RequestFactory().get('/')
user = model_factory(User, save=True)
cart = model_factory(Cart, contact=user.contact, is_active=True,
save=True)
model_factory(Order, cart=cart, save=True)
request.user = user
request.cart = cart
wizard = PrePaymentWizard()
wizard.request = request
form1 = Mock(data={'payment_method': 'dummy'})
form2 = Mock(data=[])
wizard.done([form1, form2])
self.assertTrue(form1.save.call_count == 1)
self.assertTrue(form2.save.call_count == 1)
