mirror of
https://github.com/cisagov/manage.get.gov.git
synced 2025-05-18 02:19:23 +02:00
Fix remaining tests (hopefully)
This commit is contained in:
parent
9f6d1324d9
commit
056df7151d
9 changed files with 28 additions and 41 deletions
|
@ -307,7 +307,6 @@ class DomainRequestAdminForm(forms.ModelForm):
|
||||||
|
|
||||||
return cleaned_data
|
return cleaned_data
|
||||||
|
|
||||||
|
|
||||||
def _check_for_valid_rejection_reason(self, rejection_reason) -> bool:
|
def _check_for_valid_rejection_reason(self, rejection_reason) -> bool:
|
||||||
"""
|
"""
|
||||||
Checks if the rejection_reason field is not none.
|
Checks if the rejection_reason field is not none.
|
||||||
|
@ -3197,6 +3196,7 @@ class SuborganizationAdmin(ListHeaderAdmin, ImportExportModelAdmin):
|
||||||
extra_context = {"domain_requests": domain_requests, "domains": domains}
|
extra_context = {"domain_requests": domain_requests, "domains": domains}
|
||||||
return super().change_view(request, object_id, form_url, extra_context)
|
return super().change_view(request, object_id, form_url, extra_context)
|
||||||
|
|
||||||
|
|
||||||
class AllowedEmailAdmin(ListHeaderAdmin):
|
class AllowedEmailAdmin(ListHeaderAdmin):
|
||||||
class Meta:
|
class Meta:
|
||||||
model = models.AllowedEmail
|
model = models.AllowedEmail
|
||||||
|
@ -3206,6 +3206,7 @@ class AllowedEmailAdmin(ListHeaderAdmin):
|
||||||
search_help_text = "Search by email."
|
search_help_text = "Search by email."
|
||||||
ordering = ["email"]
|
ordering = ["email"]
|
||||||
|
|
||||||
|
|
||||||
admin.site.unregister(LogEntry) # Unregister the default registration
|
admin.site.unregister(LogEntry) # Unregister the default registration
|
||||||
|
|
||||||
admin.site.register(LogEntry, CustomLogEntryAdmin)
|
admin.site.register(LogEntry, CustomLogEntryAdmin)
|
||||||
|
|
|
@ -245,9 +245,7 @@ class UserFixture:
|
||||||
|
|
||||||
# Additional emails to add to the AllowedEmail whitelist.
|
# Additional emails to add to the AllowedEmail whitelist.
|
||||||
# The format should be as follows: ["email@igorville.gov", "email2@igorville.gov"]
|
# The format should be as follows: ["email@igorville.gov", "email2@igorville.gov"]
|
||||||
ADDITIONAL_ALLOWED_EMAILS = [
|
ADDITIONAL_ALLOWED_EMAILS = ["zander.adkinson@ecstech.com"]
|
||||||
"zander.adkinson@ecstech.com"
|
|
||||||
]
|
|
||||||
|
|
||||||
def load_users(cls, users, group_name, are_superusers=False):
|
def load_users(cls, users, group_name, are_superusers=False):
|
||||||
logger.info(f"Going to load {len(users)} users in group {group_name}")
|
logger.info(f"Going to load {len(users)} users in group {group_name}")
|
||||||
|
|
|
@ -35,21 +35,17 @@ class AllowedEmail(TimeStampedModel):
|
||||||
# Check if there's a '+' in the local part
|
# Check if there's a '+' in the local part
|
||||||
if "+" in local:
|
if "+" in local:
|
||||||
base_local = local.split("+")[0]
|
base_local = local.split("+")[0]
|
||||||
base_email_exists = cls.objects.filter(
|
base_email_exists = cls.objects.filter(Q(email__iexact=f"{base_local}@{domain}")).exists()
|
||||||
Q(email__iexact=f"{base_local}@{domain}")
|
|
||||||
).exists()
|
|
||||||
|
|
||||||
# Given an example email, such as "joe.smoe+1@igorville.com"
|
# Given an example email, such as "joe.smoe+1@igorville.com"
|
||||||
# The full regex statement will be: "^joe.smoe\\+\\d+@igorville.com$"
|
# The full regex statement will be: "^joe.smoe\\+\\d+@igorville.com$"
|
||||||
pattern = f'^{re.escape(base_local)}\\+\\d+@{re.escape(domain)}$'
|
pattern = f"^{re.escape(base_local)}\\+\\d+@{re.escape(domain)}$"
|
||||||
return base_email_exists and re.match(pattern, email)
|
return base_email_exists and re.match(pattern, email)
|
||||||
else:
|
else:
|
||||||
# Edge case, the +1 record exists but the base does not,
|
# Edge case, the +1 record exists but the base does not,
|
||||||
# and the record we are checking is the base record.
|
# and the record we are checking is the base record.
|
||||||
pattern = f'^{re.escape(local)}\\+\\d+@{re.escape(domain)}$'
|
pattern = f"^{re.escape(local)}\\+\\d+@{re.escape(domain)}$"
|
||||||
plus_email_exists = cls.objects.filter(
|
plus_email_exists = cls.objects.filter(Q(email__iregex=pattern)).exists()
|
||||||
Q(email__iregex=pattern)
|
|
||||||
).exists()
|
|
||||||
return plus_email_exists
|
return plus_email_exists
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
|
|
|
@ -580,14 +580,8 @@ class DomainRequest(TimeStampedModel):
|
||||||
@classmethod
|
@classmethod
|
||||||
def get_statuses_that_send_emails(cls):
|
def get_statuses_that_send_emails(cls):
|
||||||
"""Returns a list of statuses that send an email to the user"""
|
"""Returns a list of statuses that send an email to the user"""
|
||||||
excluded_statuses = [
|
excluded_statuses = [cls.DomainRequestStatus.INELIGIBLE, cls.DomainRequestStatus.IN_REVIEW]
|
||||||
cls.DomainRequestStatus.INELIGIBLE,
|
return [status for status in cls.DomainRequestStatus if status not in excluded_statuses]
|
||||||
cls.DomainRequestStatus.IN_REVIEW
|
|
||||||
]
|
|
||||||
return [
|
|
||||||
status for status in cls.DomainRequestStatus
|
|
||||||
if status not in excluded_statuses
|
|
||||||
]
|
|
||||||
|
|
||||||
def sync_organization_type(self):
|
def sync_organization_type(self):
|
||||||
"""
|
"""
|
||||||
|
|
|
@ -56,6 +56,7 @@ class TestDomainRequestAdmin(MockEppLib):
|
||||||
@classmethod
|
@classmethod
|
||||||
def tearDownClass(cls):
|
def tearDownClass(cls):
|
||||||
super().tearDownClass()
|
super().tearDownClass()
|
||||||
|
AllowedEmail.objects.all.delete()
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def setUpClass(self):
|
def setUpClass(self):
|
||||||
|
@ -74,6 +75,8 @@ class TestDomainRequestAdmin(MockEppLib):
|
||||||
model=DomainRequest,
|
model=DomainRequest,
|
||||||
)
|
)
|
||||||
self.mock_client = MockSESClient()
|
self.mock_client = MockSESClient()
|
||||||
|
allowed_emails = [AllowedEmail(email="mayor@igorville.gov"), AllowedEmail(email="help@get.gov")]
|
||||||
|
AllowedEmail.objects.bulk_create(allowed_emails)
|
||||||
|
|
||||||
def tearDown(self):
|
def tearDown(self):
|
||||||
super().tearDown()
|
super().tearDown()
|
||||||
|
@ -604,8 +607,8 @@ class TestDomainRequestAdmin(MockEppLib):
|
||||||
):
|
):
|
||||||
"""Helper method for the email test cases.
|
"""Helper method for the email test cases.
|
||||||
email_index is the index of the email in mock_client."""
|
email_index is the index of the email in mock_client."""
|
||||||
allowed_email, _ = AllowedEmail.objects.get_or_create(email=email_address)
|
AllowedEmail.objects.get_or_create(email=email_address)
|
||||||
allowed_bcc_email, _ = AllowedEmail.objects.get_or_create(email=bcc_email_address)
|
AllowedEmail.objects.get_or_create(email=bcc_email_address)
|
||||||
with less_console_noise():
|
with less_console_noise():
|
||||||
# Access the arguments passed to send_email
|
# Access the arguments passed to send_email
|
||||||
call_args = self.mock_client.EMAILS_SENT
|
call_args = self.mock_client.EMAILS_SENT
|
||||||
|
@ -633,9 +636,6 @@ class TestDomainRequestAdmin(MockEppLib):
|
||||||
bcc_email = kwargs["Destination"]["BccAddresses"][0]
|
bcc_email = kwargs["Destination"]["BccAddresses"][0]
|
||||||
self.assertEqual(bcc_email, bcc_email_address)
|
self.assertEqual(bcc_email, bcc_email_address)
|
||||||
|
|
||||||
allowed_email.delete()
|
|
||||||
allowed_bcc_email.delete()
|
|
||||||
|
|
||||||
@override_settings(IS_PRODUCTION=True)
|
@override_settings(IS_PRODUCTION=True)
|
||||||
@less_console_noise_decorator
|
@less_console_noise_decorator
|
||||||
def test_action_needed_sends_reason_email_prod_bcc(self):
|
def test_action_needed_sends_reason_email_prod_bcc(self):
|
||||||
|
|
|
@ -40,8 +40,6 @@ def send_templated_email(
|
||||||
Raises EmailSendingError if SES client could not be accessed
|
Raises EmailSendingError if SES client could not be accessed
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if not settings.IS_PRODUCTION: # type: ignore
|
if not settings.IS_PRODUCTION: # type: ignore
|
||||||
if flag_is_active(None, "disable_email_sending"): # type: ignore
|
if flag_is_active(None, "disable_email_sending"): # type: ignore
|
||||||
message = "Could not send email. Email sending is disabled due to flag 'disable_email_sending'."
|
message = "Could not send email. Email sending is disabled due to flag 'disable_email_sending'."
|
||||||
|
@ -49,7 +47,7 @@ def send_templated_email(
|
||||||
else:
|
else:
|
||||||
# Raise an email sending error if these doesn't exist within our whitelist.
|
# Raise an email sending error if these doesn't exist within our whitelist.
|
||||||
# If these emails don't exist, this function can handle that elsewhere.
|
# If these emails don't exist, this function can handle that elsewhere.
|
||||||
AllowedEmail = apps.get_model('registrar', 'AllowedEmail')
|
AllowedEmail = apps.get_model("registrar", "AllowedEmail")
|
||||||
message = "Could not send email. The email '{}' does not exist within the whitelist."
|
message = "Could not send email. The email '{}' does not exist within the whitelist."
|
||||||
if to_address and not AllowedEmail.is_allowed_email(to_address):
|
if to_address and not AllowedEmail.is_allowed_email(to_address):
|
||||||
raise EmailSendingError(message.format(to_address))
|
raise EmailSendingError(message.format(to_address))
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue