48 lignes
898 o
Go
48 lignes
898 o
Go
package filelist
|
|
|
|
import (
|
|
"github.com/NyaaPantsu/nyaa/model"
|
|
"html/template"
|
|
"testing"
|
|
)
|
|
|
|
func makeDummyFile(path ...string) (file model.File) {
|
|
file.SetPath(path)
|
|
return
|
|
}
|
|
|
|
func dashes(n uint) (out string) {
|
|
var i uint
|
|
for i = 0; i < n; i++ {
|
|
out += "-"
|
|
}
|
|
return
|
|
}
|
|
|
|
func TestFilelist(T *testing.T) {
|
|
files := []model.File{
|
|
makeDummyFile("A", "B", "C.txt"),
|
|
makeDummyFile("A", "C", "C.txt"),
|
|
makeDummyFile("B.txt"),
|
|
}
|
|
expected := "A\n" +
|
|
"-B\n" +
|
|
"--C.txt\n" +
|
|
"-C\n" +
|
|
"--C.txt\n" +
|
|
"B.txt\n"
|
|
|
|
filelist := FileListToFolder(files)
|
|
|
|
out, err := filelist.MakeFolderTreeView("{{dashes .NestLevel}}{{.FolderName}}\n", "{{dashes .NestLevel}}{{.Filename}}\n", map[string]interface{}{
|
|
"dashes": dashes,
|
|
}, nil)
|
|
if err != nil {
|
|
T.Fatalf("%v", err)
|
|
return
|
|
}
|
|
if out != template.HTML(expected) {
|
|
T.Fatalf("Error: expected %s, got %s", expected, out)
|
|
return
|
|
}
|
|
}
|