fix failing tests

Author: IoannisPanagiotas

doc/modules/ROOT/pages/common-usage/monitoring-system.adoc

@@ -121,7 +121,7 @@ This procedure can help a user realize which of his operations might be memory d
 
 A user can access only his own graphs or tasks, but an admin can obtain information for all users.
 
-[[syntax]]
+[[mem-reporting-syntax]]
 ==== Syntax
 .List the graphs and tasks of a user along with their memory footprint.
 [source, cypher, role=noplay]
@@ -173,7 +173,7 @@ The `gds.listMemory.summary` procedure provides an overview of the total graph m
 
 A user can access only his own graphs or tasks, but an admin can obtain information for all users.
 
-[[syntax]]
+[[mem-summary-syntax]]
 ==== Syntax
 .List the graphs and tasks of a user along with their memory footprint.
 [source, cypher, role=noplay]

doc/modules/ROOT/pages/operations-reference/additional-operation-references.adoc

@@ -27,10 +27,13 @@
 | xref:common-usage/debug-sysinfo.adoc[Status of the system]                                   | `gds.debug.sysInfo` label:procedure[Procedure]
 | xref:installation/configure-apache-arrow-server.adoc[Monitoring]                              | `gds.debug.arrow` label:procedure[Procedure]
 | xref:common-usage/monitoring-system.adoc[Get an overview of the system's workload and available resources] | `gds.systemMonitor` label:procedure[Procedure]
+| xref:common-usage/monitoring-system.adoc[Detailed memory footprint reporting] | `gds.listMemory` label:procedure[Procedure]
+| xref:common-usage/monitoring-system.adoc[Memory footprint summary] | `gds.listMemory.summary` label:procedure[Procedure]
 | xref:management-ops/backup-restore.adoc[Back-up graphs and models to disk]             | `gds.backup` label:procedure[Procedure]
 | xref:management-ops/backup-restore.adoc[Restore persisted graphs and models to memory] | `gds.restore` label:procedure[Procedure]
 | xref:production-deployment/defaults-and-limits.adoc[List configured defaults] | `gds.config.defaults.list` label:procedure[Procedure]
 | xref:production-deployment/defaults-and-limits.adoc[Configure a default] | `gds.config.defaults.set` label:procedure[Procedure]
 | xref:production-deployment/defaults-and-limits.adoc#_limits_on_configuration_values[List configured limits] | `gds.config.limits.list` label:procedure[Procedure]
 | xref:production-deployment/defaults-and-limits.adoc#_limits_on_configuration_values[Configure a limit] | `gds.config.limits.set` label:procedure[Procedure]
+
 |===

open-packaging/src/test/java/org/neo4j/gds/OpenGdsProcedureSmokeTest.java

@@ -519,6 +519,9 @@ class OpenGdsProcedureSmokeTest extends BaseProcTest {
         "gds.prizeSteinerTree.stream",
         "gds.prizeSteinerTree.stream.estimate",
 
+        "gds.listMemory",
+        "gds.listMemory.summary",
+
         "gds.version"
     );
 
@@ -590,7 +593,7 @@ void countShouldMatch() {
         );
 
         // If you find yourself updating this count, please also update the count in SmokeTest.kt
-        int expectedCount = 432;
+        int expectedCount = 434;
         assertEquals(
             expectedCount,
             returnedRows,

progress-tracking/src/main/java/org/neo4j/gds/mem/TaskMemoryContainer.java

@@ -44,9 +44,13 @@ void reserve(String username, String taskName, JobId jobId,long memoryAmount){
     }
 
     long removeTask(UserTask task){
-         var mem=  memoryInUse.getOrDefault(task.username(), EMPTY_HASH_MAP).remove(task.jobId()).getRight();
-         allocatedMemory.addAndGet(-mem);
-         return  mem;
+            var memPair=  memoryInUse.getOrDefault(task.username(), EMPTY_HASH_MAP).remove(task.jobId());
+            if (memPair !=null){
+                var mem = memPair.getRight();
+                 allocatedMemory.addAndGet(-mem);
+                return mem;
+            }
+            return allocatedMemory.get();
     }
 
     long taskReservedMemory(){