mirror of
https://github.com/shadow1ng/fscan.git
synced 2026-09-25 12:41:53 +08:00
test: 补充单元测试覆盖率 29.9% → 36.6%
新建 18 个测试文件,追加 30 个已有测试文件,覆盖协议解析、 错误分类、CEL 表达式求值、YAML 反序列化、字节编码等纯函数。
This commit is contained in:
@@ -6,6 +6,7 @@ import (
|
||||
"bytes"
|
||||
"encoding/base64"
|
||||
"encoding/binary"
|
||||
"errors"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
@@ -128,3 +129,25 @@ func TestBuildMongoSCRAMClientFinalBuildsProof(t *testing.T) {
|
||||
t.Fatalf("client final = %q", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestClassifyMongoDBErrorType(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
err error
|
||||
want ErrorType
|
||||
}{
|
||||
{"nil", nil, ErrorTypeUnknown},
|
||||
{"authentication failed", errors.New("authentication failed"), ErrorTypeAuth},
|
||||
{"bad auth", errors.New("bad auth"), ErrorTypeAuth},
|
||||
{"dial tcp", errors.New("dial tcp connection refused"), ErrorTypeNetwork},
|
||||
{"eof", errors.New("eof"), ErrorTypeNetwork},
|
||||
{"unknown", errors.New("random mongodb error"), ErrorTypeUnknown},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := classifyMongoDBErrorType(tt.err); got != tt.want {
|
||||
t.Errorf("classifyMongoDBErrorType() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user