cloudreve/models/group_test.go

package model

import (
	"github.com/DATA-DOG/go-sqlmock"
	"github.com/jinzhu/gorm"
	"github.com/pkg/errors"
	"github.com/stretchr/testify/assert"
	"testing"
)

func TestGetGroupByID(t *testing.T) {
	asserts := assert.New(t)

	//找到用户组时
	groupRows := sqlmock.NewRows([]string{"id", "name", "policies"}).
		AddRow(1, "管理员", "[1]")
	mock.ExpectQuery("^SELECT (.+)").WillReturnRows(groupRows)

	group, err := GetGroupByID(1)
	asserts.NoError(err)
	asserts.Equal(Group{
		Model: gorm.Model{
			ID: 1,
		},
		Name:       "管理员",
		Policies:   "[1]",
		PolicyList: []uint{1},
	}, group)

	//未找到用户时
	mock.ExpectQuery("^SELECT (.+)").WillReturnError(errors.New("not found"))
	group, err = GetGroupByID(1)
	asserts.Error(err)
	asserts.Equal(Group{}, group)
}

func TestGroup_AfterFind(t *testing.T) {
	asserts := assert.New(t)

	testCase := Group{
		Model: gorm.Model{
			ID: 1,
		},
		Name:     "管理员",
		Policies: "[1]",
	}
	err := testCase.AfterFind()
	asserts.NoError(err)
	asserts.Equal(testCase.PolicyList, []uint{1})

	testCase.Policies = "[1,2,3,4,5]"
	err = testCase.AfterFind()
	asserts.NoError(err)
	asserts.Equal(testCase.PolicyList, []uint{1, 2, 3, 4, 5})

	testCase.Policies = "[1,2,3,4,5"
	err = testCase.AfterFind()
	asserts.Error(err)

	testCase.Policies = "[]"
	err = testCase.AfterFind()
	asserts.NoError(err)
	asserts.Equal(testCase.PolicyList, []uint{})
}

func TestGroup_GetAria2Option(t *testing.T) {
	asserts := assert.New(t)
	group := Group{}
	// 格式错误
	{
		group.Aria2Option = "123"
		asserts.Equal([3]bool{false, false, false}, group.GetAria2Option())
	}

	// 正常
	{
		group.Aria2Option = "1,1,0"
		asserts.Equal([3]bool{true, true, false}, group.GetAria2Option())
	}
}

func TestGroup_BeforeSave(t *testing.T) {
	asserts := assert.New(t)
	group := Group{
		PolicyList: []uint{1, 2, 3},
	}
	{
		err := group.BeforeSave()
		asserts.NoError(err)
		asserts.Equal("[1,2,3]", group.Policies)
	}

}