From 3ffad88b5da6babf5aa0747601ee8510cc4f7f54 Mon Sep 17 00:00:00 2001 From: zandercymatics <141044360+zandercymatics@users.noreply.github.com> Date: Wed, 3 Jan 2024 13:10:51 -0700 Subject: [PATCH] Catch strangler --- src/registrar/tests/test_admin.py | 4 +++- src/registrar/tests/test_models_domain.py | 3 ++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/registrar/tests/test_admin.py b/src/registrar/tests/test_admin.py index 153c87b42..a35dbc7e3 100644 --- a/src/registrar/tests/test_admin.py +++ b/src/registrar/tests/test_admin.py @@ -59,7 +59,9 @@ class TestDomainAdmin(MockEppLib): """ self.client.force_login(self.superuser) application = completed_application(status=DomainApplication.ApplicationStatus.IN_REVIEW) - application.approve() + with boto3_mocking.clients.handler_for("sesv2", self.mock_client): + with less_console_noise(): + application.approve() response = self.client.get("/admin/registrar/domain/") diff --git a/src/registrar/tests/test_models_domain.py b/src/registrar/tests/test_models_domain.py index 9db7257b3..b86941183 100644 --- a/src/registrar/tests/test_models_domain.py +++ b/src/registrar/tests/test_models_domain.py @@ -265,7 +265,8 @@ class TestDomainCreation(MockEppLib): user, _ = User.objects.get_or_create() application = DomainApplication.objects.create(creator=user, requested_domain=draft_domain) - with boto3_mocking.clients.handler_for("sesv2", MockSESClient): + mock_client = MockSESClient() + with boto3_mocking.clients.handler_for("sesv2", mock_client): with less_console_noise(): # skip using the submit method application.status = DomainApplication.ApplicationStatus.SUBMITTED