total size in a node

Author: sanderegg

services/storage/tests/unit/test_handlers_paths.py

@@ -530,14 +530,16 @@ async def test_path_compute_size(
     total_num_files = sum(
         len(files_in_node) for files_in_node in list_of_files.values()
     )
-    expected_project_total_size = project_params.allowed_file_sizes[0] * total_num_files
 
+    # get size of a full project
+    expected_total_size = project_params.allowed_file_sizes[0] * total_num_files
+    path = Path(project["uuid"])
     url = url_from_operation_id(
         client,
         initialized_app,
         "compute_path_total_size",
         location_id=f"{location_id}",
-        path=project["uuid"],
+        path=f"{path}",
     ).with_query(user_id=user_id)
     response = await client.post(f"{url}")
 
@@ -547,5 +549,32 @@ async def test_path_compute_size(
         PathTotalSizeCreate,
     )
     assert received
-    assert received.path == Path(project["uuid"])
-    assert received.size == expected_project_total_size
+    assert received.path == path
+    assert received.size == expected_total_size
+
+    # get size of a project/node
+    selected_node_id = NodeID(random.choice(list(project["workbench"])))  # noqa: S311
+    selected_node_s3_keys = [
+        Path(s3_object_id) for s3_object_id in list_of_files[selected_node_id]
+    ]
+    expected_total_size = project_params.allowed_file_sizes[0] * len(
+        selected_node_s3_keys
+    )
+    path = Path(project["uuid"]) / f"{selected_node_id}"
+    url = url_from_operation_id(
+        client,
+        initialized_app,
+        "compute_path_total_size",
+        location_id=f"{location_id}",
+        path=f"{path}",
+    ).with_query(user_id=user_id)
+    response = await client.post(f"{url}")
+
+    received, _ = assert_status(
+        response,
+        status.HTTP_200_OK,
+        PathTotalSizeCreate,
+    )
+    assert received
+    assert received.path == path
+    assert received.size == expected_total_size