chore: refactor (#1814)

2022-04-22 09:37:09 +08:00
parent 162e9cef86
commit bc3c9484d1
5 changed files with 32 additions and 20 deletions
--- a/core/stores/mon/model_test.go
+++ b/core/stores/mon/model_test.go
@@ -18,6 +18,7 @@ func TestModel_StartSession(t *testing.T) {
 		m := createModel(mt)
 		sess, err := m.StartSession()
 		assert.Nil(t, err)
+		defer sess.EndSession(context.Background())

 		_, err = sess.WithTransaction(context.Background(), func(sessCtx mongo.SessionContext) (interface{}, error) {
 			_ = sessCtx.StartTransaction()
@@ -26,10 +27,8 @@ func TestModel_StartSession(t *testing.T) {
 			return nil, nil
 		})
 		assert.Nil(t, err)
-
 		assert.NoError(t, sess.CommitTransaction(context.Background()))
 		assert.Error(t, sess.AbortTransaction(context.Background()))
-		sess.EndSession(context.Background())
 	})
 }