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)
