def test_company_is_verified(code, preverfiied, oauth2, expected):
company = CompanyFactory.build(
verified_with_preverified_enrolment=preverfiied,
verified_with_code=code,
verified_with_companies_house_oauth2=oauth2,
)
assert company.is_verified is expected
def test_can_publish(desciption, summary, email, is_verified, expected,
settings):
mock_verifed = mock.PropertyMock(return_value=is_verified)
with mock.patch('company.models.Company.is_verified', mock_verifed):
company = CompanyFactory.build(
description=desciption,
summary=summary,
email_address=email,
)
assert company.is_publishable is expected
def mask_company_data(self):
queryset = models.Company.objects.all()
failed = 0
succeded = 0
for company in queryset:
try:
message = f'Company {company.pk} updated'
company_factory = CompanyFactory.build()
company.mobile_number = company_factory.mobile_number
company.postal_full_name = company_factory.postal_full_name
company.address_line_1 = company_factory.address_line_1
company.address_line_2 = company_factory.address_line_2
company.postal_code = company_factory.postal_code
company.po_box = company_factory.po_box
company.email_address = company_factory.email_address
company.email_full_name = company_factory.email_full_name
company.save()
self.stdout.write(self.style.SUCCESS(message))
succeded += 1
except Exception as e:
self.stdout.write(self.style.ERROR(e))
failed += 1
self.stdout.write(self.style.SUCCESS(f'{succeded} companies updated'))
self.stdout.write(self.style.WARNING(f'{failed} companies failed'))
