You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
120 lines
2.8 KiB
120 lines
2.8 KiB
package serializer
|
|
|
|
import (
|
|
"database/sql"
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
model "github.com/HFO4/cloudreve/models"
|
|
"github.com/HFO4/cloudreve/pkg/cache"
|
|
"github.com/duo-labs/webauthn/webauthn"
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
var mock sqlmock.Sqlmock
|
|
|
|
// TestMain 初始化数据库Mock
|
|
func TestMain(m *testing.M) {
|
|
var db *sql.DB
|
|
var err error
|
|
db, mock, err = sqlmock.New()
|
|
if err != nil {
|
|
panic("An error was not expected when opening a stub database connection")
|
|
}
|
|
model.DB, _ = gorm.Open("mysql", db)
|
|
defer db.Close()
|
|
m.Run()
|
|
}
|
|
|
|
func TestBuildUser(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
user := model.User{
|
|
Policy: model.Policy{MaxSize: 1024 * 1024},
|
|
}
|
|
mock.ExpectQuery("SELECT(.+)").WillReturnRows(sqlmock.NewRows([]string{"id"}))
|
|
res := BuildUser(user)
|
|
asserts.NoError(mock.ExpectationsWereMet())
|
|
asserts.Equal("1.00mb", res.Policy.MaxSize)
|
|
|
|
}
|
|
|
|
func TestBuildUserResponse(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
user := model.User{
|
|
Policy: model.Policy{MaxSize: 1024 * 1024},
|
|
}
|
|
res := BuildUserResponse(user)
|
|
asserts.Equal("1.00mb", res.Data.(User).Policy.MaxSize)
|
|
}
|
|
|
|
func TestBuildUserStorageResponse(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
cache.Set("pack_size_0", uint64(0), 0)
|
|
|
|
{
|
|
user := model.User{
|
|
Storage: 0,
|
|
Group: model.Group{MaxStorage: 10},
|
|
}
|
|
res := BuildUserStorageResponse(user)
|
|
asserts.Equal(uint64(0), res.Data.(storage).Used)
|
|
asserts.Equal(uint64(10), res.Data.(storage).Total)
|
|
asserts.Equal(uint64(10), res.Data.(storage).Free)
|
|
}
|
|
{
|
|
user := model.User{
|
|
Storage: 6,
|
|
Group: model.Group{MaxStorage: 10},
|
|
}
|
|
res := BuildUserStorageResponse(user)
|
|
asserts.Equal(uint64(6), res.Data.(storage).Used)
|
|
asserts.Equal(uint64(10), res.Data.(storage).Total)
|
|
asserts.Equal(uint64(4), res.Data.(storage).Free)
|
|
}
|
|
{
|
|
user := model.User{
|
|
Storage: 20,
|
|
Group: model.Group{MaxStorage: 10},
|
|
}
|
|
res := BuildUserStorageResponse(user)
|
|
asserts.Equal(uint64(20), res.Data.(storage).Used)
|
|
asserts.Equal(uint64(10), res.Data.(storage).Total)
|
|
asserts.Equal(uint64(0), res.Data.(storage).Free)
|
|
}
|
|
{
|
|
user := model.User{
|
|
Storage: 6,
|
|
Group: model.Group{MaxStorage: 10},
|
|
}
|
|
res := BuildUserStorageResponse(user)
|
|
asserts.Equal(uint64(6), res.Data.(storage).Used)
|
|
asserts.Equal(uint64(10), res.Data.(storage).Total)
|
|
asserts.Equal(uint64(4), res.Data.(storage).Free)
|
|
}
|
|
}
|
|
|
|
func TestBuildTagRes(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
tags := []model.Tag{
|
|
{
|
|
Type: 0,
|
|
Expression: "exp",
|
|
},
|
|
{
|
|
Type: 1,
|
|
Expression: "exp",
|
|
},
|
|
}
|
|
res := buildTagRes(tags)
|
|
asserts.Len(res, 2)
|
|
asserts.Equal("", res[0].Expression)
|
|
asserts.Equal("exp", res[1].Expression)
|
|
}
|
|
|
|
func TestBuildWebAuthnList(t *testing.T) {
|
|
asserts := assert.New(t)
|
|
credentials := []webauthn.Credential{{}}
|
|
res := BuildWebAuthnList(credentials)
|
|
asserts.Len(res, 1)
|
|
}
|