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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
package mstate_test
import (
. "launchpad.net/gocheck"
"launchpad.net/juju-core/charm"
state "launchpad.net/juju-core/mstate"
)
type ServiceSuite struct {
UtilSuite
charm *state.Charm
service *state.Service
}
var _ = Suite(&ServiceSuite{})
func (s *ServiceSuite) SetUpTest(c *C) {
s.UtilSuite.SetUpTest(c)
s.charm = s.AddTestingCharm(c, "dummy")
var err error
s.service, err = s.State.AddService("mysql", s.charm)
c.Assert(err, IsNil)
}
func (s *ServiceSuite) TestServiceCharm(c *C) {
// Check that getting and setting the service charm URL works correctly.
testcurl, err := s.service.CharmURL()
c.Assert(err, IsNil)
c.Assert(testcurl.String(), Equals, s.charm.URL().String())
// TODO BUG https://bugs.launchpad.net/juju-core/+bug/1020318
testcurl = charm.MustParseURL("local:myseries/mydummy-1")
err = s.service.SetCharmURL(testcurl)
c.Assert(err, IsNil)
testcurl, err = s.service.CharmURL()
c.Assert(err, IsNil)
c.Assert(testcurl.String(), Equals, "local:myseries/mydummy-1")
}
func (s *ServiceSuite) TestAddUnit(c *C) {
// Check that principal units can be added on their own.
unitZero, err := s.service.AddUnit()
c.Assert(err, IsNil)
c.Assert(unitZero.Name(), Equals, "mysql/0")
principal := unitZero.IsPrincipal()
c.Assert(principal, Equals, true)
unitOne, err := s.service.AddUnit()
c.Assert(err, IsNil)
c.Assert(unitOne.Name(), Equals, "mysql/1")
principal = unitOne.IsPrincipal()
c.Assert(principal, Equals, true)
// Check that principal units cannot be added to principal units.
_, err = s.service.AddUnitSubordinateTo(unitZero)
c.Assert(err, ErrorMatches, `can't add unit of principal service "mysql" as a subordinate of "mysql/0"`)
// Assign the principal unit to a machine.
m, err := s.State.AddMachine()
c.Assert(err, IsNil)
err = unitZero.AssignToMachine(m)
c.Assert(err, IsNil)
// Add a subordinate service.
subCharm := s.AddTestingCharm(c, "logging")
logging, err := s.State.AddService("logging", subCharm)
c.Assert(err, IsNil)
// Check that subordinate units can be added to principal units
subZero, err := logging.AddUnitSubordinateTo(unitZero)
c.Assert(err, IsNil)
c.Assert(subZero.Name(), Equals, "logging/0")
principal = subZero.IsPrincipal()
c.Assert(principal, Equals, false)
// Check the subordinate unit has been assigned its principal's machine.
id, err := subZero.AssignedMachineId()
c.Assert(err, IsNil)
c.Assert(id, Equals, m.Id())
// Check that subordinate units must be added to other units.
_, err = logging.AddUnit()
c.Assert(err, ErrorMatches, `cannot directly add units to subordinate service "logging"`)
// Check that subordinate units cannnot be added to subordinate units.
_, err = logging.AddUnitSubordinateTo(subZero)
c.Assert(err, ErrorMatches, "a subordinate unit must be added to a principal unit")
}
func (s *ServiceSuite) TestReadUnit(c *C) {
_, err := s.service.AddUnit()
c.Assert(err, IsNil)
_, err = s.service.AddUnit()
c.Assert(err, IsNil)
// Check that retrieving a unit works correctly.
unit, err := s.service.Unit("mysql/0")
c.Assert(err, IsNil)
c.Assert(unit.Name(), Equals, "mysql/0")
// Check that retrieving a non-existent or an invalidly
// named unit fail nicely.
unit, err = s.service.Unit("mysql")
c.Assert(err, ErrorMatches, `can't get unit "mysql" from service "mysql":.*`)
unit, err = s.service.Unit("mysql/0/0")
c.Assert(err, ErrorMatches, `can't get unit "mysql/0/0" from service "mysql": .*`)
unit, err = s.service.Unit("pressword/0")
c.Assert(err, ErrorMatches, `can't get unit "pressword/0" from service "mysql": .*`)
// Add another service to check units are not misattributed.
mysql, err := s.State.AddService("wordpress", s.charm)
c.Assert(err, IsNil)
_, err = mysql.AddUnit()
c.Assert(err, IsNil)
// BUG: use error strings from state.
unit, err = s.service.Unit("wordpress/0")
c.Assert(err, ErrorMatches, `can't get unit "wordpress/0" from service "mysql": .*`)
// Check that retrieving all units works.
units, err := s.service.AllUnits()
c.Assert(err, IsNil)
c.Assert(len(units), Equals, 2)
c.Assert(units[0].Name(), Equals, "mysql/0")
c.Assert(units[1].Name(), Equals, "mysql/1")
}
func (s *ServiceSuite) TestRemoveUnit(c *C) {
_, err := s.service.AddUnit()
c.Assert(err, IsNil)
_, err = s.service.AddUnit()
c.Assert(err, IsNil)
// Check that removing a unit works.
unit, err := s.service.Unit("mysql/0")
c.Assert(err, IsNil)
err = s.service.RemoveUnit(unit)
c.Assert(err, IsNil)
units, err := s.service.AllUnits()
c.Assert(err, IsNil)
c.Assert(units, HasLen, 1)
c.Assert(units[0].Name(), Equals, "mysql/1")
// Check that removing a non-existent unit fails nicely.
// TODO improve error message.
// BUG: use error strings from state
err = s.service.RemoveUnit(unit)
c.Assert(err, ErrorMatches, `can't remove unit "mysql/0": .*`)
}
|