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
|
// Copyright 2014 Canonical Ltd.
// Copyright 2014 Cloudbase Solutions SRL
// Licensed under the AGPLv3, see LICENCE file for details.
package agent
import (
"io/ioutil"
"os"
"runtime"
jc "github.com/juju/testing/checkers"
gc "gopkg.in/check.v1"
"gopkg.in/juju/names.v2"
"github.com/juju/juju/apiserver/params"
"github.com/juju/juju/testing"
jujuversion "github.com/juju/juju/version"
)
type identitySuite struct {
testing.BaseSuite
mongodConfigPath string
mongodPath string
}
var _ = gc.Suite(&identitySuite{})
var attributeParams = AgentConfigParams{
Tag: names.NewMachineTag("1"),
UpgradedToVersion: jujuversion.Current,
Password: "sekrit",
CACert: "ca cert",
StateAddresses: []string{"localhost:1234"},
APIAddresses: []string{"localhost:1235"},
Nonce: "a nonce",
Model: testing.ModelTag,
}
var servingInfo = params.StateServingInfo{
Cert: "old cert",
PrivateKey: "old key",
CAPrivateKey: "old ca key",
StatePort: 69,
APIPort: 47,
SharedSecret: "shared",
SystemIdentity: "identity",
}
func (s *identitySuite) TestWriteSystemIdentityFile(c *gc.C) {
params := attributeParams
params.Paths.DataDir = c.MkDir()
conf, err := NewStateMachineConfig(params, servingInfo)
c.Assert(err, jc.ErrorIsNil)
err = WriteSystemIdentityFile(conf)
c.Assert(err, jc.ErrorIsNil)
contents, err := ioutil.ReadFile(conf.SystemIdentityPath())
c.Assert(err, jc.ErrorIsNil)
c.Check(string(contents), gc.Equals, servingInfo.SystemIdentity)
fi, err := os.Stat(conf.SystemIdentityPath())
c.Assert(err, jc.ErrorIsNil)
// Windows is not fully POSIX compliant. Chmod() and Chown() have unexpected behavior
// compared to linux/unix
if runtime.GOOS != "windows" {
c.Check(fi.Mode().Perm(), gc.Equals, os.FileMode(0600))
}
// ensure that file is deleted when SystemIdentity is empty
info := servingInfo
info.SystemIdentity = ""
conf, err = NewStateMachineConfig(params, info)
c.Assert(err, jc.ErrorIsNil)
err = WriteSystemIdentityFile(conf)
c.Assert(err, jc.ErrorIsNil)
fi, err = os.Stat(conf.SystemIdentityPath())
c.Assert(err, jc.Satisfies, os.IsNotExist)
}
|