Golang如何捕获panic异常？

在 Golang 单元测试中，验证函数是否按预期触发 panic 异常至关重要。由于 Go 的 `testing` 包本身不直接支持 panic 期望，本文介绍如何利用 `defer` 和 `recover` 机制来捕获并验证 panic。首先，在测试函数中使用 `defer` 语句调用 `recover()`，以便在 panic 发生时进行捕获。通过检查 `recover()` 的返回值或设置标志位，可以判断是否发生了 panic。为了简化重复逻辑，可以封装辅助函数 `assertPanic`。更进一步，可以扩展封装函数，例如 `assertPanicWithMessage`，以验证 panic 的具体错误信息是否符合预期。掌握这些技巧，能够有效地测试 Go 代码中的 panic 行为，保证代码的健壮性和可靠性。

要验证Go函数中是否触发panic，可使用defer和recover机制捕获并验证。具体步骤如下：1. 在测试函数中定义defer语句，并在其中调用recover()以捕获panic；2. 设置标志位或直接判断recover返回值是否为nil来确认是否发生panic；3. 若需多次使用，可封装辅助函数assertPanic简化逻辑；4. 如需验证panic内容，可进一步扩展封装函数以检查具体的错误信息。通过这种方式，可以有效测试预期的panic行为，同时保持代码简洁和可维护性。

在写 Go 代码时，有些函数设计上会在特定错误场景触发 panic，比如参数非法、状态异常等情况。这时候我们在做单元测试的时候，就需要验证这些 panic 是否如期发生。Golang 的 testing 包本身没有直接支持“期望 panic”的机制，但我们可以结合 defer 和 recover 来实现。

如何手动捕获 panic 并验证

Go 中的 panic 是可以被 recover 捕获的，只要在 defer 函数中调用 recover()。我们可以在测试函数中使用这个机制来判断是否发生了 panic。

举个例子：

func TestSomeFunctionThatPanic(t *testing.T) {
    var didPanic bool

    defer func() {
        if r := recover(); r != nil {
            didPanic = true
        }
    }()

    someFunctionThatShouldPanic() // 调用会 panic 的函数

    if !didPanic {
        t.Errorf("Expected panic but did not happen")
    }
}

这种方式虽然简单，但非常实用。你只需要确保 defer 函数在测试函数执行完后能检查到是否发生了 panic。

注意点：

• 如果函数中有多个可能 panic 的地方，你需要根据具体输入控制流程。
• recover 只能在 defer 函数中生效，不能放在函数体里直接调用。
• 如果你不处理 panic，测试会直接失败，所以一定要包裹 recover。

使用封装函数简化重复逻辑

如果你有多个测试用例需要验证 panic，每次都写一遍 defer 和 recover 就显得啰嗦了。可以封装一个辅助函数或者宏函数来统一处理。

示例封装：

func assertPanic(t *testing.T, f func()) {
    defer func() {
        if r := recover(); r == nil {
            t.Errorf("Expected panic but did not occur")
        }
    }()
    f()
}

使用方式：

func TestMyPanicFunc(t *testing.T) {
    assertPanic(t, func() {
        someFunctionThatShouldPanic()
    })
}

这样不仅让测试代码更简洁，也更容易维护和扩展，比如你可以加上对 panic 内容的检查。

验证 panic 的内容信息（可选）

有时候我们不仅想知道有没有 panic，还想确认 panic 的内容是否符合预期。这时可以稍微改进上面的封装函数：

func assertPanicWithMessage(t *testing.T, expected string, f func()) {
    defer func() {
        if r := recover(); r != nil {
            if msg, ok := r.(string); ok {
                if msg != expected {
                    t.Errorf("Expected panic message: %q, got: %q", expected, msg)
                }
            } else {
                t.Errorf("Panic value is not a string")
            }
        } else {
            t.Errorf("Expected panic but did not occur")
        }
    }()
    f()
}

然后测试：

assertPanicWithMessage(t, "invalid input", func() {
    myFunc(nil)
})

这种做法适合那些通过 panic 抛出明确错误信息的场景，比如一些配置校验或断言工具函数。

基本上就这些。用好 defer + recover，再配合 testing 包，就能很好地覆盖 panic 场景的测试。不需要太复杂的框架，但要注意 recover 的使用时机和方式。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。