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)
|
|||
|
|
}
|
|||
|
|
}
|