e6ba84ea50
Export will traverse a given PathInfo structure, and write the contents in NAR format to the passed Writer. It uses directoryLookupFn and blobLookupFn to resolve references. This is being moved over from nar-bridge. We need to keep the code there around until we can bump go.mod to storev1 with this merged, but the tests can already be moved entirely. Change-Id: Ie0de3077b09344cafa00ff1e2ddb8b52e9e631bc Reviewed-on: https://cl.tvl.fyi/c/depot/+/9602 Tested-by: BuildkiteCI Reviewed-by: Brian McGee <brian@bmcgee.ie> Autosubmit: flokli <flokli@flokli.de>
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package storev1
|
|
|
|
import (
|
|
"testing"
|
|
|
|
castorev1pb "code.tvl.fyi/tvix/castore/protos"
|
|
"github.com/google/go-cmp/cmp"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/protobuf/testing/protocmp"
|
|
)
|
|
|
|
func requireProtoEq(t *testing.T, expected interface{}, actual interface{}) {
|
|
if diff := cmp.Diff(expected, actual, protocmp.Transform()); diff != "" {
|
|
t.Errorf("unexpected difference:\n%v", diff)
|
|
}
|
|
}
|
|
|
|
func TestPopNextNode(t *testing.T) {
|
|
t.Run("empty directory", func(t *testing.T) {
|
|
d := &castorev1pb.Directory{
|
|
Directories: []*castorev1pb.DirectoryNode{},
|
|
Files: []*castorev1pb.FileNode{},
|
|
Symlinks: []*castorev1pb.SymlinkNode{},
|
|
}
|
|
|
|
n := drainNextNode(d)
|
|
require.Equal(t, nil, n)
|
|
})
|
|
t.Run("only directories", func(t *testing.T) {
|
|
ds := &castorev1pb.Directory{
|
|
Directories: []*castorev1pb.DirectoryNode{{
|
|
Name: []byte("a"),
|
|
Digest: []byte{},
|
|
Size: 0,
|
|
}, {
|
|
Name: []byte("b"),
|
|
Digest: []byte{},
|
|
Size: 0,
|
|
}},
|
|
Files: []*castorev1pb.FileNode{},
|
|
Symlinks: []*castorev1pb.SymlinkNode{},
|
|
}
|
|
|
|
n := drainNextNode(ds)
|
|
requireProtoEq(t, &castorev1pb.DirectoryNode{
|
|
Name: []byte("a"),
|
|
Digest: []byte{},
|
|
Size: 0,
|
|
}, n)
|
|
})
|
|
}
|