zydronium/manage.get.gov
1
0
Fork 0
mirror of https://github.com/cisagov/manage.get.gov.git synced 2025-05-15 17:17:02 +02:00
Code Issues Projects Releases Packages Wiki Activity

Test cases, black linting

Browse source
This commit is contained in:
zandercymatics 2024-02-09 15:08:08 -07:00
parent bfc82bda01
commit 10e8317e3c
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
3 changed files with 149 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/registrar/admin.py
View file

@ -1156,7 +1156,7 @@ class DomainAdmin(ListHeaderAdmin):
return None
# Get the date we want to update to
desired_date = date.today() + relativedelta(years=1)
desired_date = self._get_current_date() + relativedelta(years=1)
# Grab the current expiration date
try:
@ -1164,7 +1164,7 @@ class DomainAdmin(ListHeaderAdmin):
except KeyError:
# if no expiration date from registry, set it to today
logger.warning("current expiration date not set; setting to today")
exp_date = date.today()
exp_date = self._get_current_date()
# If the expiration date is super old (2020, for example), we need to
# "catch up" to the current year, so we add the difference.
@ -1178,9 +1178,10 @@ class DomainAdmin(ListHeaderAdmin):
# Renew the domain.
try:
obj.renew_domain(length=years)
self.message_user(
request,
f"Successfully extended the expiration date.",
"Successfully extended the expiration date.",
)
except RegistryError as err:
if err.is_connection_error():
@ -1203,6 +1204,13 @@ class DomainAdmin(ListHeaderAdmin):
return HttpResponseRedirect(".")
# Workaround for unit tests, as we cannot mock date directly.
# it is immutable. Rather than dealing with a convoluted workaround,
# lets wrap this in a function.
def _get_current_date(self):
"""Gets the current date"""
return date.today()
def do_delete_domain(self, request, obj):
if not isinstance(obj, Domain):
# Could be problematic if the type is similar,