fix: test after rebase

ICIJ · Nov 20, 2024 · fe4d205 · fe4d205
1 parent da04c9a
commit fe4d205
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/...app/src/test/java/org/icij/datashare/tasks/CreateNlpBatchesFromIndexParametrizedTest.java b/...app/src/test/java/org/icij/datashare/tasks/CreateNlpBatchesFromIndexParametrizedTest.java
@@ -66,7 +66,7 @@ void enqueueBatch(List<Document> batch) {
     public void setUp() {
         DatashareTaskFactory factory = mock(DatashareTaskFactory.class);
         when(factory.createBatchNlpTask(any(), any())).thenReturn(mock(BatchNlpTask.class));
-        taskManager = new TaskManagerMemory(new LinkedBlockingQueue<>(), factory);
+        taskManager = new TaskManagerMemory(new LinkedBlockingQueue<>(), factory, new PropertiesProvider());
     }
 
     @After
@@ -76,7 +76,8 @@ public void tearDown() throws IOException {
     }
 
 
-    public CreateNlpBatchesFromIndexParametrizedTest(int batchSize, int scrollSize, List<List<Language>> expectedLanguages) {
+    public CreateNlpBatchesFromIndexParametrizedTest(int batchSize, int scrollSize,
+                                                     List<List<Language>> expectedLanguages) {
         this.batchSize = batchSize;
         this.scrollSize = scrollSize;
         this.expectedLanguages = expectedLanguages;
@@ -86,7 +87,8 @@ public CreateNlpBatchesFromIndexParametrizedTest(int batchSize, int scrollSize,
     public static Collection<Object[]> taskParams() {
         return List.of(
             new Object[] {7, 3, List.of(
-                List.of(Language.ENGLISH, Language.ENGLISH, Language.ENGLISH, Language.ENGLISH, Language.ENGLISH, Language.ENGLISH, Language.ENGLISH),
+                List.of(Language.ENGLISH, Language.ENGLISH, Language.ENGLISH, Language.ENGLISH, Language.ENGLISH,
+                    Language.ENGLISH, Language.ENGLISH),
                 List.of(Language.ENGLISH, Language.ENGLISH, Language.ENGLISH),
                 List.of(Language.FRENCH, Language.FRENCH, Language.FRENCH, Language.FRENCH, Language.FRENCH),
                 List.of(Language.SPANISH, Language.SPANISH, Language.SPANISH, Language.SPANISH, Language.SPANISH)

diff --git a/datashare-app/src/test/java/org/icij/datashare/tasks/CreateNlpBatchesFromIndexTest.java b/datashare-app/src/test/java/org/icij/datashare/tasks/CreateNlpBatchesFromIndexTest.java
@@ -41,7 +41,7 @@ public class CreateNlpBatchesFromIndexTest {
     public void setUp() {
         DatashareTaskFactory factory = mock(DatashareTaskFactory.class);
         when(factory.createBatchNlpTask(any(), any())).thenReturn(mock(BatchNlpTask.class));
-        taskManager = new TaskManagerMemory(new LinkedBlockingQueue<>(), factory);
+        taskManager = new TaskManagerMemory(new LinkedBlockingQueue<>(), factory, new PropertiesProvider());
     }
 
     @After