50 lines
968 B
Go
50 lines
968 B
Go
package beego
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestBuildTemplate(t *testing.T) {
|
|
dir := "_beeTmp"
|
|
files := []string{
|
|
"1.tpl",
|
|
"2.html",
|
|
"3.htmltpl",
|
|
"4.mystyle",
|
|
}
|
|
if err := os.MkdirAll(dir, 0777); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, name := range files {
|
|
if _, err := os.Create(filepath.Join(dir, name)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
if err := BuildTemplate(dir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(BeeTemplates) != 1 {
|
|
t.Fatalf("should be 1 but got %v", len(BeeTemplates))
|
|
}
|
|
for _, v := range BeeTemplates {
|
|
if len(v.Templates()) != 3 {
|
|
t.Errorf("should be 3 but got %v", len(v.Templates()))
|
|
}
|
|
}
|
|
|
|
AddTemplateExt("mystyle")
|
|
if err := BuildTemplate(dir); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(BeeTemplates) != 1 {
|
|
t.Fatalf("should be 1 but got %v", len(BeeTemplates))
|
|
}
|
|
for _, v := range BeeTemplates {
|
|
if len(v.Templates()) != 4 {
|
|
t.Errorf("should be 4 but got %v", len(v.Templates()))
|
|
}
|
|
}
|
|
}
|