def test_base_address_build(self): """ Test that commit inherited fine """ address = model_factory(Address) form = BaseAddressForm() with patch('django.forms.ModelForm.save', Mock(return_value=address)): address = form.save(commit=False) self.assertTrue(address.pk is None)
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)