fix: tests

natesales · natesales · commit 628e9a374616 · 2026-02-26T20:43:52.000-08:00
diff --git a/main_test.go b/main_test.go
@@ -57,7 +57,7 @@ func TestMainRawFormat(t *testing.T) {
 	)
 	assert.Nil(t, err)
 	assert.Contains(t, out.String(), "v=spf1 -all")
-	assert.Contains(t, out.String(), "a.iana-servers.net")
+	assert.Containsf(t, out.String(), "hera.ns.cloudflare.com", out.String())
 }
 
 func TestMainJSONFormat(t *testing.T) {
@@ -69,7 +69,7 @@ func TestMainJSONFormat(t *testing.T) {
 	assert.Nil(t, err)
 	o := strings.ReplaceAll(out.String(), `\\"`, `"`)
 	assert.Contains(t, o, `"preference":0,"mx":"."`)
-	assert.Contains(t, o, `"ns":"a.iana-servers.net."`)
+	assert.Contains(t, o, "hera.ns.cloudflare.com")
 	assert.Contains(t, o, `"txt":["v=spf1 -all"`)
 }
 

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ func TestMainRawFormat(t *testing.T) {`
`57`	`57`	`)`
`58`	`58`	`assert.Nil(t, err)`
`59`	`59`	`assert.Contains(t, out.String(), "v=spf1 -all")`
`60`		`- assert.Contains(t, out.String(), "a.iana-servers.net")`
	`60`	`+ assert.Containsf(t, out.String(), "hera.ns.cloudflare.com", out.String())`
`61`	`61`	`}`
`62`	`62`
`63`	`63`	`func TestMainJSONFormat(t *testing.T) {`
`@@ -69,7 +69,7 @@ func TestMainJSONFormat(t *testing.T) {`
`69`	`69`	`assert.Nil(t, err)`
`70`	`70`	o := strings.ReplaceAll(out.String(), `\\"`, `"`)
`71`	`71`	assert.Contains(t, o, `"preference":0,"mx":"."`)
`72`		- assert.Contains(t, o, `"ns":"a.iana-servers.net."`)
	`72`	`+ assert.Contains(t, o, "hera.ns.cloudflare.com")`
`73`	`73`	assert.Contains(t, o, `"txt":["v=spf1 -all"`)
`74`	`74`	`}`
`75`	`75`