From 0e4ba770bf5a440b2f0c6c289bacd448c0a14580 Mon Sep 17 00:00:00 2001 From: Neil Martinsen-Burrell Date: Thu, 9 Nov 2023 15:42:55 -0600 Subject: [PATCH] Add a test case for sending to multiple emails --- .../test_transition_domain_migrations.py | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/src/registrar/tests/test_transition_domain_migrations.py b/src/registrar/tests/test_transition_domain_migrations.py index c33f9ff57..27264c54e 100644 --- a/src/registrar/tests/test_transition_domain_migrations.py +++ b/src/registrar/tests/test_transition_domain_migrations.py @@ -281,3 +281,23 @@ class TestLogins(TestCase): # should only be one domain we send email for self.assertIn("Found 1 transition domains", output) self.assertTrue("would send email to testuser@gmail.com", output) + + def test_send_domain_invitations_two_emails(self): + """Can send only a single domain invitation email.""" + with less_console_noise(): + self.run_load_domains() + self.run_transfer_domains() + + # these are two email addresses in data/test_contacts.txt + output_stream = StringIO() + # 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", + "agustina.wyman7@test.com", stdout=output_stream) + + # Check that we had the right numbers in our output + output = output_stream.getvalue() + # should only be one domain we send email for + self.assertIn("Found 2 transition domains", output) + self.assertTrue("would send email to testuser@gmail.com", output) + self.assertTrue("would send email to agustina.wyman7@test.com", output)