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
|
package main
import (
"bytes"
"errors"
"fmt"
"testing"
"time"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/internal/praefect/config"
"gitlab.com/gitlab-org/gitaly/internal/praefect/datastore"
"gitlab.com/gitlab-org/gitaly/internal/praefect/service/info"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/internal/testhelper/testdb"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
func TestMetadataSubcommand(t *testing.T) {
t.Parallel()
ctx := testhelper.Context(t)
tx := testdb.New(t).Begin(t)
defer tx.Rollback(t)
testdb.SetHealthyNodes(t, ctx, tx, map[string]map[string][]string{
"praefect": {"virtual-storage": {"primary", "secondary-1"}},
})
rs := datastore.NewPostgresRepositoryStore(tx, map[string][]string{
"virtual-storage": {"primary", "secondary-1", "secondary-2"},
})
require.NoError(t, rs.CreateRepository(ctx, 1, "virtual-storage", "relative-path", "replica-path", "primary", []string{"secondary-1"}, []string{"secondary-2"}, true, true))
require.NoError(t, rs.IncrementGeneration(ctx, 1, "primary", nil))
_, err := tx.ExecContext(ctx, "UPDATE storage_repositories SET verified_at = $1 WHERE storage = 'primary'",
time.Date(2021, time.April, 1, 10, 4, 20, 64, time.UTC),
)
require.NoError(t, err)
ln, clean := listenAndServe(t, []svcRegistrar{
registerPraefectInfoServer(info.NewServer(config.Config{}, rs, nil, nil, nil)),
})
defer clean()
for _, tc := range []struct {
desc string
args []string
error error
}{
{
desc: "missing parameters fails",
error: errors.New("repository id or virtual storage and relative path required"),
},
{
desc: "repository id with virtual storage fails",
args: []string{"-repository-id=1", "-virtual-storage=virtual-storage"},
error: errors.New("virtual storage and relative path can't be provided with a repository ID"),
},
{
desc: "repository id with relative path fails",
args: []string{"-repository-id=1", "-relative-path=relative-path"},
error: errors.New("virtual storage and relative path can't be provided with a repository ID"),
},
{
desc: "virtual storage without relative path fails",
args: []string{"-virtual-storage=virtual-storage"},
error: errors.New("relative path is required with virtual storage"),
},
{
desc: "relative path without virtual storage fails",
args: []string{"-relative-path=relative-path"},
error: errors.New("virtual storage is required with relative path"),
},
{
desc: "repository not found",
args: []string{"-repository-id=2"},
error: fmt.Errorf("get metadata: %w", status.Error(codes.NotFound, "repository not found")),
},
{
desc: "repository found with repository id",
args: []string{"-repository-id=1"},
},
{
desc: "repository found with virtual storage and relative path",
args: []string{"-virtual-storage=virtual-storage", "-relative-path=relative-path"},
},
} {
t.Run(tc.desc, func(t *testing.T) {
stdout := &bytes.Buffer{}
cmd := newMetadataSubcommand(stdout)
fs := cmd.FlagSet()
require.NoError(t, fs.Parse(tc.args))
err := cmd.Exec(fs, config.Config{SocketPath: ln.Addr().String()})
testhelper.RequireGrpcError(t, tc.error, err)
if tc.error != nil {
return
}
require.Equal(t, `Repository ID: 1
Virtual Storage: "virtual-storage"
Relative Path: "relative-path"
Replica Path: "replica-path"
Primary: "primary"
Generation: 1
Replicas:
- Storage: "primary"
Assigned: true
Generation: 1, fully up to date
Healthy: true
Valid Primary: true
Verified At: 2021-04-01 10:04:20 +0000 UTC
- Storage: "secondary-1"
Assigned: true
Generation: 0, behind by 1 changes
Healthy: true
Valid Primary: false
Verified At: unverified
- Storage: "secondary-2"
Assigned: true
Generation: replica not yet created
Healthy: false
Valid Primary: false
Verified At: unverified
`, stdout.String())
})
}
}
|