zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-18 18:39:21 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update test_transition_domain_migrations.py

Browse source
This commit is contained in:
zandercymatics 2023-12-22 09:02:26 -07:00
parent 532e2f44d2
commit 4506ee65f8
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/registrar/tests/test_transition_domain_migrations.py
View file

@ -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(