我想为运行slick的服务编写一些集成测试，然后通过回滚事务来清理postgresql数据库，但是我看不到做到这一点的方法。我知道我可以测试已经组合在一起的DBIO对象并回滚它们，但是如果我想在更高的抽象水平上进行测试，这似乎是不可能的。
用伪代码，我想这样做:
StartDbTransaction() // setup
DoSomethingInDB()
AssertSomething()
RollBackDbTransaction() // teardown
例如，如果我有这个(从play-silhouette-slick-seed简化):
class PasswordInfoDAO(db: JdbcBackend#DatabaseDef) {
// ...
def remove(loginInfo: LoginInfo): Future[Unit] =
db.run(passwordInfoSubQuery(loginInfo).delete).map(_ => ())
}
我以为我可以按照Specs2 Guide的方式编写一个ForEach特性，这给出了一个通用示例:
// a transaction with the database
trait Transaction
trait DatabaseContext extends ForEach[Transaction] {
// you need to define the "foreach" method
def foreach[R: AsResult](f: Transaction => R): Result = {
val transaction = openDatabaseTransaction
try AsResult(f(transaction))
finally closeDatabaseTransaction(transaction)
}
// create and close a transaction
def openDatabaseTransaction: Transaction = ???
def closeDatabaseTransaction(t: Transaction) = ???
}
class FixtureSpecification extends mutable.Specification with DatabaseContext {
"example 1" >> { t: Transaction =>
println("use the transaction")
ok
}
"example 2" >> { t: Transaction =>
println("use it here as well")
ok
}
}
因此，为了流畅，我尝试了以下方法:
override def foreach[R: AsResult](f: JdbcBackend#DatabaseDef => R): Result = {
val db = dbConfig.db
val session = db.createSession()
session.conn.setAutoCommit(false)
val result = AsResult(f(db))
session.conn.rollback()
result
}
然后，我打算像这样使用它:
class PasswordInfoDAOSpec(implicit ee: ExecutionEnv)
extends Specification with DatabaseContext {
"password" should {
"be removed from db" in { db =>
// arrange
db.run(...) // something to set up the database
// act
PasswordInfoDAO(db).remove(loginInfo).await
// assert
PasswordInfoDAO(db).find(loginInfo) must be None.await
}
}
}
问题在于，slick 3将忽略我的 session (通过设计)，而是使用 session 池，因此我的回滚不执行任何操作。我认为Slick期望您可以在可以组合在一起并可能在不同上下文中执行的DBIOActions级别使用它。 Slick 2有一种方法可以使用
.withSession控制 session ，但已将其删除。是创建，迁移和删除每个测试的测试数据库的唯一选择吗？
最佳答案
这是部分答案。通过深入到JDBC来回滚事务似乎是不可能的，或者至少是非常不可取的。因此，我改写了存储库以返回DBIO而不是我的业务对象。 DBIO monadic绑定(bind)操作负责事务逻辑，因此，这实际上是回滚某些内容的唯一方法。
class MyRepository {
def add(whatever: String): dbio.DBIOAction[Int, NoStream, Write with Write] = {
// return a DBIOAction
}
}
我有一个将任意操作绑定(bind)到“假”异常的函数，然后返回原始操作的Future结果并丢弃该异常:
case class IntentionalRollbackException[R](successResult: R) extends Exception("Rolling back transaction")
def runWithRollback[R, S <: slick.dbio.NoStream, E <: slick.dbio.Effect](action: DBIOAction[R, S, E]): Future[R] = {
val block = action.flatMap(r => DBIO.failed(new IntentionalRollbackException(r)))
val tryResult = dbConfig.db.run(block.transactionally.asTry)
// not sure how to eliminate these casts from Any
tryResult.map {
case Failure(IntentionalRollbackException(successResult)) => successResult.asInstanceOf[R]
case Failure(t) => throw t
case Success(r) => r.asInstanceOf[R]
}
}
因此，我可以从规范中使用它:
val insertAction1 = new MyRepository().add("whatever 1").withPinnedSession
val insertAction2 = new MyRepository().add("whatever 2").withPinnedSession
val actions = insertAction1 andThen insertAction2
val result = Await.result(runWithRollback(action), 5.seconds)
result must be ...
我敢肯定，还有一种方法可以更清晰地将specs2编写为ForEach特性或类似特性。
我从this和this汲取了这些想法
关于scala - 如何回滚Slick 3 + Specs2的集成测试？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/34905455/