def test_companies_in_post_set_to_published(self):
companies = CompanyFactory.create_batch(5, is_published=False)
published_company = CompanyFactory(is_published=True)
numbers = '{num1},{num2}'.format(num1=companies[0].number,
num2=companies[3].number)
response = self.client.post(
reverse('admin:company_company_publish'),
{'company_numbers': numbers},
)
assert response.status_code == http.client.FOUND
assert response.url == reverse('admin:company_company_changelist')
published = Company.objects.filter(is_published=True).values_list(
'number', flat=True)
assert len(published) == 3
assert companies[0].number in published
assert companies[3].number in published
assert published_company.number in published
unpublished = Company.objects.filter(is_published=False).values_list(
'number', flat=True)
assert len(unpublished) == 3
assert companies[1].number in unpublished
assert companies[2].number in unpublished
assert companies[4].number in unpublished
def test_companies_publish_form_handles_whitespace():
companies = CompanyFactory.create_batch(3)
data = ' {num1},{num2} , {num3},'.format(num1=companies[0].number,
num2=companies[1].number,
num3=companies[2].number)
form = PublishByCompanyHouseNumberForm(data={'company_numbers': data})
assert form.is_valid() is True
def test_companies_publish_form_handles_whitespace():
companies = CompanyFactory.create_batch(3)
data = ' {num1},{num2} , {num3},'.format(num1=companies[0].number,
num2=companies[1].number,
num3=companies[2].number)
form = admin.PublishByCompanyHouseNumberForm(
data={
'company_numbers': data,
'directories': ['investment_support_directory']
})
assert form.is_valid() is True
def test_companies_in_post_set_to_published(self):
companies = CompanyFactory.create_batch(
7,
is_published_investment_support_directory=False,
is_published_find_a_supplier=False,
)
published_company_isd = CompanyFactory(
is_published_investment_support_directory=True)
numbers = '{num1},{num2}'.format(num1=companies[0].number,
num2=companies[3].number)
response = self.client.post(
reverse('admin:company_company_publish'),
{
'company_numbers': numbers,
'directories':
['investment_support_directory', 'find_a_supplier'],
},
)
assert response.status_code == http.client.FOUND
assert response.url == reverse('admin:company_company_changelist')
published_isd = Company.objects.filter(
is_published_investment_support_directory=True).values_list(
'number', flat=True)
assert len(published_isd) == 3
assert companies[0].number in published_isd
assert companies[3].number in published_isd
assert published_company_isd.number in published_isd
published_fas = Company.objects.filter(
is_published_find_a_supplier=True).values_list('number', flat=True)
assert len(published_fas) == 2
assert companies[0].number in published_fas
assert companies[3].number in published_fas
unpublished = Company.objects.filter(
is_published_investment_support_directory=False,
is_published_find_a_supplier=False,
).values_list('number', flat=True)
assert len(unpublished) == 5
assert companies[1].number in unpublished
assert companies[2].number in unpublished
assert companies[4].number in unpublished
def test_retrieve_missing_company_details(mock_get_date_of_creation):
mock_get_date_of_creation.return_value = None
company_with_date = CompanyFactory(date_of_creation=date(2010, 10, 1))
companies = CompanyFactory.create_batch(2)
mock_get_date_of_creation.return_value = date(2010, 10, 10)
call_command('retrieve_missing_company_details')
assert models.Company.objects.get(
number=company_with_date.number
).date_of_creation == date(2010, 10, 1)
assert models.Company.objects.get(
number=companies[0].number
).date_of_creation == date(2010, 10, 10)
assert models.Company.objects.get(
number=companies[1].number
).date_of_creation == date(2010, 10, 10)
def new_companies():
return CompanyFactory.create_batch(10)
