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
|
package gitlabshell_test
import (
"bytes"
"encoding/json"
"io/ioutil"
"os"
"os/exec"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/internal/config"
"gitlab.com/gitlab-org/gitaly/internal/gitlabshell"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
)
func TestGitHooksConfig(t *testing.T) {
testhelper.ConfigureRuby()
defer func(cfg config.Cfg) {
config.Config = cfg
}(config.Config)
loggingDir, err := ioutil.TempDir("", t.Name())
require.NoError(t, err)
defer func() { os.RemoveAll(loggingDir) }()
config.Config.Logging.Dir = loggingDir
config.Config.Logging.Level = "fatal"
config.Config.Logging.Format = "my-custom-format"
config.Config.GitlabShell = config.GitlabShell{
CustomHooksDir: "/path/to/custom_hooks",
Dir: "../../ruby/gitlab-shell",
GitlabURL: "http://gitlaburl.com",
HTTPSettings: config.HTTPSettings{
ReadTimeout: 100,
User: "user_name",
Password: "pwpw",
CAFile: "/ca_file_path",
CAPath: "/ca_path",
SelfSigned: true,
},
SecretFile: "secret_file_path",
}
dumpConfigPath := filepath.Join(config.Config.Ruby.Dir, "gitlab-shell", "bin", "dump-config")
var stdout bytes.Buffer
cmd := exec.Command(dumpConfigPath)
gitlabshellEnv, err := gitlabshell.Env()
require.NoError(t, err)
cmd.Env = append(os.Environ(), gitlabshellEnv...)
cmd.Stdout = &stdout
cmd.Stderr = os.Stderr
require.NoError(t, cmd.Run())
rubyConfigMap := make(map[string]interface{})
require.NoError(t, json.NewDecoder(&stdout).Decode(&rubyConfigMap))
require.Equal(t, config.Config.Logging.Level, rubyConfigMap["log_level"])
require.Equal(t, config.Config.Logging.Format, rubyConfigMap["log_format"])
require.Equal(t, config.Config.GitlabShell.SecretFile, rubyConfigMap["secret_file"])
require.Equal(t, config.Config.GitlabShell.CustomHooksDir, rubyConfigMap["custom_hooks_dir"])
require.Equal(t, config.Config.GitlabShell.GitlabURL, rubyConfigMap["gitlab_url"])
// HTTP Settings
httpSettings, ok := rubyConfigMap["http_settings"].(map[string]interface{})
require.True(t, ok)
require.Equal(t, float64(config.Config.GitlabShell.HTTPSettings.ReadTimeout), httpSettings["read_timeout"])
require.Equal(t, config.Config.GitlabShell.HTTPSettings.User, httpSettings["user"])
require.Equal(t, config.Config.GitlabShell.HTTPSettings.Password, httpSettings["password"])
require.Equal(t, config.Config.GitlabShell.HTTPSettings.CAFile, httpSettings["ca_file"])
require.Equal(t, config.Config.GitlabShell.HTTPSettings.CAPath, httpSettings["ca_path"])
require.Equal(t, config.Config.GitlabShell.HTTPSettings.SelfSigned, httpSettings["self_signed_cert"])
dir := filepath.Dir(rubyConfigMap["log_file"].(string))
require.Equal(t, config.Config.Logging.Dir, dir)
}
|