使用MockWebServer进行暂停功能测试

时间:2020-06-02 21:29:38

标签: android unit-testing kotlin-coroutines mockwebserver

我正在测试使用MockWebServer的暂停功能返回结果的api,但是除非使用launch构建器,否则它不能与runBlockingTest,testCoroutineDispatcher,testCorounieScope一起使用,为什么?

abstract class AbstractPostApiTest {

    internal lateinit var mockWebServer: MockWebServer

    private val responseAsString by lazy {
        getResourceAsText(RESPONSE_JSON_PATH)
    }

    @BeforeEach
    open fun setUp() {
        mockWebServer = MockWebServer()
        println("AbstractPostApiTest setUp() $mockWebServer")
    }


    @AfterEach
    open fun tearDown() {
        mockWebServer.shutdown()
    }

    companion object {
        const val RESPONSE_JSON_PATH = "posts.json"
    }


    @Throws(IOException::class)
    fun enqueueResponse(
        code: Int = 200,
        headers: Map<String, String>? = null
    ): MockResponse {

        // Define mock response
        val mockResponse = MockResponse()
        // Set response code
        mockResponse.setResponseCode(code)

        // Set headers
        headers?.let {
            for ((key, value) in it) {
                mockResponse.addHeader(key, value)
            }
        }

        // Set body
        mockWebServer.enqueue(
            mockResponse.setBody(responseAsString)
        )

        return mockResponse
    }


}


class PostApiTest : AbstractPostApiTest() {

    private lateinit var postApi: PostApiCoroutines

    private val testCoroutineDispatcher = TestCoroutineDispatcher()

    private val testCoroutineScope = TestCoroutineScope(testCoroutineDispatcher)

    @BeforeEach
    override fun setUp() {
        super.setUp()

        val okHttpClient = OkHttpClient
            .Builder()
            .build()

        postApi = Retrofit.Builder()
            .baseUrl(mockWebServer.url("/"))
            .addConverterFactory(GsonConverterFactory.create())
            .client(okHttpClient)
            .build()
            .create(PostApiCoroutines::class.java)


        Dispatchers.setMain(testCoroutineDispatcher)

    }

    @AfterEach
    override fun tearDown() {
        super.tearDown()

        Dispatchers.resetMain()
        try {
            testCoroutineScope.cleanupTestCoroutines()
        } catch (exception: Exception) {
            exception.printStackTrace()
        }
    }

    @Test
    fun `Given we have a valid request, should be done to correct url`() =
        testCoroutineScope.runBlockingTest {

            // GIVEN
            enqueueResponse(200, RESPONSE_JSON_PATH)

            // WHEN
              postApi.getPostsResponse()

            advanceUntilIdle()

            val request = mockWebServer.takeRequest()

            // THEN
            Truth.assertThat(request.path).isEqualTo("/posts")

        }
}

结果错误:java.lang.IllegalStateException: This job has not completed yet

如果使用launch构建器,则此测试不起作用;如果使用launch构建器,则不需要testCoroutineDispatchertestCoroutineScope,这是什么原因?通常,即使使用runBlockingTest

,挂起函数也不会在其他范围内传递
 @Test
    fun `Given we have a valid request, should be done to correct url`() =
        runBlockingTest {

            // GIVEN
            enqueueResponse(200, RESPONSE_JSON_PATH)

            // WHEN
            launch {
                postApi.getPosts()
            }

            val request = mockWebServer.takeRequest()

            // THEN
            Truth.assertThat(request.path).isEqualTo("/posts")

        }

以上一种有效。

下面的测试有时也会通过。

@测试 有趣的Given api return 200, should have list of posts()= testCoroutineScope.runBlockingTest {

    // GIVEN
    enqueueResponse(200)

    // WHEN
    var posts: List<Post> = emptyList()
    launch {
        posts = postApi.getPosts()
    }

    advanceUntilIdle()

    // THEN
    Truth.assertThat(posts).isNotNull()
    Truth.assertThat(posts.size).isEqualTo(100)

}

我尝试了很多使用posts = postApi.getPosts()调用launch而不使用async的组合,将enqueueResponse(200)放入异步async { enqueueResponse(200) }.await()的情况,但是测试失败了,有时它通过了每种组合都不一样。

1 个答案:

答案 0 :(得分:1)

runBlockTest 存在一个错误,即在完成运行测试的协程之前不等待其他线程/作业完成。

我尝试使用 runBlocking 并成功(我使用了从 Hamcrest 到 Kotlin Hamkrest 的很棒的移植)

fun `run test` = runBlocking {
  mockWebServer.enqueue(MockResponse().setResponseCode(200).setBody(""))

  // make HTTP call 

  val result = mockWebServer.takeRequest(2000L, TimeUnit.MILLISECONDS)

  assertThat(result != null, equalTo(true))
}

这里有几点需要注意:

  1. 线程阻塞调用的使用应该绝不在没有超时的情况下被调用。最好是一无所获,然后永远阻塞一个线程。

  2. 使用 runBlocking 可能被某些人认为是不。然而,this blog post 概述了运行并发代码的不同方法,以及它们的不同用例。我们通常希望使用 runBlockingTest 或 (TestCoroutineDispatcher.runBlockingTest) 以便我们的测试代码和应用程​​序代码同步。通过使用相同的 Dispatcher,我们可以确保所有作业都完成,等等。TestCoroutineDispatcher 还具有方便的“时钟”功能,可以消除延迟。但是,在测试应用程序的 HTTP 层时,如果模拟服务器运行在单独的线程上,我们的同步点为 takeRequest。因此,我们可以愉快地使用 runBlocking 来允许我们使用协程,并且运行在不同线程上的模拟服务器可以毫无问题地协同工作。