1 回答

TA貢獻(xiàn)1757條經(jīng)驗 獲得超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
}
然后在測試文件中,您需要對您期望錯誤的事實有一個理解或協(xié)議(正確地)。就像下面的標(biāo)志一樣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是為了演示,您實際上應(yīng)該使用一些不同的方法來檢查錯誤,這樣更快。這取決于您的應(yīng)用程序中錯誤的設(shè)計方式。您可以使用標(biāo)記錯誤或類型/接口檢查?;蜃远x錯誤類型中基于 const 的枚舉字段。
- 1 回答
- 0 關(guān)注
- 98 瀏覽
添加回答
舉報