fix tests

raft-tech · Oct 18, 2023 · dc99ebd · dc99ebd
1 parent 0ddd03d
commit dc99ebd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tdrs-backend/tdpservice/search_indexes/test/test_model_mapping.py b/tdrs-backend/tdpservice/search_indexes/test/test_model_mapping.py
@@ -581,11 +581,11 @@ def test_can_create_and_index_ssp_m3_submission():
 
 
 @pytest.mark.django_db
-def test_can_create_and_index_ssp_m6_submission():
+def test_can_create_and_index_ssp_m6_submission(test_datafile):
     """SSP M6 submissions can be created and mapped."""
     record_num = fake.uuid4()
 
-    submission = models.tanf.SSP_M6()
+    submission = models.ssp.SSP_M6()
     submission.datafile = test_datafile
     submission.RecordType = record_num
     submission.CALENDAR_QUARTER = 1
@@ -610,7 +610,7 @@ def test_can_create_and_index_ssp_m6_submission():
 
     assert submission.id is not None
 
-    search = documents.tanf.SSP_M6DataSubmissionDocument.search().query(
+    search = documents.ssp.SSP_M6DataSubmissionDocument.search().query(
         'match',
         RecordType=record_num
     )

diff --git a/tdrs-backend/tdpservice/users/test/test_permissions.py b/tdrs-backend/tdpservice/users/test/test_permissions.py
@@ -123,6 +123,9 @@ def test_ofa_system_admin_permissions(ofa_system_admin):
         'search_indexes.add_ssp_m3',
         'search_indexes.view_ssp_m3',
         'search_indexes.change_ssp_m3',
+        'search_indexes.add_ssp_m6',
+        'search_indexes.view_ssp_m6',
+        'search_indexes.change_ssp_m6',
     }
     group_permissions = ofa_system_admin.get_group_permissions()
     assert group_permissions == expected_permissions