如何使用嵌套的悬浮函数对kotlin协程进行单元测试

时间:2018-04-06 16:40:42

标签: android unit-testing kotlin coroutine kotlinx.coroutines

下面是我试图进行单元测试的一段代码。 async(CommonPool)第一个挂起函数之前的代码是可测试的,但在此之后,测试仍然失败。我尝试使用runBlocking但无法测试嵌套的挂起异步函数。

interface Listener {
        fun onLoading(user: User?)

        fun onSuccess(user: User)
}

execute(listener: Listener) {

        listener.onLoading(null)
        val service = UserService.getInstance(context, "someurl")
        val database = UserDatabase.getInstance(context)

        launch(UI) {

            val user = async(CommonPool) {
                userDatabase.getUser()
            }.await()

            listener.onLoading(user)

            val response = service.getUsersSelf(oauthToken).await()

            async(CommonPool) {database.saveUser(userResponse.data.user)}.await()

            val user = async(CommonPool) {database.getUser()}.await()

            listener.onSuccess(user)
        }
    }

下面是我的单元测试,我使用mockito来模拟我的监听器并检查交互次数。

@Test
    fun execute() {
        runBlocking {
            userDatabase.saveUser(user)

            val listener = mock(GetUser.Listener::class.java)

            getUser.execute(listener)

            verify(listener, times(1)).onLoading(null) // Success

            verify(listener, times(1)).onLoading(user) // Fails

            verify(listener, times(1)).onSuccess(user) // Fails
        }
    }

但是最后两次verify测试失败了。任何人都可以帮我测试吗?

2 个答案:

答案 0 :(得分:1)

你的execute是一种即发即弃的功能。如果这是通过线程完成的,它将是Thread { <code> }.start()并完全放弃对线程的引用。您唯一剩下的就是Listener实例,它会在调用完成时收到通知 - 但不会出现错误。

虽然我怀疑这是一个很好的方法来使用协同程序,它们都是关于消除回调接口,如Listener,如果您知道这种方法对您有好处,那么听众是你发起的协程事件的唯一处理方式。

因此,您必须在侦听器中实现单元测试逻辑。如果没有失败回调,您可能需要等待一段有限的时间来调用方法。一种方法:

fun execute() {
    val latch = CountDownLatch(3)
    runBlocking {
        val x = object : Listener {

            override fun onLoading(user: User?) {
                if (user == null) {
                    require(latch.count == 3L)
                    latch.countDown()
                } else {
                    require(latch.count == 2L)
                    latch.countDown()
                }
            }

            override fun onSuccess(user: User) {
                require (latch.count == 1L)
                latch.countDown()
            }
        }
    }
    require(latch.await(10, TimeUnit.SECONDS))
}

答案 1 :(得分:0)

您可以在测试时使用依赖项注入将Dispatchers.Unconfined替换为UI Dispatcher。该测试成功:

@Test
fun test() {
    var result = false
    GlobalScope.launch (Dispatchers.Unconfined) {
        async(Dispatchers.Unconfined) {
            sleep(1000)
            result = true
        }.await()
    }
    assertTrue(result)
}