tests: remove test artifacts

natesales · Sep 7, 2023 · 96c59bb · 96c59bb
1 parent 4f1f371
commit 96c59bb
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/main_test.go b/main_test.go
@@ -2,6 +2,8 @@ package main
 
 import (
 	"bytes"
+	"os"
+	"path/filepath"
 	"regexp"
 	"strings"
 	"testing"
@@ -471,6 +473,14 @@ func TestMainRecAXFR(t *testing.T) {
 	}, &out))
 	assert.Contains(t, out.String(), `AXFR zonetransfer.me.`)
 	assert.Contains(t, out.String(), `AXFR internal.zonetransfer.me.`)
+
+	// Remove zonetransfer files
+	files, err := filepath.Glob("zonetransfer.me*")
+	assert.Nil(t, err)
+	for _, f := range files {
+		err := os.RemoveAll(f)
+		assert.Nil(t, err)
+	}
 }
 
 func TestMainShowAll(t *testing.T) {