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

domain mock added with mock object

Browse source
This commit is contained in:
Alysia Broddrick 2023-06-13 08:24:13 -07:00
parent bb6c953b31
commit 9ca860b26a
No known key found for this signature in database
GPG key ID: 03917052CD0F06B7
2 changed files with 17 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/registrar/models/domain.py
View file

@ -478,7 +478,7 @@ class Domain(TimeStampedModel, DomainHelper):
} }
print("\nCACHE AT TOP\n\n"+str(cache)+"\n\n\n") print("\nCACHE AT TOP\n\n"+str(cache)+"\n\n\n")
# remove null properties (to distinguish between "a value of None" and null) # remove null properties (to distinguish between "a value of None" and null)
cleaned = {k: v for k, v in cache if v is not ...} cleaned = {k: v for k, v in cache.items() if v is not ...}
print("\ncleaned is "+str(cleaned)+"\n\n") print("\ncleaned is "+str(cleaned)+"\n\n")
# get contact info, if there are any # get contact info, if there are any
if ( if (

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

@ -13,15 +13,19 @@ try:
from epplib import InfoDomain from epplib import InfoDomain
except ImportError: except ImportError:
# allow epplibwrapper to load without epplib, for testing and development # allow epplibwrapper to load without epplib, for testing and development
print("epplib didn't load")
pass pass
##delete me ##delete me
from django.core.cache import cache from django.core.cache import cache
class TestDomain(TestCase): class TestDomain(TestCase):
mockDataInfoDomain={"avail": False, class fakedEppObject(object):
"auth_info": None, def __init__(self, avail, auth_info, cr_date):
"cr_date": datetime.datetime(2023, 5, 25, 19, 45, 35) self.avail=avail
} self.auth_info=auth_info
self.cr_date=cr_date
mockDataInfoDomain=fakedEppObject(False, None, datetime.datetime(2023, 5, 25, 19, 45, 35))
# {'auth_info': <MagicMock name='send().res_data.__getitem__().auth_info' id='281473717645712'>, '_contacts': <MagicMock name='send().res_data.__getitem__().contacts' id='281473717644608'>, 'cr_date': <MagicMock name='send().res_data.__getitem__().cr_date' id='281473719330096'>, 'ex_date': <MagicMock name='send().res_data.__getitem__().ex_date' id='281473719328464'>, '_hosts': <MagicMock name='send().res_data.__getitem__().hosts' id='281473721505856'>, 'name': <MagicMock name='send().res_data.__getitem__().name' id='281473717398512'>, 'registrant': <MagicMock name='send().res_data.__getitem__().registrant' id='281473717289408'>, 'statuses': <MagicMock name='send().res_data.__getitem__().statuses' id='281473717293632'>, 'tr_date': <MagicMock name='send().res_data.__getitem__().tr_date' id='281473710170096'>, 'up_date': <MagicMock name='send().res_data.__getitem__().up_date' id='281473710170384'>} # {'auth_info': <MagicMock name='send().res_data.__getitem__().auth_info' id='281473717645712'>, '_contacts': <MagicMock name='send().res_data.__getitem__().contacts' id='281473717644608'>, 'cr_date': <MagicMock name='send().res_data.__getitem__().cr_date' id='281473719330096'>, 'ex_date': <MagicMock name='send().res_data.__getitem__().ex_date' id='281473719328464'>, '_hosts': <MagicMock name='send().res_data.__getitem__().hosts' id='281473721505856'>, 'name': <MagicMock name='send().res_data.__getitem__().name' id='281473717398512'>, 'registrant': <MagicMock name='send().res_data.__getitem__().registrant' id='281473717289408'>, 'statuses': <MagicMock name='send().res_data.__getitem__().statuses' id='281473717293632'>, 'tr_date': <MagicMock name='send().res_data.__getitem__().tr_date' id='281473710170096'>, 'up_date': <MagicMock name='send().res_data.__getitem__().up_date' id='281473710170384'>}
# mockDataContactInfo={ # mockDataContactInfo={
@ -71,15 +75,21 @@ class TestDomain(TestCase):
# print(patch) # print(patch)
# domain, _= Domain.objects.get_or_create(name="igorville.gov") # domain, _= Domain.objects.get_or_create(name="igorville.gov")
# mockSend=MagicMock(return_val) # mockSend=MagicMock(return_val)
# InfoDomain()
with patch ("registrar.models.domain.registry.send", new=self.mock_send): with patch ("registrar.models.domain.registry.send", new=self.mock_send):
# with patch("epplibwrapper.CLIENT") as registry_mock, \ # with patch("epplibwrapper.CLIENT") as registry_mock, \
# patch("epplibwrapper.CLIENT.send",side_effects=self.mock_send) as send_mock: # patch("epplibwrapper.CLIENT.send",side_effects=self.mock_send) as send_mock:
# with patch("epplibwrapper.CLIENT.send",side_effects=self.mock_send): # with patch("epplibwrapper.CLIENT.send",side_effects=self.mock_send):
domain, _ = Domain.objects.get_or_create(name="igorville.gov") domain, _ = Domain.objects.get_or_create(name="igorville.gov")
domain._get_property("auth_info") domain._get_property("auth_info")
# domain._get_property("cr_date")
print(domain._cache) print(domain._cache)
self.assertEquals(domain._cache["auth_info"],self.mockDataInfoDomain.auth_info )
self.assertEquals(domain._cache["cr_date"],self.mockDataInfoDomain.cr_date )
self.assertFalse("avail" in domain._cache.keys())
#check it a second time and make sure send is not called
# sec=domain.security_contact # sec=domain.security_contact
# print(sec) # print(sec)
# print("domain cache is as follows\n") # print("domain cache is as follows\n")