def test_company_form_null_company_number():
form = forms.CompanyForm(data={
'company_number': None,
})
form.is_valid()
assert form.cleaned_data['company_number'] is None
def test_company_form_empty_string_company_number():
form = forms.CompanyForm(data={
'company_number': '',
})
form.is_valid()
assert form.cleaned_data['company_number'] is None
def test_company_form_company_number_and_sole_trader_throws_error():
form = forms.CompanyForm(data={
'company_number': '123124',
'sole_trader': True,
})
assert form.is_valid() is False
assert form.errors['sole_trader'] == [form.MESSAGE_MUTUALLY_EXCLUSIVE]
def test_company_form_pad_company_number():
form = forms.CompanyForm(data={
'company_number': '1231245',
})
form.is_valid()
assert form.cleaned_data['company_number'] == '01231245'
def test_company_form_company_number_without_sole_trader_acceted(data):
form = forms.CompanyForm(data=data)
assert form.is_valid() is True
def test_company_form_optional_fields():
form = forms.CompanyForm(data={})
assert form.fields['company_name'].required is False
assert form.fields['company_number'].required is False