cloudreve/pkg/filesystem/path_test.go

package filesystem

import (
	"github.com/DATA-DOG/go-sqlmock"
	model "github.com/HFO4/cloudreve/models"
	"github.com/jinzhu/gorm"
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestFileSystem_IsFileExist(t *testing.T) {
	asserts := assert.New(t)
	fs := &FileSystem{User: &model.User{
		Model: gorm.Model{
			ID: 1,
		},
	}}

	// 存在
	{
		path := "/1.txt"
		// 根目录
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(1).
			WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1))
		mock.ExpectQuery("SELECT(.+)").WithArgs(1, "1.txt").WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "1.txt"))
		exist, file := fs.IsFileExist(path)
		asserts.NoError(mock.ExpectationsWereMet())
		asserts.True(exist)
		asserts.Equal(uint(1), file.ID)
	}

	// 文件不存在
	{
		path := "/1.txt"
		// 根目录
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(1).
			WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1))
		mock.ExpectQuery("SELECT(.+)").WithArgs(1, "1.txt").WillReturnRows(sqlmock.NewRows([]string{"id", "name"}))
		exist, _ := fs.IsFileExist(path)
		asserts.NoError(mock.ExpectationsWereMet())
		asserts.False(exist)
	}

	// 父目录不存在
	{
		path := "/1.txt"
		// 根目录
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(1).
			WillReturnRows(sqlmock.NewRows([]string{"id"}))
		exist, _ := fs.IsFileExist(path)
		asserts.NoError(mock.ExpectationsWereMet())
		asserts.False(exist)
	}
}

func TestFileSystem_IsPathExist(t *testing.T) {
	asserts := assert.New(t)
	fs := &FileSystem{User: &model.User{
		Model: gorm.Model{
			ID: 1,
		},
	}}

	// 查询根目录
	{
		path := "/"
		// 根目录
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(1).
			WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(1))
		exist, folder := fs.IsPathExist(path)
		asserts.NoError(mock.ExpectationsWereMet())
		asserts.True(exist)
		asserts.Equal(uint(1), folder.ID)
	}

	// 深层路径
	{
		path := "/1/2/3"
		// 根目录
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(1).
			WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1))
		// 1
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(1, 1, "1").
			WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(2, 1))
		// 2
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(2, 1, "2").
			WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(3, 1))
		// 3
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(3, 1, "3").
			WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(4, 1))
		exist, folder := fs.IsPathExist(path)
		asserts.NoError(mock.ExpectationsWereMet())
		asserts.True(exist)
		asserts.Equal(uint(4), folder.ID)
	}

	// 深层 不存在
	{
		path := "/1/2/3"
		// 根目录
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(1).
			WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(1, 1))
		// 1
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(1, 1, "1").
			WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(2, 1))
		// 2
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(2, 1, "2").
			WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}).AddRow(3, 1))
		// 3
		mock.ExpectQuery("SELECT(.+)").
			WithArgs(3, 1, "3").
			WillReturnRows(sqlmock.NewRows([]string{"id", "owner_id"}))
		exist, folder := fs.IsPathExist(path)
		asserts.NoError(mock.ExpectationsWereMet())
		asserts.False(exist)
		asserts.Nil(folder)
	}

}

func TestFileSystem_IsChildFileExist(t *testing.T) {
	asserts := assert.New(t)
	fs := &FileSystem{User: &model.User{
		Model: gorm.Model{
			ID: 1,
		},
	}}
	folder := model.Folder{
		Model:    gorm.Model{ID: 1},
		Name:     "123",
		Position: "/",
	}

	mock.ExpectQuery("SELECT(.+)").
		WithArgs(1, "321").
		WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(2, "321"))
	exist, childFile := fs.IsChildFileExist(&folder, "321")
	asserts.NoError(mock.ExpectationsWereMet())
	asserts.True(exist)
	asserts.Equal("/123/321", childFile.Position)
}