Remove data in close method.

xichen2020 · Feb 28, 2019 · 3ee0549 · 3ee0549
1 parent 9064d51
commit 3ee0549
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/integration/setup.go b/integration/setup.go
@@ -132,18 +132,15 @@ func (ts *testServerSetup) stopServer() error {
 	}
 	close(ts.doneCh)
 
-	// Remove data directory to prevent accumulation of data.
-	if err := os.RemoveAll(filepath.Join(ts.db.Options().FilePathPrefix(), "data")); err != nil {
-		return err
-	}
-
 	// Wait for graceful server shutdown.
 	<-ts.closedCh
 	return nil
 }
 
 func (ts *testServerSetup) close(t *testing.T) {
-	// TODO(wjang): Delete the database files as well.
+	// Remove data directory to prevent accumulation of data.
+	err := os.RemoveAll(filepath.Join(ts.db.Options().FilePathPrefix(), "data"))
+	require.NoError(t, err)
 }
 
 func loadConfig(t *testing.T, config string) configuration {