9950: Remove uses of environment variables.
[arvados.git] / services / arv-git-httpd / gitolite_test.go
1 package main
2
3 import (
4         "io/ioutil"
5         "os"
6         "os/exec"
7         "strings"
8
9         "git.curoverse.com/arvados.git/sdk/go/arvados"
10         "git.curoverse.com/arvados.git/sdk/go/arvadostest"
11         check "gopkg.in/check.v1"
12 )
13
14 var _ = check.Suite(&GitoliteSuite{})
15
16 // GitoliteSuite tests need an API server, an arv-git-httpd server,
17 // and a repository hosted by gitolite.
18 type GitoliteSuite struct {
19         IntegrationSuite
20         gitoliteHome string
21 }
22
23 func (s *GitoliteSuite) SetUpTest(c *check.C) {
24         var err error
25         s.gitoliteHome, err = ioutil.TempDir("", "arv-git-httpd")
26         c.Assert(err, check.Equals, nil)
27
28         runGitolite := func(prog string, args ...string) {
29                 c.Log(prog, " ", args)
30                 cmd := exec.Command(prog, args...)
31                 cmd.Dir = s.gitoliteHome
32                 cmd.Env = []string{"HOME=" + s.gitoliteHome}
33                 for _, e := range os.Environ() {
34                         if !strings.HasPrefix(e, "HOME=") {
35                                 cmd.Env = append(cmd.Env, e)
36                         }
37                 }
38                 diags, err := cmd.CombinedOutput()
39                 c.Log(string(diags))
40                 c.Assert(err, check.Equals, nil)
41         }
42
43         runGitolite("gitolite", "setup", "--admin", "root")
44
45         s.tmpRepoRoot = s.gitoliteHome + "/repositories"
46         s.Config = &Config{
47                 Client: arvados.Client{
48                         APIHost:  arvadostest.APIHost(),
49                         Insecure: true,
50                 },
51                 Listen:     ":0",
52                 GitCommand: "/usr/share/gitolite3/gitolite-shell",
53                 RepoRoot:   s.tmpRepoRoot,
54         }
55         s.IntegrationSuite.SetUpTest(c)
56
57         // Install the gitolite hooks in the bare repo we made in
58         // (*IntegrationTest)SetUpTest() -- see 2.2.4 at
59         // http://gitolite.com/gitolite/gitolite.html
60         runGitolite("gitolite", "setup")
61
62         os.Setenv("GITOLITE_HTTP_HOME", s.gitoliteHome)
63         os.Setenv("GL_BYPASS_ACCESS_CHECKS", "1")
64 }
65
66 func (s *GitoliteSuite) TearDownTest(c *check.C) {
67         // We really want Unsetenv here, but it's not worth forcing an
68         // upgrade to Go 1.4.
69         os.Setenv("GITOLITE_HTTP_HOME", "")
70         os.Setenv("GL_BYPASS_ACCESS_CHECKS", "")
71         if s.gitoliteHome != "" {
72                 err := os.RemoveAll(s.gitoliteHome)
73                 c.Check(err, check.Equals, nil)
74         }
75         s.IntegrationSuite.TearDownTest(c)
76 }
77
78 func (s *GitoliteSuite) TestFetch(c *check.C) {
79         err := s.RunGit(c, activeToken, "fetch", "active/foo.git")
80         c.Check(err, check.Equals, nil)
81 }
82
83 func (s *GitoliteSuite) TestFetchUnreadable(c *check.C) {
84         err := s.RunGit(c, anonymousToken, "fetch", "active/foo.git")
85         c.Check(err, check.ErrorMatches, `.* not found.*`)
86 }
87
88 func (s *GitoliteSuite) TestPush(c *check.C) {
89         err := s.RunGit(c, activeToken, "push", "active/foo.git", "master:gitolite-push")
90         c.Check(err, check.Equals, nil)
91
92         // Check that the commit hash appears in the gitolite log, as
93         // assurance that the gitolite hooks really did run.
94
95         sha1, err := exec.Command("git", "--git-dir", s.tmpWorkdir+"/.git",
96                 "log", "-n1", "--format=%H").CombinedOutput()
97         c.Logf("git-log in workdir: %q", string(sha1))
98         c.Assert(err, check.Equals, nil)
99         c.Assert(len(sha1), check.Equals, 41)
100
101         gitoliteLog, err := exec.Command("grep", "-r", string(sha1[:40]), s.gitoliteHome+"/.gitolite/logs").CombinedOutput()
102         c.Check(err, check.Equals, nil)
103         c.Logf("gitolite log message: %q", string(gitoliteLog))
104 }
105
106 func (s *GitoliteSuite) TestPushUnwritable(c *check.C) {
107         err := s.RunGit(c, spectatorToken, "push", "active/foo.git", "master:gitolite-push-fail")
108         c.Check(err, check.ErrorMatches, `.*HTTP code = 403.*`)
109 }