Go 在测试中设置/重置模拟函数
示例
本示例说明如何模拟与单元测试无关的函数调用，然后使用defer语句将模拟的函数调用重新分配回其原始函数。
var validate = validateDTD
// ParseXML解析b以获取XML元素和值，并将其作为以下内容的映射返回
// 字符串键/值对。
func ParseXML(b []byte) (map[string]string, error) {
// 我们不在乎在单元测试中针对DTD进行验证
if err := validate(b); err != nil {
return err
}
// 解析b等的代码
}
func validateDTD(b []byte) error {
// 从某些外部存储中获取DTD，并使用它来验证b等。
}在我们的单元测试中
func TestParseXML(t *testing.T) {
// 将原始验证函数分配给变量。
originalValidate = validate
// 在此测试中使用mockValidate函数。
validate = mockValidate
// 将重新分配推迟回原始验证功能。
defer func() {
validate = originalValidate
}()
var input []byte
actual, err := ParseXML(input)
// 断言等
}
func mockValidate(b []byte) error {
return nil // 总是返回nil
}