From 5265ce80cc8956ce708d05a4ca1cc7c9064bbef1 Mon Sep 17 00:00:00 2001 From: zandercymatics <141044360+zandercymatics@users.noreply.github.com> Date: Wed, 12 Mar 2025 13:04:36 -0600 Subject: [PATCH] Update test_models_domain.py --- src/registrar/tests/test_models_domain.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/registrar/tests/test_models_domain.py b/src/registrar/tests/test_models_domain.py index c5fabb6bc..e10b16843 100644 --- a/src/registrar/tests/test_models_domain.py +++ b/src/registrar/tests/test_models_domain.py @@ -990,18 +990,18 @@ class TestRegistrantContacts(MockEppLib): actual_contact = contact[1] if expected_contact.contact_type == PublicContact.ContactTypeChoices.SECURITY: expectedCreateCommand = self._convertPublicContactToEpp( - expected_contact, disclose_email=True, disclose_fields=["email"] + expected_contact, disclose_email=True, disclose_fields={"email"} ) elif expected_contact.contact_type == PublicContact.ContactTypeChoices.ADMINISTRATIVE: expectedCreateCommand = self._convertPublicContactToEpp( expected_contact, disclose_email=True, - disclose_fields=["email", "voice", "addr"], + disclose_fields={"email", "voice", "addr"}, disclose_types={"addr": "loc"}, ) else: expectedCreateCommand = self._convertPublicContactToEpp( - expected_contact, disclose_email=False, disclose_fields=[] + expected_contact, disclose_email=False, disclose_fields={} ) self.mockedSendFunction.assert_any_call(expectedCreateCommand, cleaned=True) # The emails should match on both items @@ -1103,7 +1103,7 @@ class TestRegistrantContacts(MockEppLib): # Verify disclosure settings self.assertEqual(result.disclose.flag, True) - self.assertEqual(result.disclose.fields, []) + self.assertEqual(result.disclose.fields, {}) self.assertIsNone(result.disclose.types) def test_not_disclosed_on_default_security_contact(self):