unit-testing - 仅测试文件中的掩码包名称

Question

为了追求 100% 的单元测试覆盖率，我们尝试在一个函数中测试多行代码。相关函数调用运行时包:

// functionName returns a string representing the function name of the function n stack frames above the caller.
// if n = 0, the name of the function calling functionName() will be returned.
func functionName(n int) string {
    pc, _, _, ok := runtime.Caller(n + 1)
    if !ok {
        return "unknown function"
    }
    me := runtime.FuncForPC(pc)
    if me == nil {
        return "unknown function"
    }

    split := strings.Split(me.Name(), ".")
    if len(split) == 0 {
        return "unknown function"
    }
    return split[len(split)-1]
}

具体来说，3 个 if 语句及其返回值目前尚未经过测试，因为运行时函数似乎不容易被操纵以返回我们想要的值。在这些情况下，我们的标准响应是模拟出有问题的项目，但这些调用是对运行时包本身内的包级函数(而不是接口(interface)的方法)的调用。

我的第一个想法是通过使用带有 Caller() 和 FuncForPC() 方法的结构来模拟运行时 token 本身，分配给测试文件中名为“runtime”的变量(因此它不会影响生产代码流，因为在正常构建期间省略了测试文件)。但是，这会触发有关在(全局) block 中重新声明“运行时”的构建错误。

我知道如果“运行时”变量在非全局范围 (example masking fmt) 中声明，这将是可能的，但我找不到一种优雅的方式来做到这一点，以便它在测试中被屏蔽，但不在生产代码本身内。我想到的唯一方法是通过更改生产代码的源代码来声明这样一个变量并替换它在测试中的值，但这远非理想，因为它使生产代码复杂化纯粹是为了测试目的。

有什么想法吗？

Answer 1

一种解决方案是声明您要模拟的那些函数的变量。

var runtimeCaller = runtime.Caller
var runtimeFuncForPC = runtime.FuncForPC

func functionName(n int) string {
    pc, _, _, ok := runtimeCaller(n + 1)
    if !ok {
        return "unknown function"
    }
    me := runtimeFuncForPC(pc)
    if me == nil {
        return "unknown function"
    }

    split := strings.Split(me.Name(), ".")
    if len(split) == 0 {
        return "unknown function"
    }
    return split[len(split)-1]
}

或者如果您更喜欢点符号...

var _runtime = struct{
    Caller    func(skip int) (pc uintptr, file string, line int, ok bool)
    FuncForPC func(pc uintptr) *runtime.Func
}{runtime.Caller, runtime.FuncForPC}

func functionName(n int) string {
    pc, _, _, ok := _runtime.Caller(n + 1)
    if !ok {
        return "unknown function"
    }
    me := _runtime.FuncForPC(pc)
    if me == nil {
        return "unknown function"
    }

    split := strings.Split(me.Name(), ".")
    if len(split) == 0 {
        return "unknown function"
    }
    return split[len(split)-1]
}

在您的测试中，在运行 functionName 之前，您可以将变量/字段设置为模拟实现。如果其他测试可能导致 functionName 被调用，请注意并发访问...我认为如果不显着更改现有代码，您将无能为力。