使用Room @Transaction函数

时间:2019-05-30 14:26:56

标签: android kotlin android-room android-architecture-components kotlin-coroutines

我正在使用AS 3.4.1和运行Android 9的模拟器进行测试。

当我在其中使用Room Dao Function annotated with @Transaction时,以下测试将无法运行。

class RecurrenceManagerTest : DatabaseTest() {

    @Rule
    @JvmField
    val instantTaskExecutorRule = InstantTaskExecutorRule()

    var recurringEntryId: Long = -1

    @Before
    override fun setup() {
        super.setup() // only initialized the db

        val recurringEntry = RecurringEntry(
            recurrence = Recurrence(DATE.toEpochMilli(), Recurrence.DAILY)
        )

        recurringEntryId = runBlocking { db.recurringEntryDao().insert(recurringEntry) }

        val recurringBookEntry = BookEntry.create(
            title = TITLE,
            date = DATE,
            value = VALUE,
            isPaid = IS_PAID,
            notes = NOTES,
            entryType = ENTRY_TYPE,
            categoryId = CATEGORY_ID,
            contacts = CONTACTS,
            recurringEntryId = recurringEntryId
        )

        runBlocking {
            db.bookEntryDao().insert(recurringBookEntry) // BreakPoint #1
        }
    }

    @Test
    fun testInsertRecurrencesAndSchedule() { 
        var recurringEntry = runBlocking { db.recurringEntryDao().get(recurringEntryId) } // BreakPoint #2

        assertThat(recurringEntry, notNullValue())

        runBlocking { RecurrenceManager.insertRecurrencesAndSchedule(ApplicationProvider.getApplicationContext(), db, recurringEntry!!) }

        val bookEntries = db.bookEntryDao().getBookEntries().liveDataValue()
    }
}

这是用于插入的功能:

@Transaction
suspend fun insert(bookEntry: BookEntry): Long {
    val id = insert(bookEntry.entity)
    bookEntry.embeddedContacts?.apply {
        forEach {
            it.id = 0
            it.bookEntryId = id
        }
    }?.let {
        insert(it)
    }

    return id
}

因此,如果我按原样运行测试(请参见BreakPoint#1),则Breakpoint#2甚至都不会被调用,因此测试在某个地方结束,没有结果。

如果我要用完全相同的代码替换BreakPoint#1处的代码,则insert函数具有该功能,则测试将正确运行。

有人知道这里有什么问题吗?

2 个答案:

答案 0 :(得分:5)

您可以使用 setTransactionExecutor 在另一个线程中运行事务

return Room
       .inMemoryDatabaseBuilder(context, MyRoomDatabase::class.java)
       .setTransactionExecutor(Executors.newSingleThreadExecutor())
       .build()

然后在测试时使用 runBlocking 而不是runBlockingTest

@Test
fun moveItem() = runBlocking {
    transactionFunction()
}

答案 1 :(得分:1)

我遇到了同样的问题,问题出在InstantTaskExecutorRule,如果您删除下面的代码块,@Transaction应该可以与suspend关键字一起找到

@Rule
@JvmField
val instantTaskExecutorRule = InstantTaskExecutorRule()

该规则似乎阻止RoomDatabase获取事务线程。在RoomDatabase.kt中,它在以下函数中受阻:

private suspend fun Executor.acquireTransactionThread(controlJob: Job): ContinuationInterceptor

希望这会有所帮助!