Add fileStat to tpl/os/os

This commit is contained in:
satotake 2018-08-05 12:37:20 +09:00 committed by Bjørn Erik Pedersen
parent a6b1eb1e91
commit d71120852a
3 changed files with 60 additions and 0 deletions

View file

@ -55,6 +55,13 @@ func init() {
},
)
ns.AddMethodMapping(ctx.FileStat,
[]string{"fileStat"},
[][2]string{
{`{{ (fileExists "files/README.txt").Size }}`, `11`},
},
)
return ns
}

View file

@ -130,3 +130,22 @@ func (ns *Namespace) FileExists(i interface{}) (bool, error) {
return status, nil
}
// FileStat Stat returns the os.FileInfo structure describing file.
func (ns *Namespace) FileStat(i interface{}) (_os.FileInfo, error) {
path, err := cast.ToStringE(i)
if err != nil {
return nil, err
}
if path == "" {
return nil, errors.New("fileStat needs a path to a file")
}
r, err := ns.readFileFs.Stat(path)
if err != nil {
return nil, err
}
return r, nil
}

View file

@ -99,3 +99,37 @@ func TestFileExists(t *testing.T) {
assert.Equal(t, test.expect, result, errMsg)
}
}
func TestFileStat(t *testing.T) {
t.Parallel()
workingDir := "/home/hugo"
v := viper.New()
v.Set("workingDir", workingDir)
ns := New(&deps.Deps{Fs: hugofs.NewMem(v)})
afero.WriteFile(ns.deps.Fs.Source, filepath.Join(workingDir, "/f/f1.txt"), []byte("f1-content"), 0755)
for i, test := range []struct {
filename string
expect interface{}
}{
{filepath.FromSlash("/f/f1.txt"), int64(10)},
{filepath.FromSlash("f/f1.txt"), int64(10)},
{"b", nil},
{"", nil},
} {
errMsg := fmt.Sprintf("[%d] %v", i, test)
result, err := ns.FileStat(test.filename)
if test.expect == nil {
require.Error(t, err, errMsg)
continue
}
require.NoError(t, err, errMsg)
assert.Equal(t, test.expect, result.Size(), errMsg)
}
}