如何编写单元测试验证runWorkflow函数的失败重试逻辑？

问题背景 #

现有独立文件中的顶层函数runWorkflow：

fun runWorkflow(data: Promise<Data>) {
    withRetries(data, data.get().timeout) {
        runAction(it.get(), client)
    }
}

其中withRetries是外部库函数，会重复执行内部代码块直到data.get().timeout分钟超时；client为静态导入。

另有独立文件中的顶层内部函数runAction：

internal fun runAction(data: Data, client: Client): Promise<Data> {
   doSomethingWithClientOrData()
   return somethingElse()
}

需要验证调用runWorkflow时的重试逻辑，但无法mock或验证withRetries的调用。尝试过mockrunAction让其先失败数次再成功，但无法区分是单次成功还是重试后成功，也不确定如何正确验证重试逻辑。

解决方案 #

1. 通过统计runAction调用次数间接验证重试 #

不用直接验证withRetries，转而通过统计runAction的调用次数来确认重试行为：

• 声明一个可变计数器记录调用次数
• 让mock的runAction按预期先抛出异常，达到指定次数后返回成功，同时每次调用递增计数器

示例（以Mockk框架为例）：

val mockClient = mockk<Client>()
var callCount = 0

// Mock顶层internal函数runAction，前2次失败，第3次成功
mockkStatic("your.package.name.RunActionKt") // 替换为runAction所在的实际包名
every { runAction(any(), mockClient) } answers {
    callCount++
    if (callCount <= 2) {
        Promise.error(RetryableException("模拟失败"))
    } else {
        Promise.success(mockk<Data>())
    }
}

// 执行测试逻辑
runWorkflow(Promise.success(mockk<Data> {
    every { timeout } returns 5 // 设置足够长的超时时间，确保重试能完成
}))

// 验证调用次数符合预期（2次重试+1次成功，共3次）
assertEquals(3, callCount)

2. 解决internal函数的测试访问问题 #

因为runAction是internal函数，测试模块默认无法直接访问，可通过两种方式处理：

• 在runAction所在模块的build.gradle.kts中配置测试模块可见性：

kotlin {
    sourceSets {
        named("test") {
            // 允许测试模块访问主模块的internal成员
            dependsOn(sourceSets["main"])
        }
    }
}

• 直接使用Mockk的mockkStatic方法，通过函数全限定名mock顶层internal函数，无需额外配置

3. 补充验证点 #

• 验证runWorkflow最终返回成功结果，确认重试后逻辑正常完成
• 若需验证超时逻辑，可设置极短的超时时间，让runAction持续失败，验证是否在超时后抛出对应异常

内容的提问来源于stack exchange，提问作者Kevin2566