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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
package backup_test
import (
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/v16/internal/backup"
"gitlab.com/gitlab-org/gitaly/v16/internal/git"
"gitlab.com/gitlab-org/gitaly/v16/internal/git/catfile"
"gitlab.com/gitlab-org/gitaly/v16/internal/git/gittest"
"gitlab.com/gitlab-org/gitaly/v16/internal/git/localrepo"
"gitlab.com/gitlab-org/gitaly/v16/internal/gitaly/config"
"gitlab.com/gitlab-org/gitaly/v16/internal/gitaly/service/setup"
"gitlab.com/gitlab-org/gitaly/v16/internal/gitaly/storage/counter"
"gitlab.com/gitlab-org/gitaly/v16/internal/gitaly/transaction"
"gitlab.com/gitlab-org/gitaly/v16/internal/grpc/client"
"gitlab.com/gitlab-org/gitaly/v16/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/v16/internal/testhelper/testcfg"
"gitlab.com/gitlab-org/gitaly/v16/internal/testhelper/testserver"
"golang.org/x/exp/slices"
)
func removeHeadReference(refs []git.Reference) []git.Reference {
for i := range refs {
if refs[i].Name == "HEAD" {
return slices.Delete(refs, i, i+1)
}
}
return refs
}
func TestRemoteRepository_ResetRefs(t *testing.T) {
cfg := testcfg.Build(t)
testcfg.BuildGitalyHooks(t, cfg)
cfg.SocketPath = testserver.RunGitalyServer(t, cfg, setup.RegisterAll)
ctx := testhelper.Context(t)
repo, repoPath := gittest.CreateRepository(t, ctx, cfg, gittest.CreateRepositoryConfig{
RelativePath: "relative-path",
Storage: cfg.Storages[0],
})
pool := client.NewPool()
defer testhelper.MustClose(t, pool)
conn, err := pool.Dial(ctx, cfg.SocketPath, "")
require.NoError(t, err)
rr := backup.NewRemoteRepository(repo, conn)
// Create some commits
c0 := gittest.WriteCommit(t, cfg, repoPath, gittest.WithBranch("main"))
c1 := gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c0), gittest.WithBranch("main"))
c2 := gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c1), gittest.WithBranch("branch-1"))
// "Snapshot" the refs to pretend this is our backup.
backupRefState, err := rr.ListRefs(ctx)
require.NoError(t, err)
backupRefState = removeHeadReference(backupRefState)
// Create some more commits
gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c1), gittest.WithBranch("main"))
gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c2), gittest.WithBranch("branch-1"))
gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c2), gittest.WithBranch("branch-2"))
intermediateRefState, err := rr.ListRefs(ctx)
require.NoError(t, err)
require.Equal(t, 4, len(intermediateRefState)) // 3 branches + HEAD
// Reset the state of the refs to the backup.
require.NoError(t, rr.ResetRefs(ctx, backupRefState))
actualRefState, err := rr.ListRefs(ctx)
require.NoError(t, err)
actualRefState = removeHeadReference(actualRefState)
require.EqualValues(t, backupRefState, actualRefState)
}
func TestLocalRepository_ResetRefs(t *testing.T) {
if testhelper.IsPraefectEnabled() {
t.Skip("local backup manager expects to operate on the local filesystem so cannot operate through praefect")
}
cfg := testcfg.Build(t)
ctx := testhelper.Context(t)
repo, repoPath := gittest.CreateRepository(t, ctx, cfg, gittest.CreateRepositoryConfig{
SkipCreationViaService: true,
RelativePath: "relative-path",
Storage: cfg.Storages[0],
})
gitCmdFactory := gittest.NewCommandFactory(t, cfg)
txManager := transaction.NewTrackingManager()
repoCounter := counter.NewRepositoryCounter(cfg.Storages)
locator := config.NewLocator(cfg)
catfileCache := catfile.NewCache(cfg)
t.Cleanup(catfileCache.Stop)
lr := localrepo.New(testhelper.SharedLogger(t), locator, gitCmdFactory, catfileCache, repo)
localRepo := backup.NewLocalRepository(
testhelper.SharedLogger(t),
locator,
gitCmdFactory,
txManager,
repoCounter,
lr)
// Create some commits
c0 := gittest.WriteCommit(t, cfg, repoPath, gittest.WithBranch("main"))
c1 := gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c0), gittest.WithBranch("main"))
c2 := gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c1), gittest.WithBranch("branch-1"))
// "Snapshot" the refs to pretend this is our backup.
backupRefState, err := lr.GetReferences(ctx)
require.NoError(t, err)
// Create some more commits
gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c1), gittest.WithBranch("main"))
gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c2), gittest.WithBranch("branch-1"))
gittest.WriteCommit(t, cfg, repoPath, gittest.WithParents(c2), gittest.WithBranch("branch-2"))
intermediateRefState, err := lr.GetReferences(ctx)
require.NoError(t, err)
require.Equal(t, 3, len(intermediateRefState)) // 3 branches
// Reset the state of the refs to the backup.
require.NoError(t, localRepo.ResetRefs(ctx, backupRefState))
actualRefState, err := lr.GetReferences(ctx)
require.NoError(t, err)
require.EqualValues(t, backupRefState, actualRefState)
}
|
