Mockitoでlambda引数をStubしたい

val consumeMock: Consume<String> = mock(Consume::class.java) as Consume<String>
// Define behavior for consume method that takes a String parameter
doAnswer { invocation ->
    val data = invocation.arguments[0] as String
    // Perform desired actions with the data
    println("Consume data: $data")
}.`when`(consumeMock).consume(anyString())

consumeMock.consume("Sample data")

verify(consumeMock).consume("Sample data")