From 321f782c892aaeb943f20d34f30c56894b8a4048 Mon Sep 17 00:00:00 2001
From: baoyangchai <baoyangchai@gmail.com>
Date: Sun, 10 Apr 2022 19:05:02 +0800
Subject: [PATCH] add test

---
 tests/create_test.go | 2 +-
 tests/tests_test.go  | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/create_test.go b/tests/create_test.go
index 2b23d440..e404822a 100644
--- a/tests/create_test.go
+++ b/tests/create_test.go
@@ -14,7 +14,7 @@ import (
 
 func TestCreate(t *testing.T) {
 	user := *GetUser("create", Config{})
-
+	defer tearDown()
 	if results := DB.Create(&user); results.Error != nil {
 		t.Fatalf("errors happened when create: %v", results.Error)
 	} else if results.RowsAffected != 1 {
diff --git a/tests/tests_test.go b/tests/tests_test.go
index 08f4f193..963e2a29 100644
--- a/tests/tests_test.go
+++ b/tests/tests_test.go
@@ -18,6 +18,10 @@ import (
 
 var DB *gorm.DB
 
+func tearDown() {
+	gorm.Close(DB)
+}
+
 func init() {
 	var err error
 	if DB, err = OpenTestConnection(); err != nil {