bugfix: prevent memory leak from tests

tauraamui · Jul 6, 2023 · 0902a06 · 0902a06
1 parent 9859815
commit 0902a06
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/storage/store_test.go b/storage/store_test.go
@@ -36,9 +36,7 @@ func TestStoreAndLoadMultipleBalloonsSuccess(t *testing.T) {
 	defer db.Close()
 
 	store := storage.New(db)
-	defer func() {
-		is.NoErr(store.Close())
-	}()
+	defer store.Close()
 
 	bigRedBalloon := Balloon{Color: "RED", Size: 695}
 	smallYellowBalloon := Balloon{Color: "YELLOW", Size: 112}
@@ -65,9 +63,7 @@ func TestStoreMultipleBalloonsSuccess(t *testing.T) {
 	defer db.Close()
 
 	store := storage.New(db)
-	defer func() {
-		is.NoErr(store.Close())
-	}()
+	defer store.Close()
 
 	bigRedBalloon := Balloon{Color: "RED", Size: 695}
 	smallYellowBalloon := Balloon{Color: "YELLOW", Size: 112}
@@ -89,9 +85,7 @@ func TestStoreMultipleBalloonsAndCakesInSuccessionRetainsCorrectRowIDs(t *testin
 	defer db.Close()
 
 	store := storage.New(db)
-	defer func() {
-		is.NoErr(store.Close())
-	}()
+	defer store.Close()
 
 	bigRedBalloon := Balloon{Color: "RED", Size: 695}
 	disguistingVeganCake := Cake{Type: "INEDIBLE", Calories: -38}