refactor TestBuildTimeFlag's cleanup

satotake · Apr 25, 2022 · 73c39bd · 73c39bd
1 parent dc25c77
commit 73c39bd
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/commands/list_test.go b/commands/list_test.go
@@ -30,17 +30,13 @@ func captureStdout(f func() error) (string, error) {
 func TestListAll(t *testing.T) {
 	c := qt.New(t)
 	dir, clean, err := createSimpleTestSite(t, testSiteConfig{})
-	defer clean()
+	t.Cleanup(clean)
 
 	c.Assert(err, qt.IsNil)
 
 	hugoCmd := newCommandsBuilder().addAll().build()
 	cmd := hugoCmd.getCommand()
 
-	defer func() {
-		os.RemoveAll(dir)
-	}()
-
 	cmd.SetArgs([]string{"-s=" + dir, "list", "all"})
 
 	out, err := captureStdout(func() error {
@@ -84,17 +80,13 @@ date: 2100-11-06
 Content
 
 `)
-	defer clean()
+	t.Cleanup(clean)
 
 	c.Assert(err, qt.IsNil)
 
 	hugoCmd := newCommandsBuilder().addAll().build()
 	cmd := hugoCmd.getCommand()
 
-	defer func() {
-		os.RemoveAll(dir)
-	}()
-
 	cmd.SetArgs([]string{"-s=" + dir, "list", "future"})
 
 	out, err := captureStdout(func() error {