portscan_test.go

package config

import (
	"reflect"
	"testing"
)

func TestPortScanConf_getScanTechniques(t *testing.T) {
	tests := []struct {
		name       string
		techniques []string
		want       []ScanTechnique
	}{
		{
			name:       "nil",
			techniques: []string{},
			want:       []ScanTechnique{},
		},
		{
			name:       "single",
			techniques: []string{"sS"},
			want:       []ScanTechnique{TCPSYN},
		},
		{
			name:       "multiple",
			techniques: []string{"sS", "sT"},
			want:       []ScanTechnique{TCPSYN, TCPConnect},
		},
		{
			name:       "unknown",
			techniques: []string{"sU"},
			want:       []ScanTechnique{NotSupportTechnique},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			c := PortScanConf{ScanTechniques: tt.techniques}
			if got := c.GetScanTechniques(); !reflect.DeepEqual(got, tt.want) {
				t.Errorf("PortScanConf.getScanTechniques() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestPortScanConf_IsZero(t *testing.T) {
	tests := []struct {
		name string
		conf PortScanConf
		want bool
	}{
		{
			name: "not zero",
			conf: PortScanConf{ScannerBinPath: "/usr/bin/nmap"},
			want: false,
		},
		{
			name: "zero",
			conf: PortScanConf{},
			want: true,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := tt.conf.IsZero(); got != tt.want {
				t.Errorf("PortScanConf.IsZero() = %v, want %v", got, tt.want)
			}
		})
	}
}