1 回答

TA贡献1757条经验 获得超8个赞
如果我理解正确,您想在某些条件下检查正确的错误返回。
这是:
func coolFunction(input int) (int, error) {
var err error
var number int
if input == 1 {
err = errors.New("This is an error")
number = 400
} else {
err = nil
number = 200
}
return number, err
}
然后在测试文件中,您需要对您期望错误的事实有一个理解或协议(正确地)。就像下面的标志一样expectError bool,这对于您的情况来说是正确的。
您还可以拥有特定错误类型的字段并检查返回的类型是否正是该类型。
func TestCoolFunction(t *testing.T) {
var testTable = []struct {
desc string
in int
expectError bool
out int
}{
{"Should_fail", 1, true, 0},
{"Should_pass", 100, false, 200},
}
for _, tt := range testTable {
t.Run(tt.desc, func(t *testing.T) {
out, err := coolFunction(tt.in)
if tt.expectError {
if err == nil {
t.Error("Failed")
}
} else {
if out != tt.out {
t.Errorf("got %d, want %d", out, tt.out)
}
}
})
}
}
DeepEqual使用@mkopriva 建议的添加特定错误检查
func TestCoolFunction(t *testing.T) {
var testTable = []struct {
desc string
in int
expectError error
out int
}{
{"Should_fail", 1, errors.New("This is an error"), 0},
{"Should_pass", 100, nil, 200},
}
for _, tt := range testTable {
t.Run(tt.desc, func(t *testing.T) {
out, err := coolFunction(tt.in)
if tt.expectError != nil {
if !reflect.DeepEqual(err, tt.expectError) {
t.Error("Failed")
}
} else {
if out != tt.out {
t.Errorf("got %d, want %d", out, tt.out)
}
}
})
}
}
同样DeepEqual是为了演示,您实际上应该使用一些不同的方法来检查错误,这样更快。这取决于您的应用程序中错误的设计方式。您可以使用标记错误或类型/接口检查。或自定义错误类型中基于 const 的枚举字段。
- 1 回答
- 0 关注
- 105 浏览
添加回答
举报