Add test

main_test.go

@@ -60,6 +60,7 @@ func TestParseFlags(t *testing.T) {
 				stderr:       new(bytes.Buffer),
 				noHTTP2:      false,
 				localAddress: "0.0.0.0",
+				resolvers:    "",
 			},
 			wantErr: false,
 		},
@@ -301,6 +302,29 @@ func TestMakeOptions(t *testing.T) {
 			},
 			wantErr: false,
 		},
+		{
+			name: "use custom DNS resolvers",
+			cli: &cli{
+				method:    "GET",
+				resolvers: "1.2.3.4,192.168.11.1:53",
+			},
+			want: &attacker.Options{
+				Rate:       0,
+				Duration:   0,
+				Timeout:    0,
+				Method:     "GET",
+				Body:       []byte{},
+				Header:     http.Header{},
+				Workers:    0,
+				MaxWorkers: 0,
+				MaxBody:    0,
+				HTTP2:      true,
+				KeepAlive:  true,
+				Buckets:    []time.Duration{},
+				Resolvers:  []string{"1.2.3.4:53", "192.168.11.1:53"},
+			},
+			wantErr: false,
+		},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {