Codebase list golang-github-gobuffalo-packr / cbdaebd v2 / walk_test.go
cbdaebd

Tree @cbdaebd (Download .tar.gz)

walk_test.go @cbdaebd — raw · history · blame 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package packr

import (
	"path/filepath"
	"testing"

	"github.com/gobuffalo/packr/v2/file"
	"github.com/stretchr/testify/require"
)

func Test_Box_Walk(t *testing.T) {
	r := require.New(t)

	box := NewBox(filepath.Join("_fixtures", "list_test"))
	r.NoError(box.AddString(filepath.Join("d", "d.txt"), "D"))

	var act []string
	r.NoError(box.Walk(func(path string, f file.File) error {
		act = append(act, path)
		return nil
	}))
	exp := []string{"a.txt", filepath.Join("b", "b.txt"), filepath.Join("b", "b2.txt"), filepath.Join("c", "c.txt"), filepath.Join("d", "d.txt")}
	r.Equal(exp, act)
}

func Test_Box_WalkPrefix(t *testing.T) {
	r := require.New(t)

	box := NewBox(filepath.Join("_fixtures", "list_test"))
	r.NoError(box.AddString(filepath.Join("d", "d.txt"), "D"))

	var act []string
	r.NoError(box.WalkPrefix("b/", func(path string, f file.File) error {
		act = append(act, path)
		return nil
	}))
	exp := []string{filepath.Join("b", "b.txt"), filepath.Join("b", "b2.txt")}
	r.Equal(exp, act)
}