Add shutdown to storage tests

iotaledger · Dec 5, 2023 · 8829fea · 8829fea
1 parent 8b9d1f8
commit 8829fea
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/storage/storage_test.go b/pkg/storage/storage_test.go
@@ -168,6 +168,7 @@ func TestStorage_PruneBySize(t *testing.T) {
 
 func TestStorage_RestoreFromDisk(t *testing.T) {
 	tf := NewTestFramework(t, t.TempDir(), storage.WithPruningDelay(1))
+	defer tf.Shutdown()
 
 	totalEpochs := 9
 	tf.GeneratePermanentData(5 * MB)
@@ -213,6 +214,7 @@ func TestStorage_RestoreFromDisk(t *testing.T) {
 
 func TestStorage_CopyFromForkedStorageEmpty(t *testing.T) {
 	tf1 := NewTestFramework(t, t.TempDir())
+	defer tf1.Shutdown()
 
 	totalEpochs := 14
 	// Generate data in the old storage (source). It contains data since the genesis and one epoch after the fork.