From a523fc4e2c612e758e155440abd7d674b01d29e7 Mon Sep 17 00:00:00 2001
From: HFO4 <912394456@qq.com>
Date: Tue, 26 Apr 2022 19:22:37 +0800
Subject: [PATCH] test: Folder.Create

---
 models/folder_test.go | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/models/folder_test.go b/models/folder_test.go
index f3137b3..97d975b 100644
--- a/models/folder_test.go
+++ b/models/folder_test.go
@@ -18,8 +18,8 @@ func TestFolder_Create(t *testing.T) {
 	}
 
 	// 不存在，插入成功
-	mock.ExpectBegin()
 	mock.ExpectQuery("SELECT(.+)folders(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}))
+	mock.ExpectBegin()
 	mock.ExpectExec("INSERT(.+)").WillReturnResult(sqlmock.NewResult(5, 1))
 	mock.ExpectCommit()
 	fid, err := folder.Create()
@@ -28,19 +28,18 @@ func TestFolder_Create(t *testing.T) {
 	asserts.NoError(mock.ExpectationsWereMet())
 
 	// 插入失败
-	mock.ExpectBegin()
 	mock.ExpectQuery("SELECT(.+)folders(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}))
+	mock.ExpectBegin()
 	mock.ExpectExec("INSERT(.+)").WillReturnError(errors.New("error"))
 	mock.ExpectRollback()
+	mock.ExpectQuery("SELECT(.+)folders(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1))
 	fid, err = folder.Create()
-	asserts.Error(err)
-	asserts.Equal(uint(0), fid)
+	asserts.NoError(err)
+	asserts.Equal(uint(1), fid)
 	asserts.NoError(mock.ExpectationsWereMet())
 
 	// 存在，直接返回
-	mock.ExpectBegin()
 	mock.ExpectQuery("SELECT(.+)folders(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(5))
-	mock.ExpectCommit()
 	fid, err = folder.Create()
 	asserts.NoError(err)
 	asserts.Equal(uint(5), fid)