add json format tests for Slot and Eligibility arms

3 files changed, 89 insertions, 3 deletions

diff --git a/src/main/scala/xyz/driver/pdsuidomain/entities/SlotArmEligibilityArm.scala b/src/main/scala/xyz/driver/pdsuidomain/entities/SlotArmEligibilityArm.scala
index f9aed6c..70114de 100644
--- a/src/main/scala/xyz/driver/pdsuidomain/entities/SlotArmEligibilityArm.scala
+++ b/src/main/scala/xyz/driver/pdsuidomain/entities/SlotArmEligibilityArm.scala
@@ -3,9 +3,7 @@ package xyz.driver.pdsuidomain.entities
 import xyz.driver.pdsuicommon.domain.LongId
 import xyz.driver.pdsuicommon.logging._
 
-final case class SlotArmEligibilityArm(slotArmId: LongId[SlotArm],
-                                       eligibilityArmId: LongId[EligibilityArm])
-
+final case class SlotArmEligibilityArm(slotArmId: LongId[SlotArm], eligibilityArmId: LongId[EligibilityArm])
 
 object SlotArmEligibilityArm {
   implicit def toPhiString(x: SlotArmEligibilityArm): PhiString = {
diff --git a/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/eligibilityarm.scala b/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/eligibilityarm.scala
new file mode 100644
index 0000000..acb790a
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/eligibilityarm.scala
@@ -0,0 +1,44 @@
+package xyz.driver.pdsuidomain.formats.json.sprayformats
+
+import spray.json._
+import xyz.driver.pdsuicommon.domain.{LongId, StringId}
+import xyz.driver.pdsuidomain.entities.{EligibilityArm, Trial}
+
+object eligibilityarm {
+  import DefaultJsonProtocol._
+  import common._
+
+  def applyUpdateToArm(json: JsValue, orig: EligibilityArm): EligibilityArm = json match {
+    case JsObject(fields) =>
+      val name = fields
+        .get("name")
+        .map(_.convertTo[String])
+        .getOrElse(deserializationError(s"Arm json object does not contain `name` field: $json"))
+      orig.copy(name = name)
+
+    case _ => deserializationError(s"Expected Json Object as partial Arm, but got $json")
+  }
+
+  def eligibilityArmFormat: RootJsonFormat[EligibilityArm] = new RootJsonFormat[EligibilityArm] {
+    override def write(obj: EligibilityArm): JsValue =
+      JsObject(
+        "id"           -> obj.id.toJson,
+        "name"         -> obj.name.toJson,
+        "originalName" -> obj.originalName.toJson,
+        "trialId"      -> obj.trialId.toJson
+      )
+
+    override def read(json: JsValue): EligibilityArm = json.asJsObject.getFields("trialId", "name") match {
+      case Seq(trialId, name) =>
+        EligibilityArm(
+          id = LongId(0),
+          name = name.convertTo[String],
+          trialId = trialId.convertTo[StringId[Trial]],
+          originalName = name.convertTo[String]
+        )
+
+      case _ => deserializationError(s"Expected Json Object as Arm, but got $json")
+    }
+  }
+
+}
diff --git a/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/slotarm.scala b/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/slotarm.scala
new file mode 100644
index 0000000..192bd6a
--- /dev/null
+++ b/src/main/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/slotarm.scala
@@ -0,0 +1,44 @@
+package xyz.driver.pdsuidomain.formats.json.sprayformats
+
+import spray.json._
+import xyz.driver.pdsuicommon.domain.{LongId, StringId}
+import xyz.driver.pdsuidomain.entities.{SlotArm, Trial}
+
+object slotarm {
+  import DefaultJsonProtocol._
+  import common._
+
+  def applyUpdateToArm(json: JsValue, orig: SlotArm): SlotArm = json match {
+    case JsObject(fields) =>
+      val name = fields
+        .get("name")
+        .map(_.convertTo[String])
+        .getOrElse(deserializationError(s"Arm json object does not contain `name` field: $json"))
+      orig.copy(name = name)
+
+    case _ => deserializationError(s"Expected Json Object as partial Arm, but got $json")
+  }
+
+  def slotArmFormat: RootJsonFormat[SlotArm] = new RootJsonFormat[SlotArm] {
+    override def write(obj: SlotArm): JsValue =
+      JsObject(
+        "id"           -> obj.id.toJson,
+        "name"         -> obj.name.toJson,
+        "originalName" -> obj.originalName.toJson,
+        "trialId"      -> obj.trialId.toJson
+      )
+
+    override def read(json: JsValue): SlotArm = json.asJsObject.getFields("trialId", "name") match {
+      case Seq(trialId, name) =>
+        SlotArm(
+          id = LongId(0),
+          name = name.convertTo[String],
+          trialId = trialId.convertTo[StringId[Trial]],
+          originalName = name.convertTo[String]
+        )
+
+      case _ => deserializationError(s"Expected Json Object as Arm, but got $json")
+    }
+  }
+
+}