From 828cc12194663f107a08251bbabc256ae0d7c936 Mon Sep 17 00:00:00 2001
From: vlad <vlad@drivergrp.com>
Date: Thu, 8 Dec 2016 18:52:37 -0800
Subject: General GADT json format

---
 src/test/scala/xyz/driver/core/JsonTest.scala  | 38 +++++++++++++++++++++++++-
 src/test/scala/xyz/driver/core/TestTypes.scala | 14 ++++++++++
 2 files changed, 51 insertions(+), 1 deletion(-)
 create mode 100644 src/test/scala/xyz/driver/core/TestTypes.scala

(limited to 'src/test/scala/xyz')

diff --git a/src/test/scala/xyz/driver/core/JsonTest.scala b/src/test/scala/xyz/driver/core/JsonTest.scala
index c113c59..eb8d5d8 100644
--- a/src/test/scala/xyz/driver/core/JsonTest.scala
+++ b/src/test/scala/xyz/driver/core/JsonTest.scala
@@ -1,9 +1,11 @@
 package xyz.driver.core
 
 import org.scalatest.{FlatSpec, Matchers}
-import xyz.driver.core.json.{EnumJsonFormat, ValueClassFormat}
+import xyz.driver.core.json.{EnumJsonFormat, GadtJsonFormat, ValueClassFormat}
 import xyz.driver.core.revision.Revision
 import xyz.driver.core.time.provider.SystemTimeProvider
+import spray.json._
+import xyz.driver.core.TestTypes.CustomGADT
 
 class JsonTest extends FlatSpec with Matchers {
 
@@ -98,4 +100,38 @@ class JsonTest extends FlatSpec with Matchers {
     parsedValue1 should be(referenceValue1)
     parsedValue2 should be(referenceValue2)
   }
+
+  "Json format for classes GADT" should "read and write correct JSON" in {
+
+    import CustomGADT._
+    import DefaultJsonProtocol._
+    implicit val case1Format = jsonFormat1(GadtCase1)
+    implicit val case2Format = jsonFormat1(GadtCase2)
+    implicit val case3Format = jsonFormat1(GadtCase3)
+
+    val format = GadtJsonFormat.create[CustomGADT]("gadtTypeField") {
+      case t1: CustomGADT.GadtCase1 => "case1"
+      case t2: CustomGADT.GadtCase2 => "case2"
+      case t3: CustomGADT.GadtCase3 => "case3"
+    } {
+      case "case1" => case1Format
+      case "case2" => case2Format
+      case "case3" => case3Format
+    }
+
+    val referenceValue1 = CustomGADT.GadtCase1("4")
+    val referenceValue2 = CustomGADT.GadtCase2("Hi!")
+
+    val writtenJson1 = format.write(referenceValue1)
+    writtenJson1 should be("{\n \"field\": \"4\",\n\"gadtTypeField\": \"case1\"\n}".parseJson)
+
+    val writtenJson2 = format.write(referenceValue2)
+    writtenJson2 should be("{\"field\":\"Hi!\",\"gadtTypeField\":\"case2\"}".parseJson)
+
+    val parsedValue1 = format.read(writtenJson1)
+    val parsedValue2 = format.read(writtenJson2)
+
+    parsedValue1 should be(referenceValue1)
+    parsedValue2 should be(referenceValue2)
+  }
 }
diff --git a/src/test/scala/xyz/driver/core/TestTypes.scala b/src/test/scala/xyz/driver/core/TestTypes.scala
new file mode 100644
index 0000000..bb25deb
--- /dev/null
+++ b/src/test/scala/xyz/driver/core/TestTypes.scala
@@ -0,0 +1,14 @@
+package xyz.driver.core
+
+object TestTypes {
+
+  sealed trait CustomGADT {
+    val field: String
+  }
+
+  object CustomGADT {
+    final case class GadtCase1(field: String) extends CustomGADT
+    final case class GadtCase2(field: String) extends CustomGADT
+    final case class GadtCase3(field: String) extends CustomGADT
+  }
+}
-- 
cgit v1.2.3