From be440e75122de5f7c19e2242a59246a92ce8edfe Mon Sep 17 00:00:00 2001
From: "jesse.tang" <1430482733@qq.com>
Date: Fri, 30 Sep 2022 11:14:34 +0800
Subject: [PATCH] fix possible nil panic in tests (#5720)

* fix maybe nil panic

* reset code
---
 tests/callbacks_test.go   | 3 +++
 tests/transaction_test.go | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/callbacks_test.go b/tests/callbacks_test.go
index 2bf9496b..4479da4c 100644
--- a/tests/callbacks_test.go
+++ b/tests/callbacks_test.go
@@ -113,6 +113,9 @@ func TestCallbacks(t *testing.T) {
 
 	for idx, data := range datas {
 		db, err := gorm.Open(nil, nil)
+		if err != nil {
+			t.Fatal(err)
+		}
 		callbacks := db.Callback()
 
 		for _, c := range data.callbacks {
diff --git a/tests/transaction_test.go b/tests/transaction_test.go
index 0ac04a04..5872da94 100644
--- a/tests/transaction_test.go
+++ b/tests/transaction_test.go
@@ -102,7 +102,7 @@ func TestTransactionWithBlock(t *testing.T) {
 		return errors.New("the error message")
 	})
 
-	if err.Error() != "the error message" {
+	if err != nil && err.Error() != "the error message" {
 		t.Fatalf("Transaction return error will equal the block returns error")
 	}