2a9852fa by miraclesu

Add template test

1 parent 6fbdbaae
1 package beego
2
3 import (
4 "os"
5 "path/filepath"
6 "testing"
7 )
8
9 func TestBuildTemplate(t *testing.T) {
10 dir := "_beeTmp"
11 files := []string{
12 "1.tpl",
13 "2.html",
14 "3.htmltpl",
15 "4.mystyle",
16 }
17 if err := os.MkdirAll(dir, 0777); err != nil {
18 t.Fatal(err)
19 }
20 for _, name := range files {
21 if _, err := os.Create(filepath.Join(dir, name)); err != nil {
22 t.Fatal(err)
23 }
24 }
25 if err := BuildTemplate(dir); err != nil {
26 t.Fatal(err)
27 }
28 if len(BeeTemplates) != 1 {
29 t.Fatalf("should be 1 but got %v", len(BeeTemplates))
30 }
31 for _, v := range BeeTemplates {
32 if len(v.Templates()) != 3 {
33 t.Errorf("should be 3 but got %v", len(v.Templates()))
34 }
35 }
36
37 AddTemplateExt("mystyle")
38 if err := BuildTemplate(dir); err != nil {
39 t.Fatal(err)
40 }
41 if len(BeeTemplates) != 1 {
42 t.Fatalf("should be 1 but got %v", len(BeeTemplates))
43 }
44 for _, v := range BeeTemplates {
45 if len(v.Templates()) != 4 {
46 t.Errorf("should be 4 but got %v", len(v.Templates()))
47 }
48 }
49 }
Styling with Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!