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

mockdb tests now more efficient

Browse source
This commit is contained in:
David Kennedy 2024-07-04 11:00:03 -04:00
parent a6f7cc6f02
commit 36985353c2
No known key found for this signature in database
GPG key ID: 6528A5386E66B96B
3 changed files with 40 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -898,8 +898,9 @@ class TestListHeaderAdmin(TestCase):
class TestMyUserAdmin(MockDb):
def setUp(self):
super().setUp()
@classmethod
def setUpClass(self):
super().setUpClass()
admin_site = AdminSite()
self.admin = MyUserAdmin(model=get_user_model(), admin_site=admin_site)
self.client = Client(HTTP_HOST="localhost:8080")
@ -907,8 +908,9 @@ class TestMyUserAdmin(MockDb):
self.staffuser = create_user()
self.test_helper = GenericTestHelper(admin=self.admin)
def tearDown(self):
super().tearDown()
@classmethod
def tearDownClass(self):
super().tearDownClass()
DomainRequest.objects.all().delete()
User.objects.all().delete()
@ -1094,6 +1096,15 @@ class TestMyUserAdmin(MockDb):
expected_href = reverse("admin:registrar_domain_change", args=[domain_deleted.pk])
self.assertNotContains(response, expected_href)
# Must clean up within test since MockDB is shared across tests for performance reasons
domain_request_started.delete()
domain_request_submitted.delete()
domain_request_in_review.delete()
domain_request_withdrawn.delete()
domain_request_approved.delete()
domain_request_rejected.delete()
domain_request_ineligible.delete()
domain_deleted.delete()
class AuditedAdminTest(TestCase):
def setUp(self):