Added swagger annotations for new entities

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main/scala/xyz/driver/pdsuicommon/utils/CustomSwaggerJsonFormats.scala b/src/main/scala/xyz/driver/pdsuicommon/utils/CustomSwaggerJsonFormats.scala
index 6d2a068..b718c1f 100644
--- a/src/main/scala/xyz/driver/pdsuicommon/utils/CustomSwaggerJsonFormats.scala
+++ b/src/main/scala/xyz/driver/pdsuicommon/utils/CustomSwaggerJsonFormats.scala
@@ -7,6 +7,8 @@ import spray.json.JsValue
 import xyz.driver.pdsuicommon.domain.{LongId, StringId, TextJson, UuidId}
 import xyz.driver.pdsuidomain.entities._
 import xyz.driver.pdsuidomain.formats.json.sprayformats.arm._
+import xyz.driver.pdsuidomain.formats.json.sprayformats.slotarm._
+import xyz.driver.pdsuidomain.formats.json.sprayformats.eligibilityarm._
 import xyz.driver.pdsuidomain.formats.json.sprayformats.criterion._
 import xyz.driver.pdsuidomain.formats.json.sprayformats.intervention._
 import xyz.driver.pdsuidomain.formats.json.sprayformats.hypothesis._
@@ -54,7 +56,10 @@ object CustomSwaggerJsonFormats {
     classOf[Hypothesis] -> hypothesisFormat.write(
       xyz.driver.pdsuidomain.fakes.entities.trialcuration.nextHypothesis()),
     classOf[StudyDesign] -> studyDesignFormat.write(
-      xyz.driver.pdsuidomain.fakes.entities.trialcuration.nextStudyDesign())
+      xyz.driver.pdsuidomain.fakes.entities.trialcuration.nextStudyDesign()),
+    classOf[EligibilityArmWithDiseases] -> eligibilityArmWithDiseasesWriter.write(
+      xyz.driver.pdsuidomain.fakes.entities.trialcuration.nextEligibilityArmWithDiseases()),
+    classOf[SlotArm] -> slotArmFormat.write(xyz.driver.pdsuidomain.fakes.entities.trialcuration.nextSlotArm())
   )
 
   // records-processing-service