worker/apiuniter: Clone uniter, fix imports

worker/apiuniter/tools_test.go

+// Copyright 2012, 2013 Canonical Ltd.
+// Licensed under the AGPLv3, see LICENCE file for details.
+
+package apiuniter_test
+
+import (
+        "io/ioutil"
+        "os"
+        "path/filepath"
+        "time"
+
+        gc "launchpad.net/gocheck"
+
+        "launchpad.net/juju-core/agent/tools"
+        "launchpad.net/juju-core/version"
+        "launchpad.net/juju-core/worker/apiuniter"
+        "launchpad.net/juju-core/worker/apiuniter/jujuc"
+)
+
+type ToolsSuite struct {
+        dataDir, toolsDir string
+}
+
+var _ = gc.Suite(&ToolsSuite{})
+
+func (s *ToolsSuite) SetUpTest(c *gc.C) {
+        s.dataDir = c.MkDir()
+        s.toolsDir = tools.SharedToolsDir(s.dataDir, version.Current)
+        err := os.MkdirAll(s.toolsDir, 0755)
+        c.Assert(err, gc.IsNil)
+        err = os.Symlink(s.toolsDir, tools.ToolsDir(s.dataDir, "unit-u-123"))
+        c.Assert(err, gc.IsNil)
+}
+
+func (s *ToolsSuite) TestEnsureJujucSymlinks(c *gc.C) {
+        jujudPath := filepath.Join(s.toolsDir, "jujud")
+        err := ioutil.WriteFile(jujudPath, []byte("assume sane"), 0755)
+        c.Assert(err, gc.IsNil)
+
+        assertLink := func(path string) time.Time {
+                target, err := os.Readlink(path)
+                c.Assert(err, gc.IsNil)
+                c.Assert(target, gc.Equals, "./jujud")
+                fi, err := os.Lstat(path)
+                c.Assert(err, gc.IsNil)
+                return fi.ModTime()
+        }
+
+        // Check that EnsureJujucSymlinks writes appropriate symlinks.
+        err = apiuniter.EnsureJujucSymlinks(s.toolsDir)
+        c.Assert(err, gc.IsNil)
+        mtimes := map[string]time.Time{}
+        for _, name := range jujuc.CommandNames() {
+                tool := filepath.Join(s.toolsDir, name)
+                mtimes[tool] = assertLink(tool)
+        }
+
+        // Check that EnsureJujucSymlinks doesn't overwrite things that don't need to be.
+        err = apiuniter.EnsureJujucSymlinks(s.toolsDir)
+        c.Assert(err, gc.IsNil)
+        for tool, mtime := range mtimes {
+                c.Assert(assertLink(tool), gc.Equals, mtime)
+        }
+}
+
+func (s *ToolsSuite) TestEnsureJujucSymlinksBadDir(c *gc.C) {
+        err := apiuniter.EnsureJujucSymlinks(filepath.Join(c.MkDir(), "noexist"))
+        c.Assert(err, gc.ErrorMatches, "cannot initialize hook commands in .*: no such file or directory")
+}