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
|
package git_test
import (
"strings"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/v16/internal/command"
"gitlab.com/gitlab-org/gitaly/v16/internal/git"
"gitlab.com/gitlab-org/gitaly/v16/internal/git/gittest"
"gitlab.com/gitlab-org/gitaly/v16/internal/grpc/metadata"
"gitlab.com/gitlab-org/gitaly/v16/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/v16/internal/testhelper/testcfg"
grpcmetadata "google.golang.org/grpc/metadata"
)
func TestWithRefHook(t *testing.T) {
t.Parallel()
ctx := testhelper.Context(t)
cfg := testcfg.Build(t)
repo, repoPath := gittest.CreateRepository(t, ctx, cfg, gittest.CreateRepositoryConfig{
SkipCreationViaService: true,
})
gittest.WriteCommit(t, cfg, repoPath, gittest.WithBranch("refs/heads/master"))
opt := git.WithRefTxHook(repo)
subCmd := git.Command{Name: "update-ref", Args: []string{"refs/heads/master", gittest.DefaultObjectHash.ZeroOID.String()}}
for _, tt := range []struct {
name string
fn func() (*command.Command, error)
}{
{
name: "NewCommand",
fn: func() (*command.Command, error) {
return gittest.NewCommandFactory(t, cfg, git.WithSkipHooks()).New(ctx, repo, subCmd, opt)
},
},
} {
t.Run(tt.name, func(t *testing.T) {
cmd, err := tt.fn()
require.NoError(t, err)
require.NoError(t, cmd.Wait())
var actualEnvVars []string
for _, env := range cmd.Env() {
kv := strings.SplitN(env, "=", 2)
require.Len(t, kv, 2)
key, val := kv[0], kv[1]
if strings.HasPrefix(key, "GL_") || strings.HasPrefix(key, "GITALY_") {
require.NotEmptyf(t, strings.TrimSpace(val),
"env var %s value should not be empty string", key)
actualEnvVars = append(actualEnvVars, key)
}
}
require.EqualValues(t, []string{
"GITALY_HOOKS_PAYLOAD",
"GITALY_LOG_DIR",
}, actualEnvVars)
})
}
}
func TestWithPackObjectsHookEnv(t *testing.T) {
t.Parallel()
ctx := testhelper.Context(t)
cfg := testcfg.Build(t)
cfg.PackObjectsCache.Enabled = true
repo, _ := gittest.CreateRepository(t, ctx, cfg, gittest.CreateRepositoryConfig{
SkipCreationViaService: true,
})
userID := "user-123"
username := "username"
protocol := "protocol"
remoteIP := "1.2.3.4"
opt := git.WithPackObjectsHookEnv(repo, protocol)
subCmd := git.Command{Name: "upload-pack", Args: []string{"a/b/c"}}
ctx = grpcmetadata.AppendToOutgoingContext(ctx, "user_id", userID, "username", username, "remote_ip", remoteIP)
ctx = metadata.OutgoingToIncoming(ctx)
cmd, err := gittest.NewCommandFactory(t, cfg, git.WithSkipHooks()).New(ctx, repo, subCmd, opt)
require.NoError(t, err)
payload, err := git.HooksPayloadFromEnv(cmd.Env())
require.NoError(t, err)
require.Equal(t, userID, payload.UserDetails.UserID)
require.Equal(t, username, payload.UserDetails.Username)
require.Equal(t, protocol, payload.UserDetails.Protocol)
require.Equal(t, remoteIP, payload.UserDetails.RemoteIP)
}
|
