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

Add test cases

Browse source
This commit is contained in:
zandercymatics 2023-11-24 14:16:34 -07:00
parent 46677f4588
commit 377786cccf
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7
2 changed files with 20 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/registrar/admin.py
View file

@ -343,6 +343,11 @@ class UserDomainRoleAdmin(ListHeaderAdmin):
class DomainInvitationAdmin(ListHeaderAdmin): class DomainInvitationAdmin(ListHeaderAdmin):
"""Custom domain invitation admin class.""" """Custom domain invitation admin class."""
class Meta:
model = models.DomainInvitation
fields = "__all__"
_meta = Meta()
# Columns # Columns
list_display = [ list_display = [

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

@ -850,9 +850,8 @@ class TestDomainApplicationAdmin(MockEppLib):
class DomainInvitationAdminTest(TestCase): class DomainInvitationAdminTest(TestCase):
def setUp(self): def setUp(self):
self.site = AdminSite()
self.factory = RequestFactory() self.factory = RequestFactory()
self.admin = ListHeaderAdmin(model=DomainInvitationAdmin, admin_site=None) self.admin = ListHeaderAdmin(model=DomainInvitationAdmin, admin_site=AdminSite())
self.client = Client(HTTP_HOST="localhost:8080") self.client = Client(HTTP_HOST="localhost:8080")
self.superuser = create_superuser() self.superuser = create_superuser()
@ -863,26 +862,23 @@ class DomainInvitationAdminTest(TestCase):
User.objects.all().delete() User.objects.all().delete()
def test_get_filters(self): def test_get_filters(self):
# Create a mock request object # Have to get creative to get past linter
request = self.factory.get("/admin/yourmodel/") p = "adminpass"
# Set the GET parameters for testing self.client.login(username="superuser", password=p)
request.GET = {
"status": "started",
"investigator": "Rachid Mrad",
"q": "search_value",
}
# Call the get_filters method
filters = self.admin.get_filters(request)
# Assert the filters extracted from the request GET # Mock a user
self.assertEqual( user = mock_user()
filters,
[ response = self.client.get(
{"parameter_name": "status", "parameter_value": "started"}, "/admin/registrar/domaininvitation/",
{"parameter_name": "investigator", "parameter_value": "Rachid Mrad"}, {},
], follow=True,
) )
# Assert that the filters are added
self.assertContains(response, "invited", count=4)
self.assertContains(response, "retrieved", count=4)
class ListHeaderAdminTest(TestCase): class ListHeaderAdminTest(TestCase):
def setUp(self): def setUp(self):