732dc68727
Change-Id: I5eaadc837a4d3a7f635574437127a22de88f556b Reviewed-on: https://cl.tvl.fyi/c/depot/+/9407 Tested-by: BuildkiteCI Reviewed-by: Connor Brewster <cbrewster@hey.com> Autosubmit: flokli <flokli@flokli.de>
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package writer
|
|
|
|
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)
|
|
})
|
|
}
|