Add TaskRegistryFactory integration test

Author: soerenreichardt

core/src/main/java/org/neo4j/gds/core/utils/progress/LocalTaskRegistryFactory.java

@@ -22,9 +22,9 @@
 public class LocalTaskRegistryFactory implements TaskRegistryFactory {
 
     private final String username;
-    private final GlobalTaskStore taskStore;
+    private final TaskStore taskStore;
 
-    LocalTaskRegistryFactory(String username, GlobalTaskStore taskStore) {
+    LocalTaskRegistryFactory(String username, TaskStore taskStore) {
         this.username = username;
         this.taskStore = taskStore;
     }

core/src/test/java/org/neo4j/gds/core/utils/progress/LocalTaskRegistryFactoryTest.java

@@ -0,0 +1,57 @@
+/*
+ * Copyright (c) "Neo4j"
+ * Neo4j Sweden AB [http://neo4j.com]
+ *
+ * This file is part of Neo4j.
+ *
+ * Neo4j is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package org.neo4j.gds.core.utils.progress;
+
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.neo4j.gds.core.utils.progress.tasks.Tasks;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class LocalTaskRegistryFactoryTest {
+
+    TaskStore taskStore;
+    TaskRegistryFactory taskRegistryFactory;
+
+    @BeforeEach
+    void setup() {
+        this.taskStore = new GlobalTaskStore();
+        this.taskRegistryFactory = new LocalTaskRegistryFactory("", taskStore);
+    }
+
+    @Test
+    void shouldPutAndRemoveDistinctTasks() {
+        var task1 = Tasks.leaf("root1");
+        var taskRegistry1 = taskRegistryFactory.newInstance();
+        taskRegistry1.registerTask(task1);
+
+        assertThat(taskStore.query("")).size().isEqualTo(1);
+
+        var task2 = Tasks.leaf("root2");
+        var taskRegistry2 = taskRegistryFactory.newInstance();
+        taskRegistry2.registerTask(task2);
+
+        assertThat(taskStore.query("")).size().isEqualTo(2);
+
+        taskRegistry1.unregisterTask();
+
+        assertThat(taskStore.query("")).containsValue(task2).doesNotContainValue(task1);
+    }
+}