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())) } } }