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

cleaned up tests

Browse source
This commit is contained in:
David Kennedy 2023-10-10 08:43:36 -04:00
parent 9e8aa2dee3
commit cb15b7d029
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
2 changed files with 29 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

25
src/registrar/tests/common.py
View file

@ -704,22 +704,19 @@ class MockEppLib(TestCase):
"alg": 1,
"pubKey": "AQPJ////4Q==",
}
dnssecExtensionWithDsData: Mapping[Any, Any] = {
"dsData": [common.DSData(**addDsData1)], # type: ignore
}
dnssecExtensionWithMultDsData: Mapping[str, Any] = {
dnssecExtensionWithDsData = extensions.DNSSECExtension(**{
"dsData": [common.DSData(**addDsData1)],
})
dnssecExtensionWithMultDsData = extensions.DNSSECExtension(**{
"dsData": [
common.DSData(**addDsData1), # type: ignore
common.DSData(**addDsData2), # type: ignore
],
}
dnssecExtensionWithKeyData: Mapping[str, Any] = {
})
dnssecExtensionWithKeyData = extensions.DNSSECExtension(**{
"keyData": [common.DNSSECKeyData(**keyDataDict)], # type: ignore
}
dnssecExtensionRemovingDsData: Mapping[Any, Any] = {
"dsData": None,
"keyData": None,
}
})
dnssecExtensionRemovingDsData = extensions.DNSSECExtension()
def mockSend(self, _request, cleaned):
"""Mocks the registry.send function used inside of domain.py
@ -765,7 +762,7 @@ class MockEppLib(TestCase):
return MagicMock(
res_data=[self.mockDataInfoDomain],
extensions=[
extensions.DNSSECExtension(**self.dnssecExtensionWithDsData)
self.dnssecExtensionWithDsData
],
)
elif getattr(_request, "name", None) == "dnssec-multdsdata.gov":
@ -775,7 +772,7 @@ class MockEppLib(TestCase):
return MagicMock(
res_data=[self.mockDataInfoDomain],
extensions=[
extensions.DNSSECExtension(**self.dnssecExtensionWithMultDsData)
self.dnssecExtensionWithMultDsData
],
)
elif getattr(_request, "name", None) == "dnssec-keydata.gov":
@ -785,7 +782,7 @@ class MockEppLib(TestCase):
return MagicMock(
res_data=[self.mockDataInfoDomain],
extensions=[
extensions.DNSSECExtension(**self.dnssecExtensionWithKeyData)
self.dnssecExtensionWithKeyData
],
)
elif getattr(_request, "name", None) == "dnssec-none.gov":