test: add unittest for DisableQTypes

wolf-joe · Oct 12, 2020 · 606f1f2 · 606f1f2
1 parent 3769b65
commit 606f1f2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/cmd/conf/conf_test.go b/cmd/conf/conf_test.go
@@ -134,6 +134,7 @@ func TestNewHandler(t *testing.T) {
 	p := gomonkey.ApplyFunc(toml.DecodeFile,
 		func(fn string, conf interface{}) (toml.MetaData, error) {
 			conf.(*Conf).DisableIPv6 = true
+			conf.(*Conf).DisableQTypes = []string{"HTTPS"}
 			conf.(*Conf).Listen = ":53/tcp"
 			return toml.MetaData{}, nil
 		})

diff --git a/inbound/server_test.go b/inbound/server_test.go
@@ -83,6 +83,13 @@ func TestHandler(t *testing.T) {
 	assert.NotNil(t, writer.r)
 	assert.Equal(t, len(writer.r.Answer), 0)
 
+	// 测试DisableIPv6
+	handler.DisableIPv6 = false
+	handler.DisableQTypes = map[string]bool{"AAAA": true}
+	handler.ServeDNS(writer, req)
+	assert.NotNil(t, writer.r)
+	assert.Equal(t, len(writer.r.Answer), 0)
+
 	req.SetQuestion("ip.cn.", dns.TypeA)
 
 	mocker := mock.Mocker{}