Add sqlite cleanup

netbirdio · Jun 12, 2024 · af06be4 · af06be4
1 parent 14639af
commit af06be4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/management/server/store.go b/management/server/store.go
@@ -196,9 +196,7 @@ func NewTestStoreFromJson(dataDir string) (Store, func(), error) {
 
 	var (
 		store   Store
-		cleanUp = func() {
-			// empty store cleanUp
-		}
+		cleanUp func()
 	)
 
 	if kind == PostgresStoreEngine {
@@ -221,6 +219,7 @@ func NewTestStoreFromJson(dataDir string) (Store, func(), error) {
 		if err != nil {
 			return nil, nil, err
 		}
+		cleanUp = func() { store.Close() }
 	}
 
 	return store, cleanUp, nil