func (s *environBrokerSuite) TestStartInstanceOpensAPIPort(c *gc.C) {
s.FakeEnviron.Spec = s.spec
s.FakeEnviron.Inst = s.BaseInstance
s.FakeEnviron.Hwc = s.hardware
// Get the API port from the fake environment config used to
// "bootstrap".
envConfig := testing.FakeConfig()
apiPort, ok := envConfig["api-port"].(int)
c.Assert(ok, jc.IsTrue)
c.Assert(apiPort, gc.Not(gc.Equals), 0)
// When StateServingInfo is not nil, verify OpenPorts was called
// for the API port.
s.StartInstArgs.InstanceConfig.StateServingInfo = ¶ms.StateServingInfo{
APIPort: apiPort,
}
result, err := s.Env.StartInstance(s.StartInstArgs)
c.Assert(err, jc.ErrorIsNil)
c.Check(result.Instance, gc.DeepEquals, s.Instance)
c.Check(result.Hardware, gc.DeepEquals, s.hardware)
called, calls := s.FakeConn.WasCalled("OpenPorts")
c.Check(called, gc.Equals, true)
c.Check(calls, gc.HasLen, 1)
c.Check(calls[0].FirewallName, gc.Equals, gce.GlobalFirewallName(s.Env))
expectPorts := []network.PortRange{{
FromPort: apiPort,
ToPort: apiPort,
Protocol: "tcp",
}}
c.Check(calls[0].PortRanges, jc.DeepEquals, expectPorts)
}
func (s *environSuite) TestBootstrapOpensAPIPort(c *gc.C) {
finalizer := func(environs.BootstrapContext, *instancecfg.InstanceConfig, environs.BootstrapDialOpts) error {
return nil
}
s.FakeCommon.BSFinalizer = finalizer
ctx := envtesting.BootstrapContext(c)
params := environs.BootstrapParams{
ControllerConfig: testing.FakeControllerConfig(),
}
_, err := s.Env.Bootstrap(ctx, params)
c.Assert(err, jc.ErrorIsNil)
apiPort := params.ControllerConfig.APIPort()
called, calls := s.FakeConn.WasCalled("OpenPorts")
c.Check(called, gc.Equals, true)
c.Check(calls, gc.HasLen, 1)
c.Check(calls[0].FirewallName, gc.Equals, gce.GlobalFirewallName(s.Env))
expectPorts := []network.PortRange{{
FromPort: apiPort,
ToPort: apiPort,
Protocol: "tcp",
}}
c.Check(calls[0].PortRanges, jc.DeepEquals, expectPorts)
}
func (s *environNetSuite) TestPortsAPI(c *gc.C) {
fwname := gce.GlobalFirewallName(s.Env)
_, err := s.Env.Ports()
c.Assert(err, jc.ErrorIsNil)
c.Check(s.FakeConn.Calls, gc.HasLen, 1)
c.Check(s.FakeConn.Calls[0].FuncName, gc.Equals, "Ports")
c.Check(s.FakeConn.Calls[0].FirewallName, gc.Equals, fwname)
}
func (s *environNetSuite) TestClosePortsAPI(c *gc.C) {
fwname := gce.GlobalFirewallName(s.Env)
err := s.Env.ClosePorts(s.Ports)
c.Assert(err, jc.ErrorIsNil)
c.Check(s.FakeConn.Calls, gc.HasLen, 1)
c.Check(s.FakeConn.Calls[0].FuncName, gc.Equals, "ClosePorts")
c.Check(s.FakeConn.Calls[0].FirewallName, gc.Equals, fwname)
c.Check(s.FakeConn.Calls[0].PortRanges, jc.DeepEquals, s.Ports)
}
func (s *environNetSuite) TestGlobalFirewallName(c *gc.C) {
uuid := s.Config.UUID()
fwname := gce.GlobalFirewallName(s.Env)
c.Check(fwname, gc.Equals, "juju-"+uuid)
}
