From 0c1cac3d89253cac3061090496da83b1d3c3d9fa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C4=B0smail?= <ismaiil_0234@hotmail.com>
Date: Mon, 4 Nov 2024 15:46:10 +0300
Subject: [PATCH] Update store.go

---
 management/server/store.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/management/server/store.go b/management/server/store.go
index d8a70c825..142ce5b1a 100644
--- a/management/server/store.go
+++ b/management/server/store.go
@@ -305,6 +305,23 @@ func NewTestStoreFromSQL(ctx context.Context, filename string, dataDir string) (
 		}
 	}
 
+	if kind == MysqlStoreEngine {
+		cleanUp, err = testutil.CreateMyDB()
+		if err != nil {
+			return nil, nil, err
+		}
+
+		dsn, ok := os.LookupEnv(mysqlDsnEnv)
+		if !ok {
+			return nil, nil, fmt.Errorf("%s is not set", mysqlDsnEnv)
+		}
+
+		store, err = NewMysqlStoreFromSqlStore(ctx, store, dsn, nil)
+		if err != nil {
+			return nil, nil, err
+		}
+	}
+
 	return store, cleanUp, nil
 }