head_test.go RawHistoryFolder

1package snapshotcodec
2
3import (
4	"errors"
5	"testing"
6)
7
8func TestHeadCodecIsDeterministicAndStrict(t *testing.T) {
9	head := Head{Generation: 7, ManifestDigest: SnapshotDigest{1}}
10	encoded, err := MarshalHead(head)
11	if err != nil {
12		t.Fatal(err)
13	}
14	encodedAgain, err := MarshalHead(head)
15	if err != nil {
16		t.Fatal(err)
17	}
18	if string(encoded) != string(encodedAgain) {
19		t.Fatal("head encoding is not deterministic")
20	}
21	got, err := UnmarshalHead(encoded)
22	if err != nil || got != head {
23		t.Fatalf("head round trip = %#v, %v", got, err)
24	}
25	for n := range len(encoded) {
26		if _, err := UnmarshalHead(encoded[:n]); err == nil {
27			t.Fatalf("prefix %d accepted", n)
28		}
29	}
30	if _, err := UnmarshalHead(append(encoded, 0)); !errors.Is(err, ErrTrailingBytes) {
31		t.Fatalf("trailing data error = %v", err)
32	}
33}
34
35func TestHeadCodecRejectsInvalidDeclarations(t *testing.T) {
36	for _, head := range []Head{
37		{ManifestDigest: SnapshotDigest{1}},
38		{Generation: 1},
39	} {
40		if _, err := MarshalHead(head); !errors.Is(err, ErrInvalidData) {
41			t.Fatalf("MarshalHead(%#v) error = %v, want invalid data", head, err)
42		}
43	}
44}