How to unit test code wrapped in `runOnUiThread`?

Question

I have a function that calls runOnUiThreadas below

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

I want to execute UnitTest myFunctionso that I myObjecthave a call myObjectFunction. But, given that it is completed in runOnUiThread, I cannot get to it.

How could I unit test ensure that the codes inside are runOnUiThreadcalled?

Answer 1

Determine how to run the test with ArgumentCaptor. Grab Runnablein functions runOnUiThread()and then run run throughrunOnUiArgCaptor.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()

}