tests: match NS records too

natesales · Jul 5, 2023 · 6f19164 · 6f19164
1 parent 2ed0398
commit 6f19164
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main_test.go b/main_test.go
@@ -163,6 +163,7 @@ func TestMainInferredServer(t *testing.T) {
 	assert.Regexp(t, regexp.MustCompile(`example.com. .* A .*`), out.String())
 	assert.Regexp(t, regexp.MustCompile(`example.com. .* AAAA .*`), out.String())
 	assert.Regexp(t, regexp.MustCompile(`example.com. .* MX .*`), out.String())
+	assert.Regexp(t, regexp.MustCompile(`example.com. .* NS .*`), out.String())
 }
 
 func TestMainInvalidReverseQuery(t *testing.T) {
@@ -321,6 +322,7 @@ func TestMainTLS12(t *testing.T) {
 	assert.Regexp(t, regexp.MustCompile(`example.com. .* A .*`), out.String())
 	assert.Regexp(t, regexp.MustCompile(`example.com. .* AAAA .*`), out.String())
 	assert.Regexp(t, regexp.MustCompile(`example.com. .* MX .*`), out.String())
+	assert.Regexp(t, regexp.MustCompile(`example.com. .* NS .*`), out.String())
 }
 
 func TestMainNSID(t *testing.T) {