mirror of
https://github.com/AlistGo/alist.git
synced 2025-11-25 19:37:41 +08:00
6024e8d832
* refactor:separate the setting method from the db package to the op package and add the cache
* refactor:separate the meta method from the db package to the op package
* fix:setting not load database data
* refactor:separate the user method from the db package to the op package
* refactor:remove user JoinPath error
* fix:op package user cache
* refactor:fs package list method
* fix:tile virtual paths (close #2743)
* Revert "refactor:remove user JoinPath error"
This reverts commit 4e20daaf9e.
* clean path directly may lead to unknown behavior
* fix: The path of the meta passed in must be prefix of reqPath
* chore: rename all virtualPath to mountPath
* fix: `getStoragesByPath` and `GetStorageVirtualFilesByPath`
is_sub_path:
/a/b isn't subpath of /a/bc
* fix: don't save setting if hook error
Co-authored-by: Noah Hsu <i@nn.ci>
92 lines
2.9 KiB
Go
92 lines
2.9 KiB
Go
package op_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/alist-org/alist/v3/internal/conf"
|
|
"github.com/alist-org/alist/v3/internal/db"
|
|
"github.com/alist-org/alist/v3/internal/model"
|
|
"github.com/alist-org/alist/v3/internal/op"
|
|
"github.com/alist-org/alist/v3/pkg/utils"
|
|
mapset "github.com/deckarep/golang-set/v2"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func init() {
|
|
dB, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
|
|
if err != nil {
|
|
panic("failed to connect database")
|
|
}
|
|
conf.Conf = conf.DefaultConfig()
|
|
db.Init(dB)
|
|
}
|
|
|
|
func TestCreateStorage(t *testing.T) {
|
|
var storages = []struct {
|
|
storage model.Storage
|
|
isErr bool
|
|
}{
|
|
{storage: model.Storage{Driver: "Local", MountPath: "/local", Addition: `{"root_folder_path":"."}`}, isErr: false},
|
|
{storage: model.Storage{Driver: "Local", MountPath: "/local", Addition: `{"root_folder_path":"."}`}, isErr: true},
|
|
{storage: model.Storage{Driver: "None", MountPath: "/none", Addition: `{"root_folder_path":"."}`}, isErr: true},
|
|
}
|
|
for _, storage := range storages {
|
|
_, err := op.CreateStorage(context.Background(), storage.storage)
|
|
if err != nil {
|
|
if !storage.isErr {
|
|
t.Errorf("failed to create storage: %+v", err)
|
|
} else {
|
|
t.Logf("expect failed to create storage: %+v", err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGetStorageVirtualFilesByPath(t *testing.T) {
|
|
setupStorages(t)
|
|
virtualFiles := op.GetStorageVirtualFilesByPath("/a")
|
|
var names []string
|
|
for _, virtualFile := range virtualFiles {
|
|
names = append(names, virtualFile.GetName())
|
|
}
|
|
var expectedNames = []string{"b", "c", "d"}
|
|
if utils.SliceEqual(names, expectedNames) {
|
|
t.Logf("passed")
|
|
} else {
|
|
t.Errorf("expected: %+v, got: %+v", expectedNames, names)
|
|
}
|
|
}
|
|
|
|
func TestGetBalancedStorage(t *testing.T) {
|
|
setupStorages(t)
|
|
set := mapset.NewSet[string]()
|
|
for i := 0; i < 5; i++ {
|
|
storage := op.GetBalancedStorage("/a/d/e1")
|
|
set.Add(storage.GetStorage().MountPath)
|
|
}
|
|
expected := mapset.NewSet([]string{"/a/d/e1", "/a/d/e1.balance"}...)
|
|
if !expected.Equal(set) {
|
|
t.Errorf("expected: %+v, got: %+v", expected, set)
|
|
}
|
|
}
|
|
|
|
func setupStorages(t *testing.T) {
|
|
var storages = []model.Storage{
|
|
{Driver: "Local", MountPath: "/a/b", Order: 0, Addition: `{"root_folder_path":"."}`},
|
|
{Driver: "Local", MountPath: "/adc", Order: 0, Addition: `{"root_folder_path":"."}`},
|
|
{Driver: "Local", MountPath: "/a/c", Order: 1, Addition: `{"root_folder_path":"."}`},
|
|
{Driver: "Local", MountPath: "/a/d", Order: 2, Addition: `{"root_folder_path":"."}`},
|
|
{Driver: "Local", MountPath: "/a/d/e1", Order: 3, Addition: `{"root_folder_path":"."}`},
|
|
{Driver: "Local", MountPath: "/a/d/e", Order: 4, Addition: `{"root_folder_path":"."}`},
|
|
{Driver: "Local", MountPath: "/a/d/e1.balance", Order: 4, Addition: `{"root_folder_path":"."}`},
|
|
}
|
|
for _, storage := range storages {
|
|
_, err := op.CreateStorage(context.Background(), storage)
|
|
if err != nil {
|
|
t.Fatalf("failed to create storage: %+v", err)
|
|
}
|
|
}
|
|
}
|