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
|
package gittest
import (
"fmt"
"io/ioutil"
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/v14/internal/git/hooks"
"gitlab.com/gitlab-org/gitaly/v14/internal/testhelper"
)
// WriteEnvToCustomHook dumps the env vars that the custom hooks receives to a file
func WriteEnvToCustomHook(t testing.TB, repoPath, hookName string) string {
tempDir := testhelper.TempDir(t)
outputPath := filepath.Join(tempDir, "hook.env")
hookContent := fmt.Sprintf("#!/bin/sh\n/usr/bin/env >%s\n", outputPath)
WriteCustomHook(t, repoPath, hookName, []byte(hookContent))
return outputPath
}
// WriteCheckNewObjectExistsHook writes a pre-receive hook which only succeeds
// if it can find the object in the quarantine directory. if
// GIT_OBJECT_DIRECTORY and GIT_ALTERNATE_OBJECT_DIRECTORIES were not passed
// through correctly to the hooks, it will fail
func WriteCheckNewObjectExistsHook(t testing.TB, gitBin, repoPath string) {
hook := fmt.Sprintf(`#!/usr/bin/env ruby
STDIN.each_line do |line|
new_object = line.split(' ')[1]
exit 1 unless new_object
exit 1 unless system(*%%W[%s cat-file -e #{new_object}])
end
`, gitBin)
WriteCustomHook(t, repoPath, "pre-receive", []byte(hook))
}
// WriteCustomHook writes a hook in the repo/path.git/custom_hooks directory
func WriteCustomHook(t testing.TB, repoPath, name string, content []byte) {
fullPath := filepath.Join(repoPath, "custom_hooks", name)
testhelper.WriteExecutable(t, fullPath, content)
}
// CaptureHookEnv creates a bogus 'update' Git hook to sniff out what
// environment variables get set for hooks.
func CaptureHookEnv(t testing.TB) (string, func()) {
tempDir := testhelper.TempDir(t)
oldOverride := hooks.Override
hooks.Override = filepath.Join(tempDir, "hooks")
hookOutputFile := filepath.Join(tempDir, "hook.env")
require.NoError(t, os.MkdirAll(hooks.Override, 0755))
script := []byte(`
#!/bin/sh
env | grep -e ^GIT -e ^GL_ > ` + hookOutputFile + "\n")
require.NoError(t, ioutil.WriteFile(filepath.Join(hooks.Override, "update"), script, 0755))
return hookOutputFile, func() {
hooks.Override = oldOverride
}
}
|