diff options
Diffstat (limited to 'main_test.go')
| -rw-r--r-- | main_test.go | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/main_test.go b/main_test.go index e231051..64c87cd 100644 --- a/main_test.go +++ b/main_test.go @@ -75,8 +75,9 @@ func TestFetchDomainsAndWrite(t *testing.T) { content: `# comment 0.0.0.0 domain1.com 0.0.0.0 domain2.com`, - want: `local-zone: "domain1.com" refuse -local-zone: "domain2.com" refuse + want: `server: + local-zone: "domain1.com" refuse + local-zone: "domain2.com" refuse `, wantSeen: map[string]struct{}{ "domain1.com": {}, @@ -88,8 +89,9 @@ local-zone: "domain2.com" refuse content: `# comment domain1.com domain2.com`, - want: `local-zone: "domain1.com" refuse -local-zone: "domain2.com" refuse + want: `server: + local-zone: "domain1.com" refuse + local-zone: "domain2.com" refuse `, wantSeen: map[string]struct{}{ "domain1.com": {}, @@ -102,8 +104,9 @@ local-zone: "domain2.com" refuse 0.0.0.0 domain1.com 0.0.0.0 DOMAIN1.COM domain2.com`, - want: `local-zone: "domain1.com" refuse -local-zone: "domain2.com" refuse + want: `server: + local-zone: "domain1.com" refuse + local-zone: "domain2.com" refuse `, wantSeen: map[string]struct{}{ "domain1.com": {}, |