mirror of
https://github.com/OpenListTeam/OpenList.git
synced 2026-10-10 13:03:09 +08:00
spark
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package teldrive
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/OpenListTeam/OpenList/v4/drivers/base"
|
|
"github.com/OpenListTeam/OpenList/v4/internal/driver"
|
|
"github.com/OpenListTeam/OpenList/v4/internal/model"
|
|
"github.com/go-resty/resty/v2"
|
|
)
|
|
|
|
// An empty directory makes the Teldrive API report totalPages == 0. List used
|
|
// to size pagesData by that value and then write pagesData[0], which panicked
|
|
// with "index out of range [0] with length 0" on every empty folder.
|
|
func TestListEmptyDir(t *testing.T) {
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = w.Write([]byte(`{"items":[],"meta":{"count":0,"totalPages":0,"currentPage":1}}`))
|
|
}))
|
|
defer srv.Close()
|
|
|
|
oldClient := base.RestyClient
|
|
base.RestyClient = resty.New()
|
|
defer func() { base.RestyClient = oldClient }()
|
|
|
|
d := &Teldrive{}
|
|
d.Address = srv.URL
|
|
|
|
objs, err := d.List(context.Background(), &model.Object{Path: "/"}, model.ListArgs{})
|
|
if err != nil {
|
|
t.Fatalf("List returned error: %v", err)
|
|
}
|
|
if len(objs) != 0 {
|
|
t.Fatalf("expected no entries for an empty dir, got %d", len(objs))
|
|
}
|
|
}
|
|
|
|
func TestListRootUsesConfiguredRootPath(t *testing.T) {
|
|
var (
|
|
mu sync.Mutex
|
|
paths []string
|
|
)
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
mu.Lock()
|
|
paths = append(paths, r.URL.Query().Get("path"))
|
|
mu.Unlock()
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = w.Write([]byte(`{"items":[{"id":"child","name":"child","type":"folder"}],"meta":{"count":1,"totalPages":1,"currentPage":1}}`))
|
|
}))
|
|
defer srv.Close()
|
|
|
|
oldClient := base.RestyClient
|
|
base.RestyClient = resty.New()
|
|
defer func() { base.RestyClient = oldClient }()
|
|
|
|
d := &Teldrive{
|
|
Addition: Addition{
|
|
RootPath: driver.RootPath{RootFolderPath: "/configured-root"},
|
|
},
|
|
}
|
|
d.Address = srv.URL
|
|
|
|
objs, err := d.List(context.Background(), &model.Object{}, model.ListArgs{})
|
|
if err != nil {
|
|
t.Fatalf("List returned error: %v", err)
|
|
}
|
|
|
|
mu.Lock()
|
|
defer mu.Unlock()
|
|
if len(paths) != 1 || paths[0] != "/configured-root" {
|
|
t.Fatalf("expected request path %q, got %q", "/configured-root", paths)
|
|
}
|
|
if len(objs) != 1 || objs[0].GetPath() != "/configured-root/child" {
|
|
t.Fatalf("expected child path %q, got %#v", "/configured-root/child", objs)
|
|
}
|
|
}
|