def test_base_address_save(self): """ Test that form save address with shipping=True and address is saved """ wizard = PrePaymentWizard() wizard.contact = model_factory(Contact, save=True) address = model_factory(Address) form = BaseAddressForm() with patch('django.forms.ModelForm.save', Mock(return_value=address)): address = form.save(wizard=wizard) self.assertTrue(address.pk is not None) self.assertEqual(address.contact, wizard.contact)
def test_billing_address_save(self): """ Test that form save address with shipping=True""" address = model_factory(Address) form = BillingAddressForm() base_save = Mock(return_value=address) commit_save = Mock(return_value=address) wizard = PrePaymentWizard() wizard.contact = model_factory(Contact, save=True) with nested(patch( 'salest.accounts.forms.BaseAddressForm.save', base_save), patch('salest.accounts.forms.InstanceSaveCommitMixin.save', commit_save)): address = form.save(wizard=wizard) base_save.assert_called_once_with(form, commit=False, wizard=wizard) commit_save.assert_called_once_with(form, commit=True) self.assertTrue(address.is_billing)