test: 补充单元测试覆盖率 29.9% → 36.6%

新建 18 个测试文件,追加 30 个已有测试文件,覆盖协议解析、
错误分类、CEL 表达式求值、YAML 反序列化、字节编码等纯函数。
This commit is contained in:
ZacharyZcR
2026-06-17 12:51:41 +08:00
parent 6eff1d5ccf
commit 0612255893
48 changed files with 7556 additions and 1 deletions
+60
View File
@@ -0,0 +1,60 @@
//go:build plugin_ssh || !plugin_selective
package services
import (
"errors"
"testing"
)
func TestClassifySSHErrorType(t *testing.T) {
tests := []struct {
name string
err error
want ErrorType
}{
{"nil error", nil, ErrorTypeUnknown},
{"unable to authenticate", errors.New("unable to authenticate"), ErrorTypeAuth},
{"no supported methods remain", errors.New("no supported methods remain"), ErrorTypeAuth},
{"handshake failed", errors.New("handshake failed"), ErrorTypeThrottle},
{"ssh disconnect", errors.New("ssh: disconnect"), ErrorTypeThrottle},
{"max startups", errors.New("max startups"), ErrorTypeThrottle},
{"connection refused", errors.New("connection refused"), ErrorTypeNetwork},
{"random error", errors.New("random error"), ErrorTypeUnknown},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got := classifySSHErrorType(tt.err)
if got != tt.want {
t.Errorf("classifySSHErrorType(%v) = %v, want %v", tt.err, got, tt.want)
}
})
}
}
func TestClassifySSHError(t *testing.T) {
authKeywords := []string{"bad password", "invalid key"}
throttleKeywords := []string{"rate limited", "too fast"}
tests := []struct {
name string
err error
want ErrorType
}{
{"nil error", nil, ErrorTypeUnknown},
{"custom auth keyword", errors.New("bad password provided"), ErrorTypeAuth},
{"custom throttle keyword", errors.New("rate limited by server"), ErrorTypeThrottle},
{"network error", errors.New("connection refused"), ErrorTypeNetwork},
{"no match", errors.New("something else"), ErrorTypeUnknown},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got := classifySSHError(tt.err, authKeywords, throttleKeywords)
if got != tt.want {
t.Errorf("classifySSHError(%v) = %v, want %v", tt.err, got, tt.want)
}
})
}
}