From 09a66162dda4243547c38d4423e81618262c1e6b Mon Sep 17 00:00:00 2001
From: Becky Smith <becky.smith@thedatalab.org>
Date: Thu, 13 Feb 2025 13:47:41 +0000
Subject: [PATCH] Show uploaded file count with icons on request overview page

---
 airlock/models.py                             | 15 ++--
 .../file_browser/request/request.html         | 12 +++
 tests/functional/test_request_pages.py        | 85 +++++++++++++++++++
 3 files changed, 106 insertions(+), 6 deletions(-)

diff --git a/airlock/models.py b/airlock/models.py
index d20b3926..797941b3 100644
--- a/airlock/models.py
+++ b/airlock/models.py
@@ -841,13 +841,16 @@ def output_files(self) -> dict[UrlPath, RequestFile]:
             if rfile.filetype == RequestFileType.OUTPUT
         }
 
+    def uploaded_files_count(self):
+        if self.status not in [RequestStatus.APPROVED, RequestStatus.RELEASED]:
+            return 0
+        return sum(1 for rfile in self.output_files().values() if rfile.uploaded)
+
     def supporting_files_count(self):
-        return len(
-            [
-                1
-                for rfile in self.all_files_by_name.values()
-                if rfile.filetype == RequestFileType.SUPPORTING
-            ]
+        return sum(
+            1
+            for rfile in self.all_files_by_name.values()
+            if rfile.filetype == RequestFileType.SUPPORTING
         )
 
     def request_filetype(self, urlpath: UrlPath):
diff --git a/airlock/templates/file_browser/request/request.html b/airlock/templates/file_browser/request/request.html
index 0244aaa6..a444cc9b 100644
--- a/airlock/templates/file_browser/request/request.html
+++ b/airlock/templates/file_browser/request/request.html
@@ -155,6 +155,18 @@
     {% #description_item title="Supporting files not for release" %}
       {{ release_request.supporting_files_count }}
     {% /description_item %}
+    {% #description_item title="Files released and uploaded" %}
+      <div class="flex flex-row flex-wrap gap-6" id="uploaded-files-count">
+        {{ release_request.uploaded_files_count }}
+        {% if release_request.status.value == "RELEASED" %}
+          {% icon_check_circle_solid class="h-5 w-5 text-green-700" %}
+        {% elif release_request.upload_in_progress %}
+          {% icon_custom_spinner class="h-5 w-5 text-bn-egg-500 animate-spin stroke-current stroke-2" %}
+        {% elif release_request.can_be_rereleased %}
+          {% icon_exclamation_triangle_solid class="h-5 w-5 text-red-800" %}
+        {% endif %}
+      </div>
+    {% /description_item %}
   {% /description_list %}
 {% /card %}
 
diff --git a/tests/functional/test_request_pages.py b/tests/functional/test_request_pages.py
index fec88482..010bfb92 100644
--- a/tests/functional/test_request_pages.py
+++ b/tests/functional/test_request_pages.py
@@ -1,4 +1,5 @@
 import pytest
+from django.conf import settings
 from playwright.sync_api import expect
 
 from airlock.enums import RequestFileType, RequestStatus, Visibility
@@ -613,3 +614,87 @@ def test_request_releaseable(live_server, context, page, bll):
     expect(release_files_button).to_be_enabled()
     for locator in [return_request_button, reject_request_button]:
         expect(locator).not_to_be_visible()
+
+
+@pytest.mark.parametrize(
+    "status,uploaded_files,failed_uploaded_files,uploaded_count,release_button_enabled,icon_colour",
+    [
+        (RequestStatus.REVIEWED, [], [], 0, True, ""),
+        (RequestStatus.APPROVED, ["file1.txt"], [], 1, False, "text-bn-egg-500"),
+        (RequestStatus.APPROVED, [], ["file1.txt"], 0, False, "text-bn-egg-500"),
+        (RequestStatus.APPROVED, ["file1.txt"], ["file2.txt"], 1, True, "text-red-800"),
+        (
+            RequestStatus.APPROVED,
+            [],
+            ["file1.txt", "file2.txt"],
+            0,
+            True,
+            "text-red-800",
+        ),
+        (
+            RequestStatus.RELEASED,
+            ["file1.txt", "file2.txt"],
+            [],
+            2,
+            False,
+            "text-green-700",
+        ),
+    ],
+)
+def test_request_uploaded_files_status(
+    live_server,
+    context,
+    page,
+    bll,
+    mock_old_api,
+    mock_notifications,
+    status,
+    uploaded_files,
+    failed_uploaded_files,
+    uploaded_count,
+    release_button_enabled,
+    icon_colour,
+):
+    release_request = factories.create_request_at_status(
+        "workspace",
+        status=status,
+        files=[
+            factories.request_file(group="group", path="file1.txt", approved=True),
+            factories.request_file(group="group", path="file2.txt", approved=True),
+        ],
+    )
+    output_checker = login_as_user(
+        live_server,
+        context,
+        user_dict={
+            "username": "output_checker",
+            "workspaces": {},
+            "output_checker": True,
+        },
+    )
+    for path in uploaded_files:
+        bll.register_file_upload(release_request, UrlPath(path), output_checker)
+
+    for path in failed_uploaded_files:
+        for _ in range(settings.UPLOAD_MAX_ATTEMPTS):
+            bll.register_file_upload_attempt(release_request, UrlPath(path))
+
+    release_request = factories.refresh_release_request(release_request)
+    page.goto(live_server.url + release_request.get_url())
+
+    release_files_button = page.locator("#release-files-button")
+    uploaded_files_count_el = page.locator("#uploaded-files-count")
+
+    expect(uploaded_files_count_el).to_contain_text(str(uploaded_count))
+    # we're using the icon component, so the easiest way to check the icon is correct
+    # is to look for its colour
+    assert icon_colour in uploaded_files_count_el.inner_html()
+
+    if status == RequestStatus.RELEASED:
+        expect(release_files_button).not_to_be_visible()
+    else:
+        expect(release_files_button).to_be_visible()
+        if release_button_enabled:
+            expect(release_files_button).to_be_enabled()
+        else:
+            expect(release_files_button).not_to_be_enabled()