zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-07-30 14:36:32 +02:00
Code Issues Projects Releases Packages Wiki Activity

added test cases for DNSSEC

Browse source
This commit is contained in:
David Kennedy 2023-09-28 22:53:15 -04:00
parent 6df2a65a90
commit b43901d675
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
2 changed files with 339 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -284,30 +284,21 @@ class Domain(TimeStampedModel, DomainHelper):
def dnssecdata(self) -> extensions.DNSSECExtension:
return self._get_property("dnssecdata")
@dnssecdata.setter
def dnssecdata(
self,
_dnssecdata: extensions.DNSSECExtension
):
@dnssecdata.setter # type: ignore
def dnssecdata(self, _dnssecdata: extensions.DNSSECExtension):
updateParams = {
"maxSigLife": _dnssecdata.maxSigLife,
"dsData": _dnssecdata.dsData,
"keyData": _dnssecdata.keyData,
"maxSigLife": _dnssecdata.get("maxSigLife", None),
"dsData": _dnssecdata.get("dsData", None),
"keyData": _dnssecdata.get("keyData", None),
"remAllDsKeyData": True,
}
request = commands.UpdateDomain(
name=self.name
)
request = commands.UpdateDomain(name=self.name)
extension = commands.UpdateDomainDNSSECExtension(**updateParams)
request.add_extension(extension)
try:
registry.send(request, cleaned=True)
except RegistryError as e:
logger.error(
"Error adding DNSSEC, code was %s error was %s"
% (e.code, e)
)
logger.error("Error adding DNSSEC, code was %s error was %s" % (e.code, e))
raise e
@nameservers.setter # type: ignore
@ -1010,10 +1001,10 @@ class Domain(TimeStampedModel, DomainHelper):
# get extensions info, if there is any
# DNSSECExtension is one possible extension, make sure to handle
# only DNSSECExtension and not other type extensions
extensions = dataResponse.extensions
returned_extensions = dataResponse.extensions
cleaned["dnssecdata"] = None
for extension in extensions:
if isinstance(extension,extensions.DNSSECExtension):
for extension in returned_extensions:
if isinstance(extension, extensions.DNSSECExtension):
cleaned["dnssecdata"] = extension
# get contact info, if there are any