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
|
package mstate_test
import (
"fmt"
"labix.org/v2/mgo"
"labix.org/v2/mgo/bson"
. "launchpad.net/gocheck"
"launchpad.net/juju-core/charm"
state "launchpad.net/juju-core/mstate"
"launchpad.net/juju-core/testing"
"net/url"
"sort"
)
// ConnSuite facilitates access to the underlying MongoDB.
// It is embedded in UtilSuite.
type ConnSuite struct {
MgoSuite
session *mgo.Session
charms *mgo.Collection
machines *mgo.Collection
services *mgo.Collection
units *mgo.Collection
}
func (cs *ConnSuite) SetUpTest(c *C) {
cs.MgoSuite.SetUpTest(c)
session, err := mgo.Dial(mgoaddr)
c.Assert(err, IsNil)
cs.session = session
cs.charms = session.DB("juju").C("charms")
cs.machines = session.DB("juju").C("machines")
cs.services = session.DB("juju").C("services")
cs.units = session.DB("juju").C("units")
}
func (cs *ConnSuite) TearDownTest(c *C) {
cs.session.Close()
cs.MgoSuite.TearDownTest(c)
}
func (s *ConnSuite) AllMachines(c *C) []int {
docs := []state.MachineDoc{}
err := s.machines.Find(bson.D{{"life", state.Alive}}).All(&docs)
c.Assert(err, IsNil)
ids := []int{}
for _, v := range docs {
ids = append(ids, v.Id)
}
sort.Ints(ids)
return ids
}
// UtilSuite provides the infrastructure for all other
// test suites (StateSuite, CharmSuite, MachineSuite, etc).
type UtilSuite struct {
MgoSuite
ConnSuite
State *state.State
}
func (s *UtilSuite) SetUpTest(c *C) {
s.ConnSuite.SetUpTest(c)
st, err := state.Dial(mgoaddr)
c.Assert(err, IsNil)
s.State = st
}
func (s *UtilSuite) TearDownTest(c *C) {
s.State.Close()
s.ConnSuite.TearDownTest(c)
}
func (s *UtilSuite) AddTestingCharm(c *C, name string) *state.Charm {
ch := testing.Charms.Dir(name)
ident := fmt.Sprintf("%s-%d", name, ch.Revision())
curl := charm.MustParseURL("local:series/" + ident)
bundleURL, err := url.Parse("http://bundles.example.com/" + ident)
c.Assert(err, IsNil)
sch, err := s.State.AddCharm(ch, curl, bundleURL, ident+"-sha256")
c.Assert(err, IsNil)
return sch
}
|