Stage/vuls
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add Test_matchListenPorts

Browse Source
This commit is contained in:
MaineK00n
2020-10-15 08:44:23 +09:00
parent 2a98450425
commit a0eb6467c1
1 changed files with 27 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
scan/base_test.go
View File

@@ -363,6 +363,33 @@ func Test_detectScanDest(t *testing.T) {
}
}
func Test_matchListenPorts(t *testing.T) {
type args struct {
open []string
port models.ListenPorts
}
tests := []struct {
name string
args args
expect []string
}{
{name: "open_empty", args: args{open: []string{}, port: models.ListenPorts{Address: "127.0.0.1", Port: "22"}}, expect: []string{}},
{name: "port_empty", args: args{open: []string{"127.0.0.1:22"}, port: models.ListenPorts{}}, expect: []string{}},
{name: "single_match", args: args{open: []string{"127.0.0.1:22"}, port: models.ListenPorts{Address: "127.0.0.1", Port: "22"}}, expect: []string{"127.0.0.1"}},
{name: "no_match_address", args: args{open: []string{"127.0.0.1:22"}, port: models.ListenPorts{Address: "192.168.1.1", Port: "22"}}, expect: []string{}},
{name: "no_match_port", args: args{open: []string{"127.0.0.1:22"}, port: models.ListenPorts{Address: "127.0.0.1", Port: "80"}}, expect: []string{}},
{name: "asterisk_match", args: args{open: []string{"127.0.0.1:22", "127.0.0.1:80", "192.168.1.1:22"}, port: models.ListenPorts{Address: "*", Port: "22"}}, expect: []string{"127.0.0.1", "192.168.1.1"}},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if match := matchListenPorts(tt.args.open, tt.args.port); !reflect.DeepEqual(match, tt.expect) {
t.Errorf("matchListenPorts() = %v, want %v", match, tt.expect)
}
})
}
}
func Test_base_parseListenPorts(t *testing.T) {
tests := []struct {
name string