aboutsummaryrefslogtreecommitdiff
path: root/object/store/packed/writer_test.go
blob: 8227caa76e0be3a64ccf5d808c2ee3f8a61572bb (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
package packed_test

import (
	"bytes"
	"os"
	"testing"

	"lindenii.org/go/furgit/internal/testgit"
	"lindenii.org/go/furgit/object/id"
	"lindenii.org/go/furgit/object/store"
	"lindenii.org/go/furgit/object/store/packed"
	"lindenii.org/go/furgit/object/typ"
)

// TestWritePack verifies that writing a pack through the store
// makes its objects readable without a manual refresh.
func TestWritePack(t *testing.T) {
	t.Parallel()

	for _, objectFormat := range id.SupportedObjectFormats() {
		t.Run(objectFormat.String(), func(t *testing.T) {
			t.Parallel()

			repo, err := testgit.NewRepo(t, testgit.RepoOptions{ObjectFormat: objectFormat})
			if err != nil {
				t.Fatalf("NewRepo: %v", err)
			}

			seeded, err := repo.SeedHistory(t)
			if err != nil {
				t.Fatalf("SeedHistory: %v", err)
			}

			stream, err := repo.PackObjectsStdout(t, seeded.All(), testgit.PackObjectsStdoutOptions{
				Revs:    false,
				Thin:    false,
				Exclude: nil,
			})
			if err != nil {
				t.Fatalf("PackObjectsStdout: %v", err)
			}

			packedStore := openEmptyStore(t, objectFormat)

			err = packedStore.WritePack(bytes.NewReader(stream), store.PackWriteOptions{
				ThinBase: nil,
				Progress: nil,
			})
			if err != nil {
				t.Fatalf("WritePack: %v", err)
			}

			probes := []struct {
				ty  typ.Type
				oid id.ObjectID
			}{
				{typ.Blob, seeded.Blobs[0]},
				{typ.Tree, seeded.Trees[0]},
				{typ.Commit, seeded.Commits[len(seeded.Commits)-1]},
				{typ.Tag, seeded.Tags[0]},
			}

			for _, probe := range probes {
				want, err := repo.CatFile(t, probe.ty, probe.oid)
				if err != nil {
					t.Fatalf("CatFile(%s): %v", probe.oid, err)
				}

				ty, content, err := packedStore.ReadBytesContent(probe.oid)
				if err != nil {
					t.Fatalf("ReadBytesContent(%s): %v", probe.oid, err)
				}

				if ty != probe.ty {
					t.Fatalf("ReadBytesContent(%s) type = %v, want %v", probe.oid, ty, probe.ty)
				}

				if !bytes.Equal(content, want) {
					t.Fatalf("ReadBytesContent(%s) content mismatch", probe.oid)
				}
			}
		})
	}
}

// openEmptyStore opens a packed store over a fresh empty directory.
func openEmptyStore(t *testing.T, objectFormat id.ObjectFormat) *packed.Packed {
	t.Helper()

	root, err := os.OpenRoot(t.TempDir())
	if err != nil {
		t.Fatalf("OpenRoot: %v", err)
	}

	t.Cleanup(func() { _ = root.Close() })

	packedStore, err := packed.New(root, objectFormat)
	if err != nil {
		t.Fatalf("New: %v", err)
	}

	t.Cleanup(func() { _ = packedStore.Close() })

	return packedStore
}