如何对包含`runOnUiThread`的代码进行单元测试？

Question

我有一个函数，如下所示调用runOnUiThread

fun myFunction(myObject: MyClass, view: MyView) {
    // Do something
    view.getActivity().runOnUiThread {
        myObject.myObjectFunction()
    }
}

我希望进行单元测试以确保myObject已调用myObjectFunction。但是由于它被包装在runOnUiThread中，我无法获取它。

如何进行单元测试以确保runOnUiThread中的代码已被调用？

- Elye

Answer 1

成功找到了一种使用ArgumentCaptor执行测试的方法。在runOnUiThread()函数中捕获Runnable，然后通过runOnUiArgCaptor.value.run()触发运行。

import com.nhaarman.mockito_kotlin.argumentCaptor
import com.nhaarman.mockito_kotlin.verify
import com.nhaarman.mockito_kotlin.whenever
import org.junit.Before
import org.junit.Test
import org.mockito.Mock

@Mock lateinit var activity: Activity
@Mock lateinit var view: MyView
@Mock lateinit var myObject: MyObject

@Before
fun setUp() {
    MockitoAnnotations.initMocks(this)
}

@Test
fun my_test_function() {
    whenever(view.getActivity()).thenReturn(activity)

    val runOnUiArgCaptor = argumentCaptor<Runnable>()
    val myTestObject = TestObject()

    myTestObject.myFunction(myObject, view)
    verify(activity).runOnUiThread(runOnUiArgCaptor.capture())
    runOnUiArgCaptor.value.run()

    verify(myObject).myObjectFunction()

}

- Elye