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