mirror of https://github.com/Xhofe/alist
48 lines
925 B
Go
48 lines
925 B
Go
package url_tree_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/alist-org/alist/v3/drivers/url_tree"
|
|
)
|
|
|
|
func testTree() (*url_tree.Node, error) {
|
|
text := `folder1:
|
|
name1:https://url1
|
|
http://url2
|
|
folder2:
|
|
http://url3
|
|
http://url4
|
|
http://url5
|
|
folder3:
|
|
http://url6
|
|
http://url7
|
|
http://url8`
|
|
return url_tree.BuildTree(text, false)
|
|
}
|
|
|
|
func TestBuildTree(t *testing.T) {
|
|
node, err := testTree()
|
|
if err != nil {
|
|
t.Errorf("failed to build tree: %+v", err)
|
|
} else {
|
|
t.Logf("tree: %+v", node)
|
|
}
|
|
}
|
|
|
|
func TestGetNode(t *testing.T) {
|
|
root, err := testTree()
|
|
if err != nil {
|
|
t.Errorf("failed to build tree: %+v", err)
|
|
return
|
|
}
|
|
node := url_tree.GetNodeFromRootByPath(root, "/")
|
|
if node != root {
|
|
t.Errorf("got wrong node: %+v", node)
|
|
}
|
|
url3 := url_tree.GetNodeFromRootByPath(root, "/folder1/folder2/url3")
|
|
if url3 != root.Children[0].Children[2].Children[0] {
|
|
t.Errorf("got wrong node: %+v", url3)
|
|
}
|
|
}
|