← Back to branch summary

~juju-qa/ubuntu/yakkety/juju/2.0-rc3-again

« back to all changes in this revision

Viewing changes to src/launchpad.net/juju-core/worker/uniter/tools_test.go

  • Committer: Package Import Robot
  • Author(s): James Page
  • Date: 2013-04-24 22:34:47 UTC
  • Revision ID: package-import@ubuntu.com-20130424223447-f0qdji7ubnyo0s71
Tags: upstream-1.10.0.1
Import upstream version 1.10.0.1

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/launchpad.net/juju-core/worker/uniter/tools_test.go
 
1
package uniter_test
 
2
 
 
3
import (
 
4
        "io/ioutil"
 
5
        . "launchpad.net/gocheck"
 
6
        "launchpad.net/juju-core/environs/agent"
 
7
        "launchpad.net/juju-core/version"
 
8
        "launchpad.net/juju-core/worker/uniter"
 
9
        "launchpad.net/juju-core/worker/uniter/jujuc"
 
10
        "os"
 
11
        "path/filepath"
 
12
        "time"
 
13
)
 
14
 
 
15
type ToolsSuite struct {
 
16
        dataDir, toolsDir string
 
17
}
 
18
 
 
19
var _ = Suite(&ToolsSuite{})
 
20
 
 
21
func (s *ToolsSuite) SetUpTest(c *C) {
 
22
        s.dataDir = c.MkDir()
 
23
        s.toolsDir = agent.SharedToolsDir(s.dataDir, version.Current)
 
24
        err := os.MkdirAll(s.toolsDir, 0755)
 
25
        c.Assert(err, IsNil)
 
26
        err = os.Symlink(s.toolsDir, agent.ToolsDir(s.dataDir, "unit-u-123"))
 
27
        c.Assert(err, IsNil)
 
28
}
 
29
 
 
30
func (s *ToolsSuite) TestEnsureJujucSymlinks(c *C) {
 
31
        jujudPath := filepath.Join(s.toolsDir, "jujud")
 
32
        err := ioutil.WriteFile(jujudPath, []byte("assume sane"), 0755)
 
33
        c.Assert(err, IsNil)
 
34
 
 
35
        assertLink := func(path string) time.Time {
 
36
                target, err := os.Readlink(path)
 
37
                c.Assert(err, IsNil)
 
38
                c.Assert(target, Equals, "./jujud")
 
39
                fi, err := os.Lstat(path)
 
40
                c.Assert(err, IsNil)
 
41
                return fi.ModTime()
 
42
        }
 
43
 
 
44
        // Check that EnsureJujucSymlinks writes appropriate symlinks.
 
45
        err = uniter.EnsureJujucSymlinks(s.toolsDir)
 
46
        c.Assert(err, IsNil)
 
47
        mtimes := map[string]time.Time{}
 
48
        for _, name := range jujuc.CommandNames() {
 
49
                tool := filepath.Join(s.toolsDir, name)
 
50
                mtimes[tool] = assertLink(tool)
 
51
        }
 
52
 
 
53
        // Check that EnsureJujucSymlinks doesn't overwrite things that don't need to be.
 
54
        err = uniter.EnsureJujucSymlinks(s.toolsDir)
 
55
        c.Assert(err, IsNil)
 
56
        for tool, mtime := range mtimes {
 
57
                c.Assert(assertLink(tool), Equals, mtime)
 
58
        }
 
59
}
 
60
 
 
61
func (s *ToolsSuite) TestEnsureJujucSymlinksBadDir(c *C) {
 
62
        err := uniter.EnsureJujucSymlinks(filepath.Join(c.MkDir(), "noexist"))
 
63
        c.Assert(err, ErrorMatches, "cannot initialize hook commands in .*: no such file or directory")
 
64
}