LEFT | RIGHT |
(no file at all) | |
1 package firewaller_test | 1 package firewaller_test |
2 | 2 |
3 import ( | 3 import ( |
4 . "launchpad.net/gocheck" | 4 . "launchpad.net/gocheck" |
5 "launchpad.net/juju-core/environs" | 5 "launchpad.net/juju-core/environs" |
6 "launchpad.net/juju-core/environs/dummy" | 6 "launchpad.net/juju-core/environs/dummy" |
7 "launchpad.net/juju-core/juju/testing" | 7 "launchpad.net/juju-core/juju/testing" |
8 "launchpad.net/juju-core/state" | 8 "launchpad.net/juju-core/state" |
9 coretesting "launchpad.net/juju-core/testing" | 9 coretesting "launchpad.net/juju-core/testing" |
10 "launchpad.net/juju-core/worker/firewaller" | 10 "launchpad.net/juju-core/worker/firewaller" |
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
65 units, err := s.Conn.AddUnits(svc, 1) | 65 units, err := s.Conn.AddUnits(svc, 1) |
66 c.Assert(err, IsNil) | 66 c.Assert(err, IsNil) |
67 u := units[0] | 67 u := units[0] |
68 id, err := u.AssignedMachineId() | 68 id, err := u.AssignedMachineId() |
69 c.Assert(err, IsNil) | 69 c.Assert(err, IsNil) |
70 m, err := s.State.Machine(id) | 70 m, err := s.State.Machine(id) |
71 c.Assert(err, IsNil) | 71 c.Assert(err, IsNil) |
72 return u, m | 72 return u, m |
73 } | 73 } |
74 | 74 |
| 75 func (s *FirewallerSuite) StateInfo(c *C, m *state.Machine) *state.Info { |
| 76 info := s.JujuConnSuite.StateInfo(c) |
| 77 info.EntityName = m.EntityName() |
| 78 info.Password = "irrelevant" |
| 79 return info |
| 80 } |
| 81 |
75 // startInstance starts a new instance for the given machine. | 82 // startInstance starts a new instance for the given machine. |
76 func (s *FirewallerSuite) startInstance(c *C, m *state.Machine) environs.Instanc
e { | 83 func (s *FirewallerSuite) startInstance(c *C, m *state.Machine) environs.Instanc
e { |
77 » inst, err := s.Conn.Environ.StartInstance(m.Id(), s.StateInfo(c), nil) | 84 » inst, err := s.Conn.Environ.StartInstance(m.Id(), s.StateInfo(c, m), nil
) |
78 c.Assert(err, IsNil) | 85 c.Assert(err, IsNil) |
79 err = m.SetInstanceId(inst.Id()) | 86 err = m.SetInstanceId(inst.Id()) |
80 c.Assert(err, IsNil) | 87 c.Assert(err, IsNil) |
81 return inst | 88 return inst |
82 } | 89 } |
83 | 90 |
84 func (s *FirewallerSuite) TestNotExposedService(c *C) { | 91 func (s *FirewallerSuite) TestNotExposedService(c *C) { |
85 fw := firewaller.NewFirewaller(s.State) | 92 fw := firewaller.NewFirewaller(s.State) |
86 defer func() { c.Assert(fw.Stop(), IsNil) }() | 93 defer func() { c.Assert(fw.Stop(), IsNil) }() |
87 | 94 |
(...skipping 156 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
244 | 251 |
245 s.assertPorts(c, inst, m.Id(), []state.Port{{"tcp", 80}, {"tcp", 8080}}) | 252 s.assertPorts(c, inst, m.Id(), []state.Port{{"tcp", 80}, {"tcp", 8080}}) |
246 | 253 |
247 err = svc.SetExposed() | 254 err = svc.SetExposed() |
248 c.Assert(err, IsNil) | 255 c.Assert(err, IsNil) |
249 } | 256 } |
250 | 257 |
251 func (s *FirewallerSuite) TestFirewallerStartWithPartialState(c *C) { | 258 func (s *FirewallerSuite) TestFirewallerStartWithPartialState(c *C) { |
252 m, err := s.State.AddMachine(state.MachinerWorker) | 259 m, err := s.State.AddMachine(state.MachinerWorker) |
253 c.Assert(err, IsNil) | 260 c.Assert(err, IsNil) |
254 » inst, err := s.Conn.Environ.StartInstance(m.Id(), s.StateInfo(c), nil) | 261 » inst, err := s.Conn.Environ.StartInstance(m.Id(), s.StateInfo(c, m), nil
) |
255 c.Assert(err, IsNil) | 262 c.Assert(err, IsNil) |
256 err = m.SetInstanceId(inst.Id()) | 263 err = m.SetInstanceId(inst.Id()) |
257 c.Assert(err, IsNil) | 264 c.Assert(err, IsNil) |
258 | 265 |
259 svc, err := s.State.AddService("wordpress", s.charm) | 266 svc, err := s.State.AddService("wordpress", s.charm) |
260 c.Assert(err, IsNil) | 267 c.Assert(err, IsNil) |
261 err = svc.SetExposed() | 268 err = svc.SetExposed() |
262 c.Assert(err, IsNil) | 269 c.Assert(err, IsNil) |
263 | 270 |
264 // Starting the firewaller, no open ports. | 271 // Starting the firewaller, no open ports. |
(...skipping 210 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
475 c.Assert(err, IsNil) | 482 c.Assert(err, IsNil) |
476 | 483 |
477 // Remove machine. Nothing bad should happen, but can't | 484 // Remove machine. Nothing bad should happen, but can't |
478 // assert port state since the machine must have been | 485 // assert port state since the machine must have been |
479 // destroyed and we lost its reference. | 486 // destroyed and we lost its reference. |
480 err = m.EnsureDead() | 487 err = m.EnsureDead() |
481 c.Assert(err, IsNil) | 488 c.Assert(err, IsNil) |
482 err = s.State.RemoveMachine(m.Id()) | 489 err = s.State.RemoveMachine(m.Id()) |
483 c.Assert(err, IsNil) | 490 c.Assert(err, IsNil) |
484 } | 491 } |
LEFT | RIGHT |