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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
// Copyright 2014 Canonical Ltd.
// Licensed under the LGPLv3, see LICENCE file for details.
package testing
import (
"io/ioutil"
"os"
"path/filepath"
gc "gopkg.in/check.v1"
jc "github.com/juju/testing/checkers"
)
type TestFile struct {
Name, Data string
}
// FakeHome stores information about the user's home
// environment so it can be cast aside for tests and
// restored afterwards.
type FakeHome struct {
files []TestFile
}
func MakeFakeHome(c *gc.C) *FakeHome {
fakeHome := c.MkDir()
err := os.Setenv("HOME", fakeHome)
c.Assert(err, jc.ErrorIsNil)
sshPath := filepath.Join(fakeHome, ".ssh")
err = os.Mkdir(sshPath, 0777)
c.Assert(err, gc.IsNil)
err = ioutil.WriteFile(filepath.Join(sshPath, "id_rsa"), []byte("private auth key\n"), 0600)
c.Assert(err, gc.IsNil)
err = ioutil.WriteFile(filepath.Join(sshPath, "id_rsa.pub"), []byte("public auth key\n"), 0666)
c.Assert(err, gc.IsNil)
return &FakeHome{
files: []TestFile{},
}
}
func (h *FakeHome) AddFiles(c *gc.C, files ...TestFile) {
for _, f := range files {
path := filepath.Join(os.Getenv("HOME"), f.Name)
err := os.MkdirAll(filepath.Dir(path), 0700)
c.Assert(err, gc.IsNil)
err = ioutil.WriteFile(path, []byte(f.Data), 0666)
c.Assert(err, gc.IsNil)
h.files = append(h.files, f)
}
}
// FileContents returns the test file contents for the
// given specified path (which may be relative, so
// we compare with the base filename only).
func (h *FakeHome) FileContents(c *gc.C, path string) string {
for _, f := range h.files {
if filepath.Base(f.Name) == filepath.Base(path) {
return f.Data
}
}
c.Fatalf("path attribute holds unknown test file: %q", path)
panic("unreachable")
}
// FileExists returns if the given relative file path exists
// in the fake home.
func (h *FakeHome) FileExists(path string) bool {
for _, f := range h.files {
if f.Name == path {
return true
}
}
return false
}
// HomePath joins the specified path snippets and returns
// an absolute path under Juju home.
func HomePath(names ...string) string {
all := append([]string{os.Getenv("HOME")}, names...)
return filepath.Join(all...)
}
// JujuXDGDataHomePath returns the test home path, it is just a convenience
// for tests, if extra path snippets are passed they will be
// joined to juju home.
// This tool assumes ~/.config/juju as the juju home.
func JujuXDGDataHomePath(names ...string) string {
all := append([]string{".local", "share", "juju"}, names...)
return HomePath(all...)
}
// FakeHomeSuite sets up a fake home directory before running tests.
type FakeHomeSuite struct {
CleanupSuite
LoggingSuite
Home *FakeHome
}
func (s *FakeHomeSuite) SetUpSuite(c *gc.C) {
s.CleanupSuite.SetUpSuite(c)
s.LoggingSuite.SetUpSuite(c)
}
func (s *FakeHomeSuite) TearDownSuite(c *gc.C) {
s.LoggingSuite.TearDownSuite(c)
s.CleanupSuite.TearDownSuite(c)
}
func (s *FakeHomeSuite) SetUpTest(c *gc.C) {
s.CleanupSuite.SetUpTest(c)
s.LoggingSuite.SetUpTest(c)
home := os.Getenv("HOME")
s.Home = MakeFakeHome(c)
s.AddCleanup(func(*gc.C) {
err := os.Setenv("HOME", home)
c.Assert(err, jc.ErrorIsNil)
})
}
func (s *FakeHomeSuite) TearDownTest(c *gc.C) {
s.LoggingSuite.TearDownTest(c)
s.CleanupSuite.TearDownTest(c)
}
|
