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'))