TRIAL-54 Study designs as enum

1 files changed, 2 insertions, 6 deletions

diff --git a/src/test/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/StudyDesignFormatSuite.scala b/src/test/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/StudyDesignFormatSuite.scala
index 9cf9b5f..e847445 100644
--- a/src/test/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/StudyDesignFormatSuite.scala
+++ b/src/test/scala/xyz/driver/pdsuidomain/formats/json/sprayformats/StudyDesignFormatSuite.scala
@@ -2,20 +2,16 @@ package xyz.driver.pdsuidomain.formats.json.sprayformats
 
 import spray.json._
 import org.scalatest.{FlatSpec, Matchers}
-import xyz.driver.pdsuicommon.domain.LongId
 import xyz.driver.pdsuidomain.entities.StudyDesign
 
 class StudyDesignFormatSuite extends FlatSpec with Matchers {
   import studydesign._
 
   "Json format for StudyDesign" should "read and write correct JSON" in {
-    val studyDesign = StudyDesign(
-      id = LongId(10),
-      name = "study design name"
-    )
+    val studyDesign = StudyDesign.Randomized
     val writtenJson = studyDesignFormat.write(studyDesign)
 
-    writtenJson should be("""{"id":10,"name":"study design name"}""".parseJson)
+    writtenJson should be("""{"id":1,"name":"Randomized"}""".parseJson)
 
     val parsedStudyDesign = studyDesignFormat.read(writtenJson)
     parsedStudyDesign should be(studyDesign)