From 4506ee65f8c3425b0cc3a3202efb496a8d1bfccd Mon Sep 17 00:00:00 2001
From: zandercymatics <141044360+zandercymatics@users.noreply.github.com>
Date: Fri, 22 Dec 2023 09:02:26 -0700
Subject: [PATCH] Update test_transition_domain_migrations.py

---
 src/registrar/tests/test_transition_domain_migrations.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/src/registrar/tests/test_transition_domain_migrations.py b/src/registrar/tests/test_transition_domain_migrations.py
index 62df96efc..a757059dd 100644
--- a/src/registrar/tests/test_transition_domain_migrations.py
+++ b/src/registrar/tests/test_transition_domain_migrations.py
@@ -18,7 +18,7 @@ from unittest.mock import MagicMock, patch
 
 from registrar.models.contact import Contact
 
-from .common import MockEppLib, less_console_noise
+from .common import MockEppLib, MockSESClient, less_console_noise
 import boto3_mocking  # type: ignore
 
 
@@ -1030,8 +1030,7 @@ class TestMigrations(TestCase):
         # this is one of the email addresses in data/test_contacts.txt
         output_stream = StringIO()
 
-        mock_client = MagicMock()
-        with boto3_mocking.clients.handler_for("sesv2", mock_client):
+        with boto3_mocking.clients.handler_for("sesv2", MockSESClient):
             # also have to re-point the logging handlers to output_stream
             with less_console_noise(output_stream):
                 call_command("send_domain_invitations", "testuser@gmail.com", stdout=output_stream)
@@ -1052,8 +1051,7 @@ class TestMigrations(TestCase):
         # these are two email addresses in data/test_contacts.txt
         output_stream = StringIO()
 
-        mock_client = MagicMock()
-        with boto3_mocking.clients.handler_for("sesv2", mock_client):
+        with boto3_mocking.clients.handler_for("sesv2", MockSESClient):
             # also have to re-point the logging handlers to output_stream
             with less_console_noise(output_stream):
                 call_command(