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

Fix unit test

Browse source
This commit is contained in:
zandercymatics 2024-02-09 10:13:10 -07:00
parent 773d6e0cee
commit 3189b50baa
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 6 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/registrar/admin.py
View file

@ -1177,10 +1177,9 @@ class DomainAdmin(ListHeaderAdmin):
# Renew the domain. # Renew the domain.
try: try:
obj.renew_domain(length=years) obj.renew_domain(length=years)
updated_domain = Domain.objects.filter(id=obj.id).get()
self.message_user( self.message_user(
request, request,
f"Successfully extended expiration date to {updated_domain.registry_expiration_date}.", f"Successfully extended expiration date.",
) )
except RegistryError as err: except RegistryError as err:
if err.is_connection_error(): if err.is_connection_error():

13
src/registrar/tests/test_admin.py
View file

@ -37,7 +37,7 @@ from .common import (
) )
from django.contrib.sessions.backends.db import SessionStore from django.contrib.sessions.backends.db import SessionStore
from django.contrib.auth import get_user_model from django.contrib.auth import get_user_model
from unittest.mock import call, patch from unittest.mock import ANY, call, patch
from unittest import skip from unittest import skip
from django.conf import settings from django.conf import settings
@ -101,19 +101,16 @@ class TestDomainAdmin(MockEppLib, WebTest):
self.assertContains(response, "Extend expiration date") self.assertContains(response, "Extend expiration date")
# Ensure the message we recieve is in line with what we expect # Ensure the message we recieve is in line with what we expect
expected_message = f"Successfully extended expiration date to 2024-01-01." expected_message = f"Successfully extended expiration date."
#self.assertContains(response, expected_message)
expected_call = call( expected_call = call(
response, # The WGSI request doesn't need to be tested
ANY,
messages.INFO, messages.INFO,
expected_message, expected_message,
extra_tags="", extra_tags="",
fail_silently=False, fail_silently=False,
) )
mock_add_message.assert_has_calls(expected_call, 1) mock_add_message.assert_has_calls([expected_call], 1)
# Assert that the domain was updated correctly
expected_date = date(year=2025, month=1, day=1)
self.assertEqual(domain.expiration_date, expected_date)
def test_short_org_name_in_domains_list(self): def test_short_org_name_in_domains_list(self):
""" """