From 2693db218ad2224a749a2b21dcc14a48502a2fac Mon Sep 17 00:00:00 2001
From: plutoyty <2631223275@qq.com>
Date: Wed, 12 Jul 2023 11:35:06 +0800
Subject: [PATCH] For "add database retry" add test

---
 pkg/common/db/relation/mysql_init.go      |  2 +-
 pkg/common/db/relation/mysql_init_test.go | 37 +++++++++++++++++++++++
 2 files changed, 38 insertions(+), 1 deletion(-)
 create mode 100644 pkg/common/db/relation/mysql_init_test.go

diff --git a/pkg/common/db/relation/mysql_init.go b/pkg/common/db/relation/mysql_init.go
index fe467b675..8586ac338 100644
--- a/pkg/common/db/relation/mysql_init.go
+++ b/pkg/common/db/relation/mysql_init.go
@@ -31,7 +31,7 @@ import (
 )
 
 const (
-	maxRetry = 100
+	maxRetry = 100 //number of retries
 )
 
 //newMysqlGormDB Initialize the database connection
diff --git a/pkg/common/db/relation/mysql_init_test.go b/pkg/common/db/relation/mysql_init_test.go
new file mode 100644
index 000000000..c6fd3e14b
--- /dev/null
+++ b/pkg/common/db/relation/mysql_init_test.go
@@ -0,0 +1,37 @@
+// Copyright © 2023 OpenIM. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package relation
+
+import (
+	"fmt"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
+	"testing"
+)
+
+//TestNewGormDB Test the retry of sporadic errors and the direct exit of wrong password.
+func TestNewGormDB(t *testing.T) {
+	err := config.InitConfig("config_folder_path")
+	if err != nil {
+		return
+	}
+	db, err := newMysqlGormDB()
+	if err != nil {
+		fmt.Println("password error")
+		return
+	}
+	if db != nil {
+		fmt.Println("success connect")
+	}
+}