zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-08-22 09:11:04 +02:00
Code Issues Projects Releases Packages Wiki Activity

Update test_models_domain.py

Browse source
This commit is contained in:
zandercymatics 2025-03-12 13:04:36 -06:00
parent ca82f103d8
commit 5265ce80cc
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -990,18 +990,18 @@ class TestRegistrantContacts(MockEppLib):
actual_contact = contact[1] actual_contact = contact[1]
if expected_contact.contact_type == PublicContact.ContactTypeChoices.SECURITY: if expected_contact.contact_type == PublicContact.ContactTypeChoices.SECURITY:
expectedCreateCommand = self._convertPublicContactToEpp( 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: elif expected_contact.contact_type == PublicContact.ContactTypeChoices.ADMINISTRATIVE:
expectedCreateCommand = self._convertPublicContactToEpp( expectedCreateCommand = self._convertPublicContactToEpp(
expected_contact, expected_contact,
disclose_email=True, disclose_email=True,
disclose_fields=["email", "voice", "addr"], disclose_fields={"email", "voice", "addr"},
disclose_types={"addr": "loc"}, disclose_types={"addr": "loc"},
) )
else: else:
expectedCreateCommand = self._convertPublicContactToEpp( 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) self.mockedSendFunction.assert_any_call(expectedCreateCommand, cleaned=True)
# The emails should match on both items # The emails should match on both items
@ -1103,7 +1103,7 @@ class TestRegistrantContacts(MockEppLib):
# Verify disclosure settings # Verify disclosure settings
self.assertEqual(result.disclose.flag, True) self.assertEqual(result.disclose.flag, True)
self.assertEqual(result.disclose.fields, []) self.assertEqual(result.disclose.fields, {})
self.assertIsNone(result.disclose.types) self.assertIsNone(result.disclose.types)
def test_not_disclosed_on_default_security_contact(self): def test_not_disclosed_on_default_security_contact(self):