Add test for generics

GiGurra · May 21, 2016 · 576da79 · 576da79
1 parent 2980c2f
commit 576da79
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/src/test/scala/se/gigurra/heisenberg/HeisenbergSpec.scala b/src/test/scala/se/gigurra/heisenberg/HeisenbergSpec.scala
@@ -115,6 +115,24 @@ class HeisenbergSpec
       instance.flatten shouldBe MapData(req_str -> "Hello", req_str_w_def -> "mydefault").source
     }
 
+    "Handle generic types" in {
+      import GenericsTest._
+      {
+        val schema = MyType.schema[Int]
+        import schema._
+        val source = MapData(t -> 42).source
+        val o = MyType[Int](source)
+        o.t shouldBe 42
+      }
+      {
+        val schema = MyType.schema[String]
+        import schema._
+        val source = MapData(t -> "s").source
+        val o = MyType[String](source)
+        o.t shouldBe "s"
+      }
+    }
+
     "flatten with default values recursively" in {
       import NestedTestType._
       import SimpleTestType._

diff --git a/src/test/scala/se/gigurra/heisenberg/TestTypes.scala b/src/test/scala/se/gigurra/heisenberg/TestTypes.scala
@@ -3,6 +3,9 @@ package se.gigurra.heisenberg
 import se.gigurra.heisenberg.MapData._
 import se.gigurra.heisenberg.Migration._
 
+import scala.collection.concurrent.TrieMap
+import scala.reflect.runtime.universe.WeakTypeTag
+
 /**
   * Created by kjolh on 12/27/2015.
   */
@@ -102,6 +105,25 @@ object TestTypes {
 
   }
 
+  object GenericsTest {
+
+    case class MyType[T : WeakTypeTag : MapDataParser : MapDataProducer](source: SourceData = Map.empty) extends Parsed[MyTypeSchema[T]] {
+      val t = parse(schema.t)
+    }
+
+    object MyType {
+      private val schemas = new TrieMap[WeakTypeTag[_], MyTypeSchema[_]]()
+      implicit def schema[T: WeakTypeTag : MapDataParser : MapDataProducer]: MyTypeSchema[T] = {
+        schemas.getOrElseUpdate(implicitly[WeakTypeTag[T]], MyTypeSchema()).asInstanceOf[MyTypeSchema[T]]
+      }
+    }
+
+    case class MyTypeSchema[T : WeakTypeTag : MapDataParser : MapDataProducer]() extends Schema[MyType[T]] {
+      def apply(objectData: SourceData) = MyType.apply(objectData)
+      val t = required[T]("t")
+    }
+  }
+
   object ParseFieldMoreThanOnceTest {
 
     object SomeTestType extends Schema[SomeTestType] {