aboutsummaryrefslogtreecommitdiff
path: root/repository/traversal_test.go
blob: 61560fded39f2eabfa5626ac18fdd1add180a4b6 (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
package repository_test

import (
	"testing"

	"codeberg.org/lindenii/furgit/internal/testgit"
	"codeberg.org/lindenii/furgit/object"
	"codeberg.org/lindenii/furgit/objectid"
	"codeberg.org/lindenii/furgit/repository"
)

func TestRepositoryDepthFirstEnumerationFromHEAD(t *testing.T) {
	t.Parallel()

	testgit.ForEachAlgorithm(t, func(t *testing.T, algo objectid.Algorithm) { //nolint:thelper
		repoHarness := testgit.NewRepo(t, testgit.RepoOptions{
			ObjectFormat: algo,
			Bare:         true,
			RefFormat:    "files",
		})

		_, _, commit1 := repoHarness.MakeCommit(t, "walk-one")
		blob2, tree2 := repoHarness.MakeSingleFileTree(t, "second.txt", []byte("second\n"))
		commit2 := repoHarness.CommitTree(t, tree2, "walk-two", commit1)
		_ = blob2
		repoHarness.UpdateRef(t, "refs/heads/main", commit2)
		repoHarness.SymbolicRef(t, "HEAD", "refs/heads/main")

		repo, err := repository.Open(repoHarness.Dir())
		if err != nil {
			t.Fatalf("repository.Open: %v", err)
		}
		defer func() { _ = repo.Close() }()

		head, err := repo.ResolveRefFully("HEAD")
		if err != nil {
			t.Fatalf("ResolveRefFully(HEAD): %v", err)
		}

		visited := make(map[objectid.ObjectID]bool)
		queue := []objectid.ObjectID{head.ID}
		objectsRead := 0

		for len(queue) > 0 {
			id := queue[0]
			queue = queue[1:]

			if visited[id] {
				continue
			}
			visited[id] = true

			stored, err := repo.ReadStored(id)
			if err != nil {
				t.Fatalf("ReadStored(%s): %v", id, err)
			}
			objectsRead++

			switch obj := stored.Object().(type) {
			case *object.Commit:
				queue = append(queue, obj.Tree)
				queue = append(queue, obj.Parents...)
			case *object.Tree:
				for _, entry := range obj.Entries {
					queue = append(queue, entry.ID)
				}
			case *object.Tag:
				queue = append(queue, obj.Target)
			case *object.Blob:
			default:
				t.Fatalf("unexpected object type: %T", obj)
			}
		}

		if objectsRead == 0 {
			t.Fatalf("no objects were enumerated from HEAD")
		}
	})
}