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
|
package state
import (
"fmt"
"io/ioutil"
"labix.org/v2/mgo"
. "launchpad.net/gocheck"
"launchpad.net/juju-core/charm"
"launchpad.net/juju-core/environs/config"
"launchpad.net/juju-core/testing"
"net/url"
"path/filepath"
)
// TestingEnvironConfig returns a default environment configuration.
func TestingEnvironConfig(c *C) *config.Config {
cfg, err := config.New(map[string]interface{}{
"type": "test",
"name": "test-name",
"default-series": "test-series",
"authorized-keys": "test-keys",
"agent-version": "9.9.9.9",
"ca-cert": testing.CACert,
"ca-private-key": "",
})
c.Assert(err, IsNil)
return cfg
}
// TestingInitialize initializes the state and returns it. If state was not
// already initialized, and cfg is nil, the minimal default environment
// configuration will be used.
func TestingInitialize(c *C, cfg *config.Config) *State {
if cfg == nil {
cfg = TestingEnvironConfig(c)
}
st, err := Initialize(TestingStateInfo(), cfg, TestingDialOpts())
c.Assert(err, IsNil)
return st
}
type (
CharmDoc charmDoc
MachineDoc machineDoc
RelationDoc relationDoc
ServiceDoc serviceDoc
UnitDoc unitDoc
)
func (doc *MachineDoc) String() string {
m := &Machine{doc: machineDoc(*doc)}
return m.String()
}
func ServiceSettingsRefCount(st *State, serviceName string, curl *charm.URL) (int, error) {
key := serviceSettingsKey(serviceName, curl)
var doc settingsRefsDoc
if err := st.settingsrefs.FindId(key).One(&doc); err == nil {
return doc.RefCount, nil
}
return 0, mgo.ErrNotFound
}
func AddTestingCharm(c *C, st *State, name string) *Charm {
return addCharm(c, st, "series", testing.Charms.Dir(name))
}
func AddCustomCharm(c *C, st *State, name, filename, content, series string, revision int) *Charm {
path := testing.Charms.ClonedDirPath(c.MkDir(), name)
if filename != "" {
config := filepath.Join(path, filename)
err := ioutil.WriteFile(config, []byte(content), 0644)
c.Assert(err, IsNil)
}
ch, err := charm.ReadDir(path)
c.Assert(err, IsNil)
if revision != -1 {
ch.SetRevision(revision)
}
return addCharm(c, st, series, ch)
}
func addCharm(c *C, st *State, series string, ch charm.Charm) *Charm {
ident := fmt.Sprintf("%s-%s-%d", series, ch.Meta().Name, ch.Revision())
curl := charm.MustParseURL("local:" + series + "/" + ident)
bundleURL, err := url.Parse("http://bundles.example.com/" + ident)
c.Assert(err, IsNil)
sch, err := st.AddCharm(ch, curl, bundleURL, ident+"-sha256")
c.Assert(err, IsNil)
return sch
}
func init() {
logSize = logSizeTests
}
|