39 lines
1.1 KiB
Go
39 lines
1.1 KiB
Go
package agent
|
||
|
||
import "testing"
|
||
|
||
func TestParseUFWStatus(t *testing.T) {
|
||
output := `Status: active
|
||
|
||
To Action From
|
||
-- ------ ----
|
||
22/tcp ALLOW Anywhere
|
||
7443/tcp ALLOW Anywhere
|
||
`
|
||
active, open := parseUFWStatus(output, 7443)
|
||
if !active || !open {
|
||
t.Fatalf("应识别出 ufw 已启用且 7443 已放行：active=%v open=%v", active, open)
|
||
}
|
||
_, open = parseUFWStatus(output, 9443)
|
||
if open {
|
||
t.Fatal("未放行的端口不应判定为已放行")
|
||
}
|
||
}
|
||
|
||
func TestParseUFWStatusInactive(t *testing.T) {
|
||
active, open := parseUFWStatus("Status: inactive\n", 7443)
|
||
if active || open {
|
||
t.Fatalf("inactive 时不应判定为启用：active=%v open=%v", active, open)
|
||
}
|
||
}
|
||
|
||
func TestCandidateHTTPSPorts(t *testing.T) {
|
||
if got := candidateHTTPSPorts(9443); len(got) != 1 || got[0] != 9443 {
|
||
t.Fatalf("显式端口应只返回该端口：%v", got)
|
||
}
|
||
got := candidateHTTPSPorts(0)
|
||
if len(got) != 3 || got[0] != 7443 || got[1] != 8443 || got[2] != 9443 {
|
||
t.Fatalf("自动模式应返回 7443/8443/9443：%v", got)
|
||
}
|
||
}
|