PrimaryKeyTest.scala

package com.typesafe.slick.testkit.tests

import com.typesafe.slick.testkit.util.{RelationalTestDB, AsyncTest}

class PrimaryKeyTest extends AsyncTest[RelationalTestDB] {
  import tdb.profile.api._

  def test = {

    class A(tag: Tag) extends Table[(Int, Int, String)](tag, "a") {
      def k1 = column[Int]("k1")
      def k2 = column[Int]("k2")
      def s = column[String]("s")
      def * = (k1, k2, s)
      def pk = primaryKey("pk_a", (k1, k2))
    }
    val as = TableQuery[A]

    as.baseTableRow.primaryKeys.map(_.name).toSet shouldBe Set("pk_a")

    seq(
      as.schema.create,
      as ++= Seq(
        (1, 1, "a11"),
        (1, 2, "a12"),
        (2, 1, "a21"),
        (2, 2, "a22")
      ),
      (as += (1, 1, "a11-conflict")).failed,
      as.schema.drop
    )
  }
}