From 76977dff1d4ca882a10b0a361d5081e14420a35a Mon Sep 17 00:00:00 2001
From: baoyangchai <baoyangchai@gmail.com>
Date: Sun, 10 Apr 2022 19:23:33 +0800
Subject: [PATCH] add test

---
 tests/tests_test.go | 28 +++++++++++++++++++++++++---
 1 file changed, 25 insertions(+), 3 deletions(-)

diff --git a/tests/tests_test.go b/tests/tests_test.go
index 963e2a29..9a2970bd 100644
--- a/tests/tests_test.go
+++ b/tests/tests_test.go
@@ -18,10 +18,32 @@ import (
 
 var DB *gorm.DB
 
-func tearDown() {
-	gorm.Close(DB)
-}
+func TestClose(t *testing.T) {
+	if DB != nil {
+		err := gorm.Close(DB)
+		if err != nil {
+			t.Errorf("gorm close meet error:%+v\n", err)
+		}
+		if DB, err = OpenTestConnection(); err != nil {
+			log.Printf("failed to connect database, got error %v", err)
+			os.Exit(1)
+		} else {
+			sqlDB, err := DB.DB()
+			if err != nil {
+				log.Printf("failed to connect database, got error %v", err)
+				os.Exit(1)
+			}
 
+			err = sqlDB.Ping()
+			if err != nil {
+				log.Printf("failed to ping sqlDB, got error %v", err)
+				os.Exit(1)
+			}
+		}
+
+	}
+
+}
 func init() {
 	var err error
 	if DB, err = OpenTestConnection(); err != nil {