From e5dddf4bbf693b3d95bc69cab1278d89e4d2a210 Mon Sep 17 00:00:00 2001 From: ZacharyZcR Date: Thu, 1 Oct 2026 13:18:58 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3=E7=AB=AF=E5=8F=A3=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E4=B8=8E=E6=A8=A1=E5=9D=97=E9=BB=98=E8=AE=A4=E7=AB=AF?= =?UTF-8?q?=E5=8F=A3=E7=9A=84=E7=BB=84=E5=90=88=E5=9B=9E=E5=BD=92=E7=94=A8?= =?UTF-8?q?=E4=BE=8B?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main_cli_plugins_test.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/main_cli_plugins_test.go b/main_cli_plugins_test.go index 646c206..9d5073f 100644 --- a/main_cli_plugins_test.go +++ b/main_cli_plugins_test.go @@ -18,6 +18,10 @@ import ( ) func TestCLIPluginDefaultPorts(t *testing.T) { + portsFile := filepath.Join(t.TempDir(), "ports.txt") + if err := os.WriteFile(portsFile, []byte("9139\n8445\n"), 0600); err != nil { + t.Fatal(err) + } tests := []struct { name string args []string @@ -27,7 +31,7 @@ func TestCLIPluginDefaultPorts(t *testing.T) { {name: "multiple plugins", args: []string{"-m", "ssh, netbios,ssh,"}, want: "22,137,139,2200,2222,22222"}, {name: "nonstandard port", args: []string{"-m", "ssh", "-p", "2222"}, want: "2222"}, {name: "explicit common ports", args: []string{"-m", "netbios", "-p", config.MainPorts}, want: config.MainPorts}, - {name: "ports file", args: []string{"-m", "netbios", "-pf", "ports.txt"}, want: config.MainPorts}, + {name: "ports file", args: []string{"-m", "netbios", "-pf", portsFile}, want: "9139,8445"}, {name: "all", args: []string{"-m", "all"}, want: config.MainPorts}, {name: "web plugin", args: []string{"-m", "webtitle"}, want: config.MainPorts}, {name: "web and service", args: []string{"-m", "ssh,webtitle"}, want: config.MainPorts},