diff options
Diffstat (limited to 'tvix/castore/protos/castore_test.go')
-rw-r--r-- | tvix/castore/protos/castore_test.go | 298 |
1 files changed, 0 insertions, 298 deletions
diff --git a/tvix/castore/protos/castore_test.go b/tvix/castore/protos/castore_test.go deleted file mode 100644 index fda87a6cfb66..000000000000 --- a/tvix/castore/protos/castore_test.go +++ /dev/null @@ -1,298 +0,0 @@ -package castorev1_test - -import ( - "testing" - - castorev1pb "code.tvl.fyi/tvix/castore/protos" - "github.com/stretchr/testify/assert" -) - -var ( - dummyDigest = []byte{ - 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, - 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, - 0x00, 0x00, 0x00, 0x00, - } -) - -func TestDirectorySize(t *testing.T) { - t.Run("empty", func(t *testing.T) { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - assert.Equal(t, uint32(0), d.Size()) - }) - - t.Run("containing single empty directory", func(t *testing.T) { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{{ - Name: []byte([]byte("foo")), - Digest: dummyDigest, - Size: 0, - }}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - assert.Equal(t, uint32(1), d.Size()) - }) - - t.Run("containing single non-empty directory", func(t *testing.T) { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{{ - Name: []byte("foo"), - Digest: dummyDigest, - Size: 4, - }}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - assert.Equal(t, uint32(5), d.Size()) - }) - - t.Run("containing single file", func(t *testing.T) { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{{ - Name: []byte("foo"), - Digest: dummyDigest, - Size: 42, - Executable: false, - }}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - assert.Equal(t, uint32(1), d.Size()) - }) - - t.Run("containing single symlink", func(t *testing.T) { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{{ - Name: []byte("foo"), - Target: []byte("bar"), - }}, - } - - assert.Equal(t, uint32(1), d.Size()) - }) - -} -func TestDirectoryDigest(t *testing.T) { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - dgst, err := d.Digest() - assert.NoError(t, err, "calling Digest() on a directory shouldn't error") - assert.Equal(t, []byte{ - 0xaf, 0x13, 0x49, 0xb9, 0xf5, 0xf9, 0xa1, 0xa6, 0xa0, 0x40, 0x4d, 0xea, 0x36, 0xdc, - 0xc9, 0x49, 0x9b, 0xcb, 0x25, 0xc9, 0xad, 0xc1, 0x12, 0xb7, 0xcc, 0x9a, 0x93, 0xca, - 0xe4, 0x1f, 0x32, 0x62, - }, dgst) -} - -func TestDirectoryValidate(t *testing.T) { - t.Run("empty", func(t *testing.T) { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - assert.NoError(t, d.Validate()) - }) - - t.Run("invalid names", func(t *testing.T) { - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{{ - Name: []byte{}, - Digest: dummyDigest, - Size: 42, - }}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - assert.ErrorContains(t, d.Validate(), "invalid node name") - } - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{{ - Name: []byte("."), - Digest: dummyDigest, - Size: 42, - }}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - assert.ErrorContains(t, d.Validate(), "invalid node name") - } - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{{ - Name: []byte(".."), - Digest: dummyDigest, - Size: 42, - Executable: false, - }}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - assert.ErrorContains(t, d.Validate(), "invalid node name") - } - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{{ - Name: []byte("\x00"), - Target: []byte("foo"), - }}, - } - - assert.ErrorContains(t, d.Validate(), "invalid node name") - } - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{{ - Name: []byte("foo/bar"), - Target: []byte("foo"), - }}, - } - - assert.ErrorContains(t, d.Validate(), "invalid node name") - } - }) - - t.Run("invalid digest", func(t *testing.T) { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{{ - Name: []byte("foo"), - Digest: nil, - Size: 42, - }}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - - assert.ErrorContains(t, d.Validate(), "invalid digest length") - }) - - t.Run("invalid symlink targets", func(t *testing.T) { - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{{ - Name: []byte("foo"), - Target: []byte{}, - }}, - } - - assert.ErrorContains(t, d.Validate(), "invalid symlink target") - } - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{{ - Name: []byte("foo"), - Target: []byte{0x66, 0x6f, 0x6f, 0}, - }}, - } - - assert.ErrorContains(t, d.Validate(), "invalid symlink target") - } - }) - - t.Run("sorting", func(t *testing.T) { - // "b" comes before "a", bad. - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{{ - Name: []byte("b"), - Digest: dummyDigest, - Size: 42, - }, { - Name: []byte("a"), - Digest: dummyDigest, - Size: 42, - }}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - assert.ErrorContains(t, d.Validate(), "is not in sorted order") - } - - // "a" exists twice, bad. - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{{ - Name: []byte("a"), - Digest: dummyDigest, - Size: 42, - }}, - Files: []*castorev1pb.FileNode{{ - Name: []byte("a"), - Digest: dummyDigest, - Size: 42, - Executable: false, - }}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - assert.ErrorContains(t, d.Validate(), "duplicate name") - } - - // "a" comes before "b", all good. - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{{ - Name: []byte("a"), - Digest: dummyDigest, - Size: 42, - }, { - Name: []byte("b"), - Digest: dummyDigest, - Size: 42, - }}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{}, - } - assert.NoError(t, d.Validate(), "shouldn't error") - } - - // [b, c] and [a] are both properly sorted. - { - d := castorev1pb.Directory{ - Directories: []*castorev1pb.DirectoryNode{{ - Name: []byte("b"), - Digest: dummyDigest, - Size: 42, - }, { - Name: []byte("c"), - Digest: dummyDigest, - Size: 42, - }}, - Files: []*castorev1pb.FileNode{}, - Symlinks: []*castorev1pb.SymlinkNode{{ - Name: []byte("a"), - Target: []byte("foo"), - }}, - } - assert.NoError(t, d.Validate(), "shouldn't error") - } - }) -} |