formatted code

This commit is contained in:
David Kennedy 2023-11-16 08:42:53 -05:00
parent 944c82b9f9
commit 3107a9213d
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
3 changed files with 7 additions and 11 deletions

View file

@ -218,7 +218,7 @@ class Domain(TimeStampedModel, DomainHelper):
def registry_expiration_date(self, ex_date: date): def registry_expiration_date(self, ex_date: date):
""" """
Direct setting of the expiration date in the registry is not implemented. Direct setting of the expiration date in the registry is not implemented.
To update the expiration date, use renew_domain method.""" To update the expiration date, use renew_domain method."""
raise NotImplementedError() raise NotImplementedError()
@ -226,7 +226,7 @@ class Domain(TimeStampedModel, DomainHelper):
""" """
Renew the domain to a length and unit of time relative to the current Renew the domain to a length and unit of time relative to the current
expiration date. expiration date.
Default length and unit of time are 1 year. Default length and unit of time are 1 year.
""" """
# if no expiration date from registry, set to today # if no expiration date from registry, set to today
@ -236,11 +236,7 @@ class Domain(TimeStampedModel, DomainHelper):
cur_exp_date = date.today() cur_exp_date = date.today()
# create RenewDomain request # create RenewDomain request
request = commands.RenewDomain( request = commands.RenewDomain(name=self.name, cur_exp_date=cur_exp_date, period=epp.Period(length, unit))
name=self.name,
cur_exp_date=cur_exp_date,
period = epp.Period(length, unit)
)
try: try:
# update expiration date in registry, and set the updated # update expiration date in registry, and set the updated

View file

@ -907,7 +907,7 @@ class MockEppLib(TestCase):
res_data=[self.mockRenewedDomainExpDate], res_data=[self.mockRenewedDomainExpDate],
code=ErrorCode.COMMAND_COMPLETED_SUCCESSFULLY, code=ErrorCode.COMMAND_COMPLETED_SUCCESSFULLY,
) )
def mockInfoDomainCommands(self, _request, cleaned): def mockInfoDomainCommands(self, _request, cleaned):
request_name = getattr(_request, "name", None) request_name = getattr(_request, "name", None)

View file

@ -1972,7 +1972,7 @@ class TestExpirationDate(MockEppLib):
def test_expiration_date_setter_not_implemented(self): def test_expiration_date_setter_not_implemented(self):
"""assert that the setter for expiration date is not implemented and will raise error""" """assert that the setter for expiration date is not implemented and will raise error"""
with self.assertRaises(NotImplementedError) as err: with self.assertRaises(NotImplementedError):
self.domain.registry_expiration_date = datetime.date.today() self.domain.registry_expiration_date = datetime.date.today()
def test_renew_domain(self): def test_renew_domain(self):
@ -1984,10 +1984,10 @@ class TestExpirationDate(MockEppLib):
def test_renew_domain_error(self): def test_renew_domain_error(self):
"""assert that the renew_domain raises an exception when registry raises error""" """assert that the renew_domain raises an exception when registry raises error"""
with self.assertRaises(RegistryError) as err: with self.assertRaises(RegistryError):
self.domain_w_error.renew_domain() self.domain_w_error.renew_domain()
class TestAnalystClientHold(MockEppLib): class TestAnalystClientHold(MockEppLib):
"""Rule: Analysts may suspend or restore a domain by using client hold""" """Rule: Analysts may suspend or restore a domain by using client hold"""