added: test for forceInsertQuery fixes

slick · Apr 26, 2016 · bf5561253182017d676d42d171cb27828bd9723a · bf55612
1 parent d00eaf7
commit bf5561253182017d676d42d171cb27828bd9723a
Unified Split

Showing with 46 additions and 0 deletions.

+1 −0 slick-testkit/src/main/resources/testkit-reference.conf

+45 −0 slick-testkit/src/main/scala/com/typesafe/slick/testkit/tests/ForceInsertQueryTest.scala
diff --git a/slick-testkit/src/main/resources/testkit-reference.conf b/slick-testkit/src/main/resources/testkit-reference.conf
@@ -46,6 +46,7 @@ testkit {
    ${testPackage}.TransactionTest
    ${testPackage}.UnionTest
    ${testPackage}.ForUpdateTest
+    ${testPackage}.ForceInsertQueryTest
  ]
 }

diff --git a/slick-testkit/src/main/scala/com/typesafe/slick/testkit/tests/ForceInsertQueryTest.scala b/slick-testkit/src/main/scala/com/typesafe/slick/testkit/tests/ForceInsertQueryTest.scala
@@ -0,0 +1,45 @@
+package com.typesafe.slick.testkit.tests
+
+import com.typesafe.slick.testkit.util.{AsyncTest, JdbcTestDB}
+
+class ForceInsertQueryTest extends AsyncTest[JdbcTestDB] {
+
+  import tdb.profile.api._
+
+  case class Person(id: Option[Long] = None,
+                    name: Option[String] = None,
+                    hairColor: Option[String] = None,
+                    eyeColor: Option[String] = None)
+
+  class People(tag: Tag) extends Table[Person](tag, "people") {
+    def id = column[Long]("id", O.PrimaryKey, O.AutoInc)
+
+    def name = column[Option[String]]("name")
+
+    def hairColor = column[Option[String]]("hair_color")
+
+    def eyeColor = column[Option[String]]("eye_color")
+
+    def * = (id.?, name, hairColor, eyeColor) <>(Person.tupled, Person.unapply)
+  }
+
+  object peopleTable extends TableQuery(new People(_)) {
+    def someForcedInsert(person: Person) = {
+      this.map(p => (p.name, p.hairColor, p.eyeColor))
+        .forceInsertQuery {
+          Query((person.name, person.hairColor, person.eyeColor))
+        }
+    }
+  }
+
+  def testForceInsert = {
+    DBIO.seq(
+      peopleTable.schema.create,
+      peopleTable.someForcedInsert(Person(name = Some("John"), hairColor = Some("Brown"), eyeColor = Some("Brown"))),
+      peopleTable.someForcedInsert(Person(name = Some("John"), hairColor = None, eyeColor = None)),
+      peopleTable.someForcedInsert(Person(name = None, hairColor = None, eyeColor = None)),
+      peopleTable.someForcedInsert(Person(name = Some("John"), hairColor = None, eyeColor = Some("Blue"))),
+      peopleTable.schema.drop
+    )
+  }
+}