Question

我希望将Slick 3框架用于Scala应用程序来管理数据库交互。我已经能够使用Slick自动生成必要的表对象，但是我还想要一个集成测试来验证数据库中的模式是否与对象中的模式匹配。这是因为有时表得到在没有我的团队被警告的情况下进行了更改，因此我们更愿意在集成测试而不是生产应用程序中捕获更改。

执行此操作的一种方法是在测试运行器中的每个表上运行select查询。但是，我觉得应该有更直接的方式。此外，我不清楚如何系统地运行文件中定义的所有表，除了手动将表对象附加到测试运行器通过的某个序列。我注意到有一个架构字段，但它只能生成create和drop语句。

非常感谢任何帮助。谢谢！

编辑：

这是我的解决方案，但我希望有一个更好的解决方案：

class TablesIT extends FunSuite with BeforeAndAfter with ScalaFutures {
  var db: Database = _
  before{ db = Database.forURL( /* personal details */ )}

  object ResultMap extends GetResult[Map[String,Any]] { //Object borrowed from http://stackoverflow.com/questions/20262036/slick-query-multiple-tables-databases-with-getting-column-names
    def apply(pr: PositionedResult) = {
      val rs = pr.rs // <- jdbc result set
      val md = rs.getMetaData
      val res = (1 to pr.numColumns).map{ i=> md.getColumnName(i) -> rs.getObject(i) }.toMap
      pr.nextRow // <- use Slick's advance method to avoid endless loop
      res
    }
  }

  def testTableHasCols[A <: Table[_]](table: slick.lifted.TableQuery[A]): Unit = {
    whenReady(db.run(table.take(1).result.headOption.asTry)) { case Success(t) => t match {
      case Some(r) => logTrace(r.toString)
      case None => logTrace("Empty table")
    }
    case Failure(ex) => fail("Query exception: " + ex.toString)
    }
  }

  def plainSqlSelect[A](query: String)(implicit gr: GetResult[A]): Future[Seq[A]] = {
    val stmt = sql"""#$query""".as[A]
    db.run(stmt)
  }


  def compareNumOfCols[A <: Table[_]](table: slick.lifted.TableQuery[A]) = {
    val tableName = table.baseTableRow.tableName
    val selectStar = whenReady(db.run(sql"""select * from #$tableName limit 1""".as(ResultMap).headOption)) {
      case Some(m) => m.size
      case None => 0
    }
    val model = whenReady(db.run(sql"""#${table.take(1).result.statements.head}""".as(ResultMap).headOption)) {
      case Some(m) => m.size
      case None => 0
    }
    assert(selectStar === model, "The number of columns do not match")
  }

  test("Test table1") {
    testTableHasCols(Table1)
    compareNumOfCols(Table1)
  }

// And on for each table
}

Answer 1

我最终设计了一个使用以下想法的更好的解决方案。它或多或少是相同的，不幸的是我仍然需要为每个表手动创建一个测试，但我认为该方法更清晰。但请注意，由于信息模式，这仅适用于PostgreSQL，但其他数据库系统也有其他方法。

class TablesIT extends FunSuite with BeforeAndAfter with ScalaFutures {
  var db: Database = _
  before{ db = Database.forURL( /* personal details */ )}

  def testTableHasCols[A <: Table[_]](table: slick.lifted.TableQuery[A]): Unit = {
    whenReady(db.run(table.take(1).result.headOption.asTry)) { case Success(t) => t match {
      case Some(r) => logTrace(r.toString)
      case None => logTrace("Empty table")
    }
    case Failure(ex) => fail("Query exception: " + ex.toString)
    }
  }

  def compareNumOfCols[A <: Table[_]](table: slick.lifted.TableQuery[A]) = {
    val tableName = table.baseTableRow.tableName
    val selectStar = whenReady(db.run(sql"""select column_name from information_schema.columns where table_name='#$tableName'""".as[String])) {
      case m: Seq[String] => m.size
      case _ => 0
    }
    val model = table.baseTableRow.create_*.map(_.name).toSeq.size

    assert(selectStar === model, "The number of columns do not match")
  }

  test("Test table1") {
    testTableHasCols(Table1)
    compareNumOfCols(Table1)
  }

// And on for each table
}

使用Slick 3验证架构

1 个答案: