forked from mirror/pkger
52 lines
846 B
Go
52 lines
846 B
Go
|
package mem
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/markbates/pkger/here"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Test_Pkger_MarshalEmbed(t *testing.T) {
|
||
|
|
r := require.New(t)
|
||
|
|
|
||
|
|
info, err := here.Current()
|
||
|
|
r.NoError(err)
|
||
|
|
|
||
|
|
pkg, err := New(info)
|
||
|
|
r.NoError(err)
|
||
|
|
|
||
|
|
const N = 10
|
||
|
|
for i := 0; i < N; i++ {
|
||
|
|
name := fmt.Sprintf("/%d.txt", i)
|
||
|
|
f, err := pkg.Create(name)
|
||
|
|
r.NoError(err)
|
||
|
|
f.Write([]byte(name))
|
||
|
|
r.NoError(f.Close())
|
||
|
|
}
|
||
|
|
|
||
|
|
em, err := pkg.MarshalEmbed()
|
||
|
|
r.NoError(err)
|
||
|
|
|
||
|
|
p2, err := UnmarshalEmbed(em)
|
||
|
|
r.NoError(err)
|
||
|
|
|
||
|
|
pinfo, err := p2.Current()
|
||
|
|
r.NoError(err)
|
||
|
|
r.Equal(info, pinfo)
|
||
|
|
|
||
|
|
var act []os.FileInfo
|
||
|
|
err = p2.Walk("/", func(path string, info os.FileInfo, err error) error {
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
act = append(act, info)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
r.NoError(err)
|
||
|
|
r.Len(act, N+1) // +1 for the / dir
|
||
|
|
}
|