mockk 验证 lambda 是在模拟中传递的

Question

我正在尝试测试该方法 getSongsList

class SongsRemoteDataSource @Inject constructor(
    private val resultParser: ResultParser,
    private val songsService: SongsService
) {

    suspend fun getSongsList(query: String): Result<SongsResponse> =
        resultParser.parse { songsService.getSongsList(query) }
}

基本上我正在尝试测试是否使用正确的 lambda 作为参数调用了模拟

class SongsRemoteDataSourceTest {

    @RelaxedMockK
    private lateinit var resultParser: ResultParser
    @RelaxedMockK
    private lateinit var songsService: SongsService

    private lateinit var songsRemoteDataSource: SongsRemoteDataSource

    @Before
    fun setUp() {
        MockKAnnotations.init(this)
        songsRemoteDataSource = SongsRemoteDataSource(resultParser, songsService)
    }

    @Test
    fun getSongsList() = runBlockingTest {
        val query = "query"

        songsRemoteDataSource.getSongsList(query)

        coVerify { resultParser.parse { songsService.getSongsList(query) } }
    }
}

但是测试失败

java.lang.AssertionError: Verification failed: call 1 of 1: ResultParser(resultParser#1).parse(eq(continuation {}), any())). Only one matching call to ResultParser(resultParser#1)/parse(Function1, Continuation) happened, but arguments are not matching:
[0]: argument: continuation {}, matcher: eq(continuation {}), result: -
[1]: argument: continuation {}, matcher: any(), result: +

结果解析器

class ResultParser @Inject constructor() {

    suspend fun <T> parse(call: suspend () -> Response<T>): Result<T> {
        ...
    }
}

歌曲服务

interface SongsService {

    @GET("search")
    suspend fun getSongsList(
        @Query("term") query: String,
        @Query("mediaType") mediaType: String = "music"
    ): Response<SongsResponse>
}

我不明白为什么它失败了。我究竟做错了什么？

Answer 1

函数总是通过身份比较

该语言不知道如何将一个函数的内容与另一个函数的内容进行比较。在不同地方创建的两个 lambda 函数，即使它们做完全相同的事情，也不被视为彼此相等。

你可以用一个简单的例子来证明这一点：

val a = { "Hello, World!" }
val b = { "Hello, World!" }
println(a == b) // prints 'false'

您的verify调用失败了，因为您实际上有两个不同的 lambda 函数，即使它们包含相同的代码。在 中创建的 lambda 与在 中创建的 lambdaSongsRemoteDataSource是不同的对象SongsRemoteDataSourceTest，因此 MockK 认为它们彼此不相等。

测试 lambda 函数内容的唯一真正方法是运行它并查看它的作用。为此，您有几个选择。

使用answers运行lambda函数

解决此问题的一种方法是将模拟配置ResultParser为始终运行它收到的 lambda 函数。

coEvery { 
    resultParser.parse(any()) 
} coAnswers { 
    firstArg<(suspend () -> Response<SongsResponse>)>().invoke() 
}

现在，每次ResultParser调用 时，它都会立即运行它作为输入接收到的任何 lambda 函数。然后，在您调用 之后getSongsList，您可以验证 是否SongsService被调用。

songsRemoteDataSource.getSongsList(query)
coVerify { 
    resultParser.parse(any())
    songsService.getSongsList(query)
}

捕获 lambda 函数并自行运行

如果您想更明确，可以改为捕获 lambda。这让您可以将 lambda 函数分配给一个变量，您可以在其中执行您想要的操作。您仍然无法比较它的相等性，但您仍然可以运行它并测试它的功能。

首先，您创建一个slot将保存捕获的函数。然后你使用coVerify插槽作为参数匹配器。

val lambdaSlot = slot<(suspend () -> Response<SongsResponse>)>()

songsRemoteDataSource.getSongsList(query)
coVerify { resultParser.parse(capture(lambdaSlot)) }

lambdaSlot.captured.invoke() // runs the lambda function
coVerify { songsService.getSongsList(query) }