func TestTemplate(t *testing.T) {
files := make(map[string]*file.File)
files["test_file.txt"] = &file.File{
Name: "test_file.txt",
Path: "static/test_file.txt",
Data: `[]byte("\x12\x34\x56\x78\x10")`,
}
dirs := new(dir.Dir)
dirs.Insert("static/")
tp := new(Template)
tp.Set("files")
tp.Variables = struct {
Pkg string
Files map[string]*file.File
Spread bool
DirList []string
}{
Pkg: "main",
Files: files,
Spread: false,
DirList: dirs.Clean(),
}
tmpl, err := tp.Exec()
assert.NoError(t, err)
assert.NotEmpty(t, tmpl)
s := string(tmpl)
assert.True(t, strings.Contains(s, `var FileStaticTestFileTxt = []byte("\x12\x34\x56\x78\x10")`))
assert.True(t, strings.Contains(s, `err = FS.Mkdir("static/", 0777)`))
assert.True(t, strings.Contains(s, `f, err = FS.OpenFile("static/test_file.txt", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)`))
// now with spread
tp.Set("file")
tp.Variables = struct {
Pkg string
Path string
Name string
Dir [][]string
Data string
}{
Pkg: "main",
Path: files["test_file.txt"].Path,
Name: files["test_file.txt"].Name,
Dir: dirs.List,
Data: files["test_file.txt"].Data,
}
tmpl, err = tp.Exec()
assert.NoError(t, err)
assert.NotEmpty(t, tmpl)
s = string(tmpl)
assert.True(t, strings.Contains(s, `var FileStaticTestFileTxt = []byte("\x12\x34\x56\x78\x10")`))
assert.True(t, strings.Contains(s, `f, err := FS.OpenFile("static/test_file.txt", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)`))
}
func TestTemplate(t *testing.T) {
var err error
files := make(map[string]*file.File)
files["test_file.txt"] = &file.File{
Name: "test_file.txt",
Path: "static/test_file.txt",
Data: `[]byte("\x12\x34\x56\x78\x10")`,
}
dirs := new(dir.Dir)
dirs.Insert("static/")
tp := new(Template)
err = tp.Set("ayy lmao")
assert.Error(t, err)
assert.Equal(t, `Error: Template must be "files" or "file"`, err.Error())
err = tp.Set("files")
assert.NoError(t, err)
assert.Equal(t, "files", tp.name)
defaultCompression := compression.NewGzip()
tp.Variables = struct {
Pkg string
Files map[string]*file.File
Spread bool
DirList []string
Compression *compression.Options
Debug bool
}{
Pkg: "main",
Files: files,
Spread: false,
DirList: dirs.Clean(),
Compression: defaultCompression.Options,
}
tp.template = "wrong {{.Err pudding"
tmpl, err := tp.Exec()
assert.Error(t, err)
assert.Empty(t, tmpl)
tp.template = "wrong{{if .Error}} pudding {{end}}"
tmpl, err = tp.Exec()
assert.Error(t, err)
assert.Empty(t, tmpl)
err = tp.Set("files")
tmpl, err = tp.Exec()
assert.NoError(t, err)
assert.NotEmpty(t, tmpl)
s := string(tmpl)
assert.True(t, strings.Contains(s, `var FileStaticTestFileTxt = []byte("\x12\x34\x56\x78\x10")`))
assert.True(t, strings.Contains(s, `err = FS.Mkdir("static/", 0777)`))
assert.True(t, strings.Contains(s, `f, err = FS.OpenFile("static/test_file.txt", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)`))
// now with spread
err = tp.Set("file")
assert.NoError(t, err)
assert.Equal(t, "file", tp.name)
defaultCompression = compression.NewGzip()
tp.Variables = struct {
Pkg string
Path string
Name string
Dir [][]string
Data string
Compression *compression.Options
}{
Pkg: "main",
Path: files["test_file.txt"].Path,
Name: files["test_file.txt"].Name,
Dir: dirs.List,
Data: files["test_file.txt"].Data,
Compression: defaultCompression.Options,
}
tmpl, err = tp.Exec()
assert.NoError(t, err)
assert.NotEmpty(t, tmpl)
s = string(tmpl)
assert.True(t, strings.Contains(s, `var FileStaticTestFileTxt = []byte("\x12\x34\x56\x78\x10")`))
assert.True(t, strings.Contains(s, `f, err := FS.OpenFile("static/test_file.txt", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)`))
}
