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
|
package main
import (
"bytes"
"encoding/json"
"flag"
"fmt"
"io"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/v14/internal/git/gittest"
"gitlab.com/gitlab-org/gitaly/v14/internal/gitaly/service/setup"
"gitlab.com/gitlab-org/gitaly/v14/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/v14/internal/testhelper/testcfg"
"gitlab.com/gitlab-org/gitaly/v14/internal/testhelper/testserver"
"gitlab.com/gitlab-org/gitaly/v14/proto/go/gitalypb"
)
func TestCreateSubcommand(t *testing.T) {
cfg := testcfg.Build(t)
cfg.SocketPath = testserver.RunGitalyServer(t, cfg, nil, setup.RegisterAll)
ctx := testhelper.Context(t)
path := testhelper.TempDir(t)
var repos []*gitalypb.Repository
for i := 0; i < 5; i++ {
repo, _ := gittest.CreateRepository(ctx, t, cfg, gittest.CreateRepositoryConfig{
Seed: gittest.SeedGitLabTest,
RelativePath: fmt.Sprintf("repo-%d", i),
})
repos = append(repos, repo)
}
var stdin bytes.Buffer
encoder := json.NewEncoder(&stdin)
for _, repo := range repos {
require.NoError(t, encoder.Encode(map[string]string{
"address": cfg.SocketPath,
"token": cfg.Auth.Token,
"storage_name": repo.StorageName,
"relative_path": repo.RelativePath,
"gl_project_path": repo.GlProjectPath,
}))
}
require.NoError(t, encoder.Encode(map[string]string{
"address": "invalid",
"token": "invalid",
"relative_path": "invalid",
}))
cmd := createSubcommand{backupPath: path}
fs := flag.NewFlagSet("create", flag.ContinueOnError)
cmd.Flags(fs)
require.NoError(t, fs.Parse([]string{"-path", path}))
require.EqualError(t,
cmd.Run(ctx, &stdin, io.Discard),
"create: pipeline: 1 failures encountered:\n - invalid: manager: isEmpty: could not dial source: invalid connection string: \"invalid\"\n")
for _, repo := range repos {
bundlePath := filepath.Join(path, repo.RelativePath+".bundle")
require.FileExists(t, bundlePath)
}
}
|