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
|
package admin
import (
"bytes"
"log"
"net/http"
"net/http/httptest"
"os"
"path/filepath"
"testing"
"time"
"github.com/go-git/go-billy/v5/util"
"github.com/go-git/go-git/v5"
"github.com/go-git/go-git/v5/plumbing/object"
"sigs.k8s.io/yaml"
)
func junkTestHandler() http.HandlerFunc {
return func(rw http.ResponseWriter, req *http.Request) {
rw.WriteHeader(http.StatusOK)
_, err := rw.Write([]byte("Im a body"))
if err != nil {
log.Fatalf("couldn't write http body %s", err)
}
}
}
func TestHooks(t *testing.T) {
tempDir := t.TempDir()
tempRepoDir := filepath.Join(tempDir, "mgmt.git")
err := os.Mkdir(tempRepoDir, 0750)
if err != nil && !os.IsExist(err) {
log.Fatal(err)
}
// setup server with repo for management
defaultConfigBytes, err := yaml.Marshal(defaultServerConfig)
if err != nil {
t.Fatal(err)
}
if err := configDefaultTestRepo(tempRepoDir, defaultConfigBytes); err != nil {
t.Fatal(err)
}
destModelFile, destPolicyFile, _ := tempModelPolicyConfig(t, tempDir, tempRepoDir)
// Load the server/config from the repo
adminService, err := NewService(destModelFile,
destPolicyFile,
"gitserver.yaml",
tempDir,
true)
if err != nil {
t.Fatal(err)
}
repo, err := git.PlainOpen(tempRepoDir)
if err != nil {
t.Fatal(err)
}
worktree, err := repo.Worktree()
if err != nil {
t.Fatal("couldn't get working tree", err)
}
fileToCommit := "gitserver.yaml"
newTestConfigRepo := *defaultRepo
newTestConfigRepo.Name = "mymiddlewaretest"
newServerConfig := *defaultServerConfig
newServerConfig.Repos = append(newServerConfig.Repos, &newTestConfigRepo)
newConfigBytes, err := yaml.Marshal(newServerConfig)
if err != nil {
t.Fatal("couldn't marshal new server config", err)
}
err = util.WriteFile(worktree.Filesystem, fileToCommit, newConfigBytes, os.FileMode(int(0644)))
if err != nil {
t.Fatal("couldn't write server config", err)
}
if _, err := worktree.Add(fileToCommit); err != nil {
t.Fatal("couldn't add server config to worktree", err)
}
_, err = worktree.Commit(fileToCommit,
&git.CommitOptions{
Author: &object.Signature{
Name: "go git server",
Email: "go-git-server@go-git-server",
},
},
)
if err != nil {
t.Fatal("couldn't commit file", err)
}
req := httptest.NewRequest(http.MethodPost, "http://localhost:3456/mgmt/git-receive-pack", bytes.NewBuffer([]byte("stuff")))
recorder := httptest.NewRecorder()
adminHandler := Hooks(adminService, junkTestHandler())
adminHandler.ServeHTTP(recorder, req)
result := recorder.Result()
defer result.Body.Close()
found := false
// wait for bg job to complete
time.Sleep(1 * time.Second)
for _, v := range adminService.Conf.Repos {
if v.Name == newTestConfigRepo.Name {
found = true
}
}
if !found {
t.Error("didn't find expected repo in config")
t.Fail()
}
}
|
