1 files changed, 35 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/export.scala b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/export.scala
new file mode 100644
index 0000000..7f3c410
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/export.scala
@@ -0,0 +1,35 @@
+package xyz.driver.pdsuidomain.fakes.entities
+
+import xyz.driver.entities.labels.Label
+import xyz.driver.pdsuidomain.entities.{Arm, Criterion, Trial}
+import xyz.driver.pdsuidomain.entities.export.trial._
+
+object export {
+  import xyz.driver.core.generators._
+  import common._
+
+  def nextExportTrialArm(): ExportTrialArm =
+    ExportTrialArm(armId = nextLongId[Arm], armName = nextString(100))
+
+  def nextExportTrialLabelCriterion(): ExportTrialLabelCriterion =
+    ExportTrialLabelCriterion(
+      criterionId = nextLongId[Criterion],
+      value = nextOption[Boolean](nextBoolean()),
+      labelId = nextLongId[Label],
+      armIds = setOf(nextLongId[Arm]),
+      criteria = nextString(100),
+      isCompound = nextBoolean(),
+      isDefining = nextBoolean()
+    )
+
+  def nextExportTrialWithLabels(): ExportTrialWithLabels =
+    ExportTrialWithLabels(
+      nctId = nextStringId[Trial],
+      trialId = nextUuidId[Trial],
+      condition = nextString(100),
+      lastReviewed = nextLocalDateTime,
+      labelVersion = nextInt(100).toLong,
+      arms = listOf(nextExportTrialArm()),
+      criteria = listOf(nextExportTrialLabelCriterion())
+    )
+}