Fix a unit tests that did not yet specify event types and minimum lik…

…elihood in the GepiRequestData.
JULIELab · Sep 20, 2024 · 8222f05 · 8222f05
1 parent 9757c79
commit 8222f05
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/gepi/gepi-core/src/main/java/de/julielab/gepi/core/retrieval/data/GepiRequestData.java b/gepi/gepi-core/src/main/java/de/julielab/gepi/core/retrieval/data/GepiRequestData.java
@@ -194,6 +194,11 @@ public GepiRequestData withDocId(String docid) {
         return this;
     }
 
+    public GepiRequestData withMinimumEventLikelihood(int eventLikelihood) {
+        this.eventLikelihood = eventLikelihood;
+        return this;
+    }
+
     @Override
     public String toString() {
         return "GepiRequestData{" +

diff --git a/gepi/gepi-core/src/test/java/de/julielab/gepi/core/services/GePiDataServiceTest.java b/gepi/gepi-core/src/test/java/de/julielab/gepi/core/services/GePiDataServiceTest.java
@@ -82,7 +82,7 @@ public void writeExcelSummary() throws Exception {
         final EventRetrievalResult eventRetrievalResult = new EventRetrievalResult();
         eventRetrievalResult.setEvents(events);
 
-        final GepiRequestData requestData = new GepiRequestData().withDataSessionId(1234).withInputMode(EnumSet.of(InputMode.A));
+        final GepiRequestData requestData = new GepiRequestData().withDataSessionId(1234).withInputMode(EnumSet.of(InputMode.A)).withEventTypes("Phosphorylation").withMinimumEventLikelihood(1);
         Path outputFile = gePiDataService.getOverviewExcel(CompletableFuture.completedFuture(eventRetrievalResult), requestData);
         assertThat(outputFile).exists();
     }