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
|
package repository
import (
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/internal/git"
"gitlab.com/gitlab-org/gitaly/internal/git/catfile"
"gitlab.com/gitlab-org/gitaly/internal/git/gittest"
"gitlab.com/gitlab-org/gitaly/internal/git/objectpool"
"gitlab.com/gitlab-org/gitaly/internal/gitaly/config"
"gitlab.com/gitlab-org/gitaly/internal/helper/text"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/proto/go/gitalypb"
"google.golang.org/grpc/metadata"
)
func newTestObjectPool(t *testing.T, cfg config.Cfg) (*objectpool.ObjectPool, *gitalypb.Repository) {
t.Helper()
relativePath := gittest.NewObjectPoolName(t)
repo := gittest.InitRepoDir(t, cfg.Storages[0].Path, relativePath)
gitCmdFactory := git.NewExecCommandFactory(cfg)
pool, err := objectpool.NewObjectPool(
cfg,
config.NewLocator(cfg),
gitCmdFactory,
catfile.NewCache(gitCmdFactory, cfg),
repo.GetStorageName(),
relativePath,
)
require.NoError(t, err)
return pool, repo
}
// getForkDestination creates a repo struct and path, but does not actually create the directory
func getForkDestination(t *testing.T, storage config.Storage) (*gitalypb.Repository, string, func()) {
t.Helper()
folder, err := text.RandomHex(20)
require.NoError(t, err)
forkRepoPath := filepath.Join(storage.Path, folder)
forkedRepo := &gitalypb.Repository{StorageName: storage.Name, RelativePath: folder, GlRepository: "project-1"}
return forkedRepo, forkRepoPath, func() { os.RemoveAll(forkRepoPath) }
}
func TestCloneFromPoolInternal(t *testing.T) {
cfg, repo, repoPath, client := setupRepositoryService(t)
ctxOuter, cancel := testhelper.Context()
defer cancel()
md := testhelper.GitalyServersMetadataFromCfg(t, cfg)
ctx := metadata.NewOutgoingContext(ctxOuter, md)
pool, poolRepo := newTestObjectPool(t, cfg)
defer func() {
require.NoError(t, pool.Remove(ctx))
}()
require.NoError(t, pool.Create(ctx, repo))
require.NoError(t, pool.Link(ctx, repo))
fullRepack(t, cfg, repoPath)
gittest.WriteCommit(t, cfg, repoPath, gittest.WithBranch("branch"))
forkedRepo, forkRepoPath, forkRepoCleanup := getForkDestination(t, cfg.Storages[0])
defer forkRepoCleanup()
req := &gitalypb.CloneFromPoolInternalRequest{
Repository: forkedRepo,
SourceRepository: repo,
Pool: &gitalypb.ObjectPool{
Repository: poolRepo,
},
}
_, err := client.CloneFromPoolInternal(ctx, req)
require.NoError(t, err)
assert.True(t, gittest.GetGitObjectDirSize(t, forkRepoPath) < 100)
isLinked, err := pool.LinkedToRepository(repo)
require.NoError(t, err)
require.True(t, isLinked)
// feature is a branch known to exist in the source repository. By looking it up in the target
// we establish that the target has branches, even though (as we saw above) it has no objects.
gittest.Exec(t, cfg, "-C", forkRepoPath, "show-ref", "--verify", "refs/heads/feature")
gittest.Exec(t, cfg, "-C", forkRepoPath, "show-ref", "--verify", "refs/heads/branch")
}
// fullRepack does a full repack on the repository, which means if it has a pool repository linked, it will get rid of redundant objects that are reachable in the pool
func fullRepack(t *testing.T, cfg config.Cfg, repoPath string) {
gittest.Exec(t, cfg, "-C", repoPath, "repack", "-A", "-l", "-d")
}
|
