0

我在以下表格上有一些代码：

@Language("SQL")
val someSql = """
    SELECT foo
    FROM bar 
    WHERE foo = :foo
    """
return session.select(some, mapOf("foo" to foo)) {
    MyObject(
            foo = it.string("foo"),
    )
}.firstOrNull()

它使用来自com.github.andrewoma.kwery.core的以下内容。注意方法签名中的 lambda：

fun <R> select(@Language("SQL") sql: String,
               parameters: Map<String, Any?> = mapOf(),
               options: StatementOptions = defaultOptions,
               mapper: (Row) -> R): List<R>

我使用 mockitokotlin2。

当使用选择查询（包含“SELECT foo”）调用会话选择方法时，我需要返回 MyObject 的实例。

我在想我可以将一个模拟传递给 lambda，如下所示（但它与我试图模拟的方法调用不匹配）。下面的代码是一个尝试。但它从不匹配eq(function2)：

val function2: (Row) -> Int = mock {
    onGeneric { invoke(any()) }.thenReturn(MyObject(foo="test-foo"))
}

val session = mock<Session> {
    on { select(sql = any(), parameters = any(), options = any(), mapper = eq(function2))}.thenReturn(listOf(MyObject(foo="test-foo")))
}

在我的情况下，function2 并不是真正的映射器，它与我要模拟的内容不相等，它永远不会匹配并且永远不会调用模拟。

那么在上面的代码中，我应该在 mock 中session, select而不是eq(function2)中放置什么来返回 MyObject 对象？

4

1 回答 1

1

我认为您只需要指定他们在设置会话模拟时预期您的映射器返回的类型 - 在您的情况下看起来是Function1<Row, MyObject>

val session = mock<Session> {
    on { select(sql = anyString(), parameters = anyMap(), options = any(), mapper = any<Function1<Row, MyObject>>())}.thenReturn(listOf(MyObject(foo="test-foo")))
}
于 2018-09-19T09:42:49.190 回答