我想为运行流畅的服务编写一些集成测试,然后通过回滚事务来清理 postgresql 数据库,但我没有找到一种方法来做到这一点。我知道我可以测试组合在一起的 DBIO 对象并将它们回滚,但如果我想在更高的抽象级别进行测试,这似乎是不可能的。
在伪代码中,我想这样做:
StartDbTransaction() // setup
DoSomethingInDB()
AssertSomething()
RollBackDbTransaction() // teardown
例如,如果我有这个(从玩剪影光滑种子):
class PasswordInfoDAO(db: JdbcBackend#DatabaseDef) {
// ...
def remove(loginInfo: LoginInfo): Future[Unit] =
db.run(passwordInfoSubQuery(loginInfo).delete).map(_ => ())
}
我想我可以按照以下方式编写一个 ForEach 特征规格2指南,这给出了一个通用示例:
// 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 将忽略我的会话(按照设计)并使用会话池,因此我的回滚不会执行任何操作。我认为 Slick 期望您应该在 DBIOActions 级别使用它,这些 DBIOActions 可以组合在一起,并且可能在不同的上下文中执行。 Slick 2 有一种方法可以控制会话.withSession
,但已被删除。
是在每次测试时创建、迁移和删除测试数据库的唯一选择吗?