From 5467887e1ccceb90af8035dd5dc7739f4966e4f7 Mon Sep 17 00:00:00 2001
From: pluto <83957921+plutoyty@users.noreply.github.com>
Date: Wed, 12 Jul 2023 15:23:25 +0800
Subject: [PATCH] For "add database retry" add test (#500)

* Add database retry

* For "add database retry" add test

* For "add database retry" add test

* For 'add database retry' add test

Signed-off-by: plutoyty <2631223275@qq.com>

* Remove .idea directory and add it to .gitignore

---------

Signed-off-by: plutoyty <2631223275@qq.com>
---
 pkg/common/db/relation/mysql_init.go      |  2 +-
 pkg/common/db/relation/mysql_init_test.go | 38 +++++++++++++++++++++++
 2 files changed, 39 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 d480ae6e8..4a6758cba 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..a4f38f7bc
--- /dev/null
+++ b/pkg/common/db/relation/mysql_init_test.go
@@ -0,0 +1,38 @@
+// 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 {
+		fmt.Println("config load error")
+		return
+	}
+	db, err := newMysqlGormDB()
+	if err != nil {
+		fmt.Println("password error")
+		return
+	}
+	if db != nil {
+		fmt.Println("success connect")
+	}
+}