def test_company_contact_name_ext(): company = CompanyContact(name="TestCompany") assert company.name == "TestCompany" assert company.full_name == "TestCompany" company.name_ext = "California" company.save() assert company.name == "TestCompany" assert company.full_name == "TestCompany / California"
def save(self, commit=True): def populate_address(needle): data = {} delete = [] for k, value in six.iteritems(self.cleaned_data): if k.startswith(needle): key = k.replace(needle, "") data[key] = value delete.append(k) # sweep unneeded keys for k in delete: del self.cleaned_data[k] return data contact_address_data = populate_address("contact_") company_address_data = populate_address("company_") user = super(CompanyRegistrationForm, self).save(commit) website = company_address_data.pop("www") contact_address = MutableAddress.from_data(contact_address_data) contact_address.save() company_address = MutableAddress.from_data(company_address_data) company_address.save() contact = PersonContact() contact.is_active = False contact.user = user contact.email = user.email contact.default_shipping_address = contact_address contact.default_billing_address = contact_address contact.first_name = contact_address_data["first_name"] contact.last_name = contact_address_data["last_name"] contact.phone = contact_address.phone contact.save() company = CompanyContact() company.default_shipping_address = company_address company.default_billing_address = company_address company.is_active = False company.phone = company_address.phone company.www = website company.name = company_address_data["name"] company.name_ext = company_address_data["name_ext"] company.tax_number = company_address_data["tax_number"] company.email = company_address_data["email"] company.save() company.members.add(contact) return user