Fix test folder cleanup

mesg-foundation · Oct 23, 2019 · 8fdb6cd · 8fdb6cd
1 parent aa45510
commit 8fdb6cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/database/ownership_test.go b/database/ownership_test.go
@@ -17,7 +17,7 @@ func TestOwnershipDB(t *testing.T) {
 	ownership.RegisterCodec(cdc)
 
 	dir, _ := ioutil.TempDir("", "ownership.db.test")
-	defer os.Remove(dir)
+	defer os.RemoveAll(dir)
 
 	store, err := store.NewLevelDBStore(dir)
 	require.NoError(t, err)

diff --git a/database/process_db_test.go b/database/process_db_test.go
@@ -12,7 +12,7 @@ import (
 
 func TestProcessDB(t *testing.T) {
 	dir, _ := ioutil.TempDir("", "process.db.test")
-	defer os.Remove(dir)
+	defer os.RemoveAll(dir)
 
 	db, err := NewProcessDB(dir)
 	require.NoError(t, err)
@@ -66,7 +66,7 @@ func TestProcessDB(t *testing.T) {
 
 func TestHashIsDifferent(t *testing.T) {
 	dir, _ := ioutil.TempDir("", "process.db.test")
-	defer os.Remove(dir)
+	defer os.RemoveAll(dir)
 
 	db, err := NewProcessDB(dir)
 	require.NoError(t, err)