shad-go/distbuild/pkg/tarstream/stream_test.go

66 lines
1.5 KiB
Go
Raw Normal View History

2020-03-27 22:52:54 +00:00
package tarstream
import (
"bytes"
"io/ioutil"
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
2020-04-11 20:34:14 +00:00
"golang.org/x/sys/unix"
2020-03-27 22:52:54 +00:00
)
func TestTarStream(t *testing.T) {
tmpDir, err := ioutil.TempDir("", "tarstream")
require.NoError(t, err)
t.Logf("running inside %s", tmpDir)
from := filepath.Join(tmpDir, "from")
to := filepath.Join(tmpDir, "to")
require.NoError(t, os.Mkdir(from, 0777))
require.NoError(t, os.Mkdir(to, 0777))
var buf bytes.Buffer
require.NoError(t, os.Mkdir(filepath.Join(from, "a"), 0777))
require.NoError(t, os.MkdirAll(filepath.Join(from, "b", "c", "d"), 0777))
require.NoError(t, ioutil.WriteFile(filepath.Join(from, "a", "x.bin"), []byte("xxx"), 0777))
require.NoError(t, ioutil.WriteFile(filepath.Join(from, "b", "c", "y.txt"), []byte("yyy"), 0666))
require.NoError(t, Send(from, &buf))
require.NoError(t, Receive(to, &buf))
checkDir := func(path string) {
st, err := os.Stat(path)
require.NoError(t, err)
require.True(t, st.IsDir())
}
checkDir(filepath.Join(to, "a"))
checkDir(filepath.Join(to, "b", "c", "d"))
checkFile := func(path string, content []byte, mode os.FileMode) {
t.Helper()
st, err := os.Stat(path)
require.NoError(t, err)
require.Equal(t, mode.String(), st.Mode().String())
b, err := ioutil.ReadFile(path)
require.NoError(t, err)
require.Equal(t, content, b)
}
2020-04-14 11:22:08 +00:00
checkFile(filepath.Join(to, "a", "x.bin"), []byte("xxx"), 0755)
checkFile(filepath.Join(to, "b", "c", "y.txt"), []byte("yyy"), 0644)
2020-03-27 22:52:54 +00:00
}
2020-04-11 20:34:14 +00:00
func init() {
unix.Umask(0022)
}