blob: aaf38a0f269f6d6e8fc0cf3d44956803a7b4c302 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
package com.pitchedapps.frost
import com.raizlabs.android.dbflow.config.DatabaseConfig
import com.raizlabs.android.dbflow.config.FlowConfig
import com.raizlabs.android.dbflow.config.FlowManager
import org.junit.rules.TestRule
import org.junit.runner.Description
import org.junit.runners.model.Statement
import org.robolectric.RuntimeEnvironment
/**
* Created by Allan Wang on 2017-05-30.
*/
class DBFlowTestRule : TestRule {
override fun apply(base: Statement, description: Description): Statement {
return object : Statement() {
@Throws(Throwable::class)
override fun evaluate() {
FlowManager.init(FlowConfig.Builder(RuntimeEnvironment.application)
.addDatabaseConfig(DatabaseConfig.Builder(TestDatabase::class.java)
.transactionManagerCreator(::ImmediateTransactionManager)
.build()).build())
try {
base.evaluate()
} finally {
FlowManager.destroy()
}
}
}
}
companion object {
fun create() = DBFlowTestRule()
}
}
|