Test: filesystem/file, filesystem, filesystem/hook, model/file/Create

models/file_test.go

@@ -1,6 +1,7 @@
 package model
 
 import (
+	"errors"
 	"github.com/DATA-DOG/go-sqlmock"
 	"github.com/stretchr/testify/assert"
 	"testing"
@@ -16,3 +17,27 @@ func TestGetFileByPathAndName(t *testing.T) {
 	asserts.Equal("1.cia", file.Name)
 	asserts.NoError(mock.ExpectationsWereMet())
 }
+
+func TestFile_Create(t *testing.T) {
+	asserts := assert.New(t)
+	file := File{
+		Name: "123",
+	}
+
+	mock.ExpectBegin()
+	mock.ExpectExec("INSERT(.+)").WillReturnResult(sqlmock.NewResult(5, 1))
+	mock.ExpectCommit()
+	fileID, err := file.Create()
+	asserts.NoError(err)
+	asserts.Equal(uint(5), fileID)
+	asserts.Equal(uint(5), file.ID)
+	asserts.NoError(mock.ExpectationsWereMet())
+
+	mock.ExpectBegin()
+	mock.ExpectExec("INSERT(.+)").WillReturnError(errors.New("error"))
+	mock.ExpectRollback()
+	fileID, err = file.Create()
+	asserts.Error(err)
+	asserts.NoError(mock.ExpectationsWereMet())
+
+}

models/init.go

@@ -52,7 +52,7 @@ func Init() {
 
 	//db.SetLogger(util.Log())
 	if err != nil {
-		util.Log().Panic("连接数据库不成功", err)
+		util.Log().Panic("连接数据库不成功, %s", err)
 	}
 
 	//设置连接池

models/migration.go

@@ -45,7 +45,7 @@ func migration() {
 	// 迁移完毕后写入版本锁 version.lock
 	err := conf.WriteVersionLock()
 	if err != nil {
-		util.Log().Warning("无法写入版本控制锁 version.lock, ", err)
+		util.Log().Warning("无法写入版本控制锁 version.lock, %s", err)
 	}
 
 }
@@ -66,7 +66,7 @@ func addDefaultPolicy() {
 			IsOriginLinkEnable: false,
 		}
 		if err := DB.Create(&defaultPolicy).Error; err != nil {
-			util.Log().Panic("无法创建初始存储策略, ", err)
+			util.Log().Panic("无法创建初始存储策略, %s", err)
 		}
 	}
 }