From 0feda9e9937f94b6aa69264c7c30fbf50897c519 Mon Sep 17 00:00:00 2001
From: ionutboangiu <ionut.boangiu@itsyscom.com>
Date: Mon, 26 May 2025 21:15:35 +0300
Subject: [PATCH] close db connections in test flush helpers

Previously, database connections would stay alive until "go test"
would finish, sometimes leading to max conn errors.
---
 engine/libtest.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/engine/libtest.go b/engine/libtest.go
index 1bf7c07cc..539a6b661 100644
--- a/engine/libtest.go
+++ b/engine/libtest.go
@@ -292,6 +292,7 @@ func InitDataDb(cfg *config.CGRConfig) error {
 	if err != nil {
 		return err
 	}
+	defer d.Close()
 	dm := NewDataManager(d, cfg.CacheCfg(), connMgr)
 
 	if err := dm.DataDB().Flush(""); err != nil {
@@ -315,6 +316,7 @@ func InitStorDb(cfg *config.CGRConfig) error {
 	if err != nil {
 		return err
 	}
+	defer storDb.Close()
 	db_Path := strings.Trim(cfg.StorDbCfg().Type, "*")
 	if err := storDb.Flush(path.Join(cfg.DataFolderPath, "storage",
 		db_Path)); err != nil {