1 files changed, 19 insertions, 12 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeTrialService.scala b/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeTrialService.scala
index 9aeb820..cd62d48 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeTrialService.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/services/fake/FakeTrialService.scala
@@ -58,19 +58,27 @@ class FakeTrialService extends TrialService {
     Future.successful(GetListReply.EntityList(Seq(trial), 1, None))
 
   override def getTrialWithLabels(trialId: StringId[Trial], condition: String)(
-          implicit requestContext: AuthenticatedRequestContext): Future[GetTrialWithLabelsReply] = {
-    Future.successful(
-      GetTrialWithLabelsReply.Entity(ExportTrialWithLabels(
-        StringId[Trial]("NCT" + generators.nextInt(999999).toString),
-        UuidId[Trial](generators.nextUuid()),
-        generators.oneOf("adenocarcinoma", "breast", "prostate"),
-        LocalDateTime.now(),
-        labelVersion = 1L,
-        generators.listOf(new ExportTrialArm(
+          implicit requestContext: AuthenticatedRequestContext): Future[GetTrialWithLabelsReply] =
+    Future.successful(GetTrialWithLabelsReply.Entity(nextExportTrialWithLabels()))
+
+  override def getTrialsWithLabels(condition: String)(
+          implicit requestContext: AuthenticatedRequestContext): Future[GetTrialsWithLabelsReply] =
+    Future.successful(GetTrialsWithLabelsReply.EntityList(generators.seqOf(nextExportTrialWithLabels())))
+
+  private def nextExportTrialWithLabels() =
+    ExportTrialWithLabels(
+      StringId[Trial]("NCT" + generators.nextInt(999999).toString),
+      UuidId[Trial](generators.nextUuid()),
+      generators.oneOf("adenocarcinoma", "breast", "prostate"),
+      LocalDateTime.now(),
+      labelVersion = 1L,
+      generators.listOf(
+        new ExportTrialArm(
           LongId[Arm](generators.nextInt(999999).toLong),
           generators.nextName().value
         )),
-        generators.listOf(new ExportTrialLabelCriterion(
+      generators.listOf(
+        new ExportTrialLabelCriterion(
           LongId[Criterion](generators.nextInt(999999).toLong),
           generators.nextOption(generators.nextBoolean()),
           LongId[Label](generators.nextInt(999999).toLong),
@@ -79,8 +87,7 @@ class FakeTrialService extends TrialService {
           generators.nextBoolean(),
           generators.nextBoolean()
         ))
-      )))
-  }
+    )
 
   def update(origTrial: Trial, draftTrial: Trial)(
           implicit requestContext: AuthenticatedRequestContext): Future[UpdateReply] =