Question

我正在尝试测试一些涉及OkHttp3的下载代码并且惨遭失败。目标：测试下载图像文件并验证其是否有效。平台：Android。此代码正在生产中，但测试代码没有任何意义。

产品代码

class FileDownloaderImpl internal constructor(
    private val ioScheduler: Scheduler,
    private val logger: LoggingInterceptor,
    private val parser: ((String) -> HttpUrl)? // for testing only
) : FileDownloader {

    @Inject constructor(logger: LoggingInterceptor) : this(Schedulers.io(), logger, null)

    override fun downloadFile(url: String, destination: File): Single<File> {
        Logger.d(TAG, "downloadFile\nurl = $url\nfile = $destination")

        val client = OkHttpClient.Builder()
            .addInterceptor(logger)
            .build()

        val call = client.newCall(newRequest(url))
        return Single.fromCallable { call.execute() }
            .doOnDispose { call.cancel() }
            .subscribeOn(ioScheduler)
            .map { response ->
                Logger.d(TAG, "Successfully downloaded board: $response")
                return@map response.body()!!.use { body ->
                    Okio.buffer(Okio.sink(destination)).use { sink ->
                        sink.writeAll(body.source())
                    }
                    destination
                }
            }
    }

    /**
     * Creates the request, optionally parsing the URL into an [HttpUrl]. The primary (maybe only)
     * use-case for that is for wrapping the URL in a `MockWebServer`.
     */
    private fun newRequest(url: String): Request {
        val httpUrl = parser?.invoke(url)
        val builder = Request.Builder()
        httpUrl?.let { builder.url(it) } ?: builder.url(url)
        return builder.build()
    }
}

测试代码（JUnit5）

@ExtendWith(TempDirectory::class)
internal class FileDownloaderImplTest {

    private val mockWebServer = MockWebServer()
    private val logger = LoggingInterceptor(HttpLoggingInterceptor.Level.BODY) { msg -> println(msg) }
    private val fileDownloader = FileDownloaderImpl(Schedulers.trampoline(), logger) {
        mockWebServer.url("/$it")
    }

    @BeforeEach fun setup() {
        mockWebServer.start()
    }

    @AfterEach fun teardown() {
        mockWebServer.shutdown()
    }

    @Test fun downloadFile(@TempDir tempDirectory: Path) {
        // Given
        val res = javaClass.classLoader.getResource("green20.webp")
        val f = File(res.path)
        val buffer = Okio.buffer(Okio.source(f)).buffer()
        mockWebServer.enqueue(MockResponse().setBody(buffer))
        val destFile = tempDirectory.resolve("temp.webp").toFile()

        // Verify initial condition
        destFile.exists() shouldBe false

        // When
        fileDownloader.downloadFile("test.html", destFile)

            // Then
            .test()
            .assertValue { file ->
                file.exists() shouldBe true
                file.length() shouldEqualTo 66 // FAIL: always 0
                true
            }
    }
}

更多细节

＆＃34; green20.webp＆＃34;是app/test/resources中存在的文件。当我调试时，所有迹象都表明它存在。关于调试的问题，我在prod代码中有断点，看起来像Response对象（可能是MockResponse）没有正文。我不知道为什么会这样。

当前的想法：

我没有正确添加模拟回复正文
该文件以某种方式打开＆＃34;所以它的长度总是为0，即使它实际上并不是空的。

修改

我尝试从测试中移除MockWebServer并启动了真正的下载，我的测试确实通过了。所以，我认为我对MockResponse及其身体做错了。任何帮助将不胜感激。

Answer 1

由于我不清楚的原因，Okio.buffer(Okio.source(file)).buffer()始终返回空Buffer。但是，以下内容有效：

mockWebServer.enqueue(MockResponse().setBody(Buffer().apply {
    writeAll(Okio.source(file))
}))

我现在正在做的是手动创建一个新缓冲区并将整个文件写入其中。现在我的MockResponse有一个真实的身体。

我仍然希望有人能解释为什么这个......

Answer 2

buffer()上的BufferedSource方法不会将整个流读入该缓冲区以返回它。相反，它只是让你访问它从文件中预加载的字节，这将在下次读取时返回。

这是将文件加载到缓冲区的代码：

val buffer = Buffer()
file.source().use {
  buffer.writeAll(it)
}

使用MockWebServer和MockResponse使用Buffer主体测试OkHttp

2 个答案: