def test_validate_gafi(self): """ test the gafi validation """ # --- invalid street shareholder = ShareholderGenerator().generate() # must be in switzerland shareholder.company.country = Country.objects.get( iso_code__iexact='ch') shareholder.user.userprofile.street = '' shareholder.user.userprofile.save() self.assertFalse(shareholder.validate_gafi()['is_valid']) # --- invalid company name shareholder = ShareholderGenerator().generate() # must be in switzerland shareholder.company.country = Country.objects.get( iso_code__iexact='ch') shareholder.user.userprofile.company_name = None shareholder.user.userprofile.save() self.assertFalse(shareholder.validate_gafi()['is_valid']) # --- valid data shareholder = ShareholderGenerator().generate() # must be in switzerland shareholder.company.country = Country.objects.get( iso_code__iexact='ch') self.assertTrue(shareholder.validate_gafi()['is_valid'])
def test_validate_gafi_with_missing_userprofile(self): shareholder = ShareholderGenerator().generate() # must be in switzerland shareholder.company.country = Country.objects.get( iso_code__iexact='ch') profile = shareholder.user.userprofile profile.delete() shareholder = Shareholder.objects.get(id=shareholder.id) # must be in switzerland shareholder.company.country = Country.objects.get( iso_code__iexact='ch') self.assertFalse(hasattr(shareholder.user, 'userprofile')) self.assertFalse(shareholder.validate_gafi()['is_valid'])