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)