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
|
package objectpool
import (
"io/ioutil"
"os"
"path/filepath"
"strings"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/internal/git"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
)
func TestLink(t *testing.T) {
ctx, cancel := testhelper.Context()
defer cancel()
testRepo, _, cleanupFn := testhelper.NewTestRepo(t)
defer cleanupFn()
pool, err := NewObjectPool(testRepo.GetStorageName(), testhelper.NewTestObjectPoolName(t))
require.NoError(t, err)
require.NoError(t, pool.Remove(ctx), "make sure pool does not exist prior to creation")
require.NoError(t, pool.Create(ctx, testRepo), "create pool")
altPath, err := git.InfoAlternatesPath(testRepo)
require.NoError(t, err)
_, err = os.Stat(altPath)
require.True(t, os.IsNotExist(err))
require.NoError(t, pool.Link(ctx, testRepo))
require.FileExists(t, altPath, "alternates file must exist after Link")
content, err := ioutil.ReadFile(altPath)
require.NoError(t, err)
require.True(t, strings.HasPrefix(string(content), "../"), "expected %q to be relative path", content)
require.NoError(t, pool.Link(ctx, testRepo))
newContent, err := ioutil.ReadFile(altPath)
require.NoError(t, err)
require.Equal(t, content, newContent)
require.False(t, testhelper.RemoteExists(t, pool.FullPath(), testRepo.GetGlRepository()), "pool remotes should not include %v", testRepo)
}
func TestLinkRemoveBitmap(t *testing.T) {
ctx, cancel := testhelper.Context()
defer cancel()
testRepo, testRepoPath, cleanupFn := testhelper.NewTestRepo(t)
defer cleanupFn()
pool, err := NewObjectPool(testRepo.GetStorageName(), testhelper.NewTestObjectPoolName(t))
require.NoError(t, err)
require.NoError(t, pool.Init(ctx))
poolPath := pool.FullPath()
testhelper.MustRunCommand(t, nil, "git", "-C", poolPath, "fetch", testRepoPath, "+refs/*:refs/*")
testhelper.MustRunCommand(t, nil, "git", "-C", poolPath, "repack", "-adb")
require.Len(t, listBitmaps(t, pool.FullPath()), 1, "pool bitmaps before")
testhelper.MustRunCommand(t, nil, "git", "-C", testRepoPath, "repack", "-adb")
require.Len(t, listBitmaps(t, testRepoPath), 1, "member bitmaps before")
refsBefore := testhelper.MustRunCommand(t, nil, "git", "-C", testRepoPath, "for-each-ref")
require.NoError(t, pool.Link(ctx, testRepo))
require.Len(t, listBitmaps(t, pool.FullPath()), 1, "pool bitmaps after")
require.Len(t, listBitmaps(t, testRepoPath), 0, "member bitmaps after")
testhelper.MustRunCommand(t, nil, "git", "-C", testRepoPath, "fsck")
refsAfter := testhelper.MustRunCommand(t, nil, "git", "-C", testRepoPath, "for-each-ref")
require.Equal(t, refsBefore, refsAfter, "compare member refs before/after link")
}
func listBitmaps(t *testing.T, repoPath string) []string {
entries, err := ioutil.ReadDir(filepath.Join(repoPath, "objects/pack"))
require.NoError(t, err)
var bitmaps []string
for _, entry := range entries {
if strings.HasSuffix(entry.Name(), ".bitmap") {
bitmaps = append(bitmaps, entry.Name())
}
}
return bitmaps
}
func TestUnlink(t *testing.T) {
ctx, cancel := testhelper.Context()
defer cancel()
testRepo, _, cleanupFn := testhelper.NewTestRepo(t)
defer cleanupFn()
pool, err := NewObjectPool(testRepo.GetStorageName(), t.Name())
require.NoError(t, err)
defer pool.Remove(ctx)
require.Error(t, pool.Unlink(ctx, testRepo), "removing a non-existing pool should be an error")
require.NoError(t, pool.Create(ctx, testRepo), "create pool")
require.NoError(t, pool.Link(ctx, testRepo), "link test repo to pool")
require.False(t, testhelper.RemoteExists(t, pool.FullPath(), testRepo.GetGlRepository()), "pool remotes should include %v", testRepo)
require.NoError(t, pool.Unlink(ctx, testRepo), "unlink repo")
require.False(t, testhelper.RemoteExists(t, pool.FullPath(), testRepo.GetGlRepository()), "pool remotes should no longer include %v", testRepo)
}
|
