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

新建 18 个测试文件，追加 30 个已有测试文件，覆盖协议解析、错误分类、CEL 表达式求值、YAML 反序列化、字节编码等纯函数。
2026-09-27 05:31:53 +08:00 · 2026-06-17 12:51:41 +08:00
parent 6eff1d5ccf
commit 0612255893
48 changed files with 7556 additions and 1 deletions
@@ -0,0 +1,100 @@
+//go:build plugin_rdp || !plugin_selective
+
+package services
+
+import (
+	"testing"
+
+	"github.com/shadow1ng/fscan/common/i18n"
+)
+
+func TestBuildBanner(t *testing.T) {
+	p := &RDPPlugin{}
+	fallback := i18n.GetText("rdp_remote_desktop_service")
+
+	tests := []struct {
+		name   string
+		osInfo map[string]any
+		want   string
+	}{
+		{
+			name:   "nil map",
+			osInfo: nil,
+			want:   fallback,
+		},
+		{
+			name:   "empty map",
+			osInfo: map[string]any{},
+			want:   fallback,
+		},
+		{
+			name:   "OsVerion and NetBIOSComputerName",
+			osInfo: map[string]any{"OsVerion": "Windows 10", "NetBIOSComputerName": "DESKTOP-01"},
+			want:   "RDP (Windows 10, DESKTOP-01)",
+		},
+		{
+			name:   "only OsVerion",
+			osInfo: map[string]any{"OsVerion": "Windows Server 2019"},
+			want:   "RDP (Windows Server 2019)",
+		},
+		{
+			name:   "only NetBIOSComputerName",
+			osInfo: map[string]any{"NetBIOSComputerName": "MY-HOST"},
+			want:   "RDP (Hostname:MY-HOST)",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := p.buildBanner(tt.osInfo)
+			if got != tt.want {
+				t.Errorf("buildBanner() = %q, want %q", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestExtractStringField(t *testing.T) {
+	p := &RDPPlugin{}
+
+	tests := []struct {
+		name   string
+		osInfo map[string]any
+		key    string
+		want   string
+	}{
+		{
+			name:   "key exists and is string",
+			osInfo: map[string]any{"foo": "bar"},
+			key:    "foo",
+			want:   "bar",
+		},
+		{
+			name:   "key exists but not string",
+			osInfo: map[string]any{"foo": 42},
+			key:    "foo",
+			want:   "",
+		},
+		{
+			name:   "key does not exist",
+			osInfo: map[string]any{"foo": "bar"},
+			key:    "missing",
+			want:   "",
+		},
+		{
+			name:   "nil map",
+			osInfo: nil,
+			key:    "foo",
+			want:   "",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := p.extractStringField(tt.osInfo, tt.key)
+			if got != tt.want {
+				t.Errorf("extractStringField(%q) = %q, want %q", tt.key, got, tt.want)
+			}
+		})
+	}
+}