mirror of https://github.com/markbates/pkger.git
70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
|
package memfs
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io"
|
||
|
|
"os"
|
||
|
|
"sort"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/markbates/pkger/here"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_Walk(t *testing.T) {
|
||
|
|
r := require.New(t)
|
||
|
|
|
||
|
|
files := []struct {
|
||
|
|
name string
|
||
|
|
body string
|
||
|
|
}{
|
||
|
|
{name: "/a/a.txt", body: "A"},
|
||
|
|
{name: "/a/a.md", body: "Amd"},
|
||
|
|
{name: "/b/c/d.txt", body: "B"},
|
||
|
|
{name: "/f.txt", body: "F"},
|
||
|
|
}
|
||
|
|
|
||
|
|
sort.Slice(files, func(a, b int) bool {
|
||
|
|
return files[a].name < files[b].name
|
||
|
|
})
|
||
|
|
|
||
|
|
fs, err := New(here.Info{})
|
||
|
|
r.NoError(err)
|
||
|
|
|
||
|
|
for _, file := range files {
|
||
|
|
f, err := fs.Create(file.name)
|
||
|
|
r.NoError(err)
|
||
|
|
_, err = io.Copy(f, strings.NewReader(file.body))
|
||
|
|
r.NoError(err)
|
||
|
|
r.NoError(f.Close())
|
||
|
|
}
|
||
|
|
|
||
|
|
var found []string
|
||
|
|
err = fs.Walk("/", func(path string, info os.FileInfo, err error) error {
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
found = append(found, path)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
r.NoError(err)
|
||
|
|
|
||
|
|
expected := []string{":/", ":/a", ":/a/a.md", ":/a/a.txt", ":/b", ":/b/c", ":/b/c/d.txt", ":/f.txt"}
|
||
|
|
r.Equal(expected, found)
|
||
|
|
|
||
|
|
found = []string{}
|
||
|
|
err = fs.Walk("/a/", func(path string, info os.FileInfo, err error) error {
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
found = append(found, path)
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
r.NoError(err)
|
||
|
|
|
||
|
|
expected = []string{":/a/a.md", ":/a/a.txt"}
|
||
|
|
r.Equal(expected, found)
|
||
|
|
}
|