Skip to content

Commit 8ef5ea7

Browse files
committed
- Update test to use correct logic
- Remove auto filtering on datafiles and zap scans
1 parent 9b64982 commit 8ef5ea7

File tree

3 files changed

+9
-17
lines changed

3 files changed

+9
-17
lines changed

tdrs-backend/tdpservice/data_files/admin/admin.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -146,13 +146,10 @@ def get_queryset(self, request):
146146
if not (request.user.has_fra_access or request.user.is_an_admin):
147147
filtered_for_fra = qs.exclude(
148148
section__in=DataFile.get_fra_section_list(),
149-
user__account_approval_status=AccountApprovalStatusChoices.DEACTIVATED
150149
)
151150
return filtered_for_fra
152151
else:
153-
return qs.exclude(
154-
user__account_approval_status=AccountApprovalStatusChoices.DEACTIVATED
155-
)
152+
return qs
156153

157154
def reparse(self, request, queryset):
158155
"""Reparse the selected data files."""

tdrs-backend/tdpservice/security/admin.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,6 @@ def file_size_human(self, obj):
3030
"""Return human friendly file size, converted to appropriate unit."""
3131
return obj.file_size_humanized
3232

33-
def get_queryset(self, request):
34-
"""Override to exclude scans uploaded by deactivated users."""
35-
qs = super().get_queryset(request)
36-
return qs.exclude(uploaded_by__account_approval_status=AccountApprovalStatusChoices.DEACTIVATED)
37-
3833

3934
@admin.register(OwaspZapScan)
4035
class OwaspZapScanAdmin(ReadOnlyAdminMixin, admin.ModelAdmin):
@@ -54,11 +49,6 @@ class OwaspZapScanAdmin(ReadOnlyAdminMixin, admin.ModelAdmin):
5449
"app_target",
5550
]
5651

57-
def get_queryset(self, request):
58-
"""Override to exclude scans uploaded by deactivated users."""
59-
qs = super().get_queryset(request)
60-
return qs.exclude(uploaded_by__account_approval_status=AccountApprovalStatusChoices.DEACTIVATED)
61-
6252

6353
@admin.register(SecurityEventToken)
6454
class SecurityEventTokenAdmin(ReadOnlyAdminMixin, admin.ModelAdmin):

tdrs-backend/tdpservice/users/test/test_models.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -82,11 +82,15 @@ def test_user_with_fra_access(client, ofa_system_admin):
8282

8383
response = client.get(f"/admin/data_files/datafile/{datafile.id}/change/")
8484
assert response.status_code == 200
85-
assert '<div class="readonly">Fra Work Outcome Tanf Exiters</div>' in response.content.decode('utf-8')
85+
assert (
86+
'<div class="readonly">Fra Work Outcome Tanf Exiters</div>'
87+
in response.content.decode("utf-8")
88+
)
89+
8690

8791
@pytest.mark.django_db
8892
def test_user_without_fra_access(client, data_analyst):
89-
"""Test that a user in data analyst role cannot access ."""
93+
"""Test that a user in data analyst role with FRA permission cannot access FRA datafiles."""
9094
data_analyst.is_staff = True
9195

9296
data_analyst.clean()
@@ -100,4 +104,5 @@ def test_user_without_fra_access(client, data_analyst):
100104
datafile.save()
101105

102106
response = client.get(f"/admin/data_files/datafile/{datafile.id}/change/")
103-
assert response.status_code == 200
107+
# We get redirected to a blank admin datafiles page
108+
assert response.status_code == 302

0 commit comments

Comments
 (0)