github.com/axw/juju@v0.0.0-20161005053422-4bd6544d08d4/provider/gce/environ_network_test.go (about)

     1  // Copyright 2014 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  package gce_test
     5  
     6  import (
     7  	jc "github.com/juju/testing/checkers"
     8  	gc "gopkg.in/check.v1"
     9  
    10  	"github.com/juju/juju/provider/gce"
    11  )
    12  
    13  type environNetSuite struct {
    14  	gce.BaseSuite
    15  }
    16  
    17  var _ = gc.Suite(&environNetSuite{})
    18  
    19  func (s *environNetSuite) TestGlobalFirewallName(c *gc.C) {
    20  	uuid := s.Config.UUID()
    21  	fwname := gce.GlobalFirewallName(s.Env)
    22  
    23  	c.Check(fwname, gc.Equals, "juju-"+uuid)
    24  }
    25  
    26  func (s *environNetSuite) TestOpenPorts(c *gc.C) {
    27  	err := s.Env.OpenPorts(s.Ports)
    28  
    29  	c.Check(err, jc.ErrorIsNil)
    30  }
    31  
    32  func (s *environNetSuite) TestOpenPortsAPI(c *gc.C) {
    33  	fwname := gce.GlobalFirewallName(s.Env)
    34  	err := s.Env.OpenPorts(s.Ports)
    35  	c.Assert(err, jc.ErrorIsNil)
    36  
    37  	c.Check(s.FakeConn.Calls, gc.HasLen, 1)
    38  	c.Check(s.FakeConn.Calls[0].FuncName, gc.Equals, "OpenPorts")
    39  	c.Check(s.FakeConn.Calls[0].FirewallName, gc.Equals, fwname)
    40  	c.Check(s.FakeConn.Calls[0].PortRanges, jc.DeepEquals, s.Ports)
    41  }
    42  
    43  func (s *environNetSuite) TestClosePorts(c *gc.C) {
    44  	err := s.Env.ClosePorts(s.Ports)
    45  
    46  	c.Check(err, jc.ErrorIsNil)
    47  }
    48  
    49  func (s *environNetSuite) TestClosePortsAPI(c *gc.C) {
    50  	fwname := gce.GlobalFirewallName(s.Env)
    51  	err := s.Env.ClosePorts(s.Ports)
    52  	c.Assert(err, jc.ErrorIsNil)
    53  
    54  	c.Check(s.FakeConn.Calls, gc.HasLen, 1)
    55  	c.Check(s.FakeConn.Calls[0].FuncName, gc.Equals, "ClosePorts")
    56  	c.Check(s.FakeConn.Calls[0].FirewallName, gc.Equals, fwname)
    57  	c.Check(s.FakeConn.Calls[0].PortRanges, jc.DeepEquals, s.Ports)
    58  }
    59  
    60  func (s *environNetSuite) TestPorts(c *gc.C) {
    61  	s.FakeConn.PortRanges = s.Ports
    62  
    63  	ports, err := s.Env.Ports()
    64  	c.Assert(err, jc.ErrorIsNil)
    65  
    66  	c.Check(ports, jc.DeepEquals, s.Ports)
    67  }
    68  
    69  func (s *environNetSuite) TestPortsAPI(c *gc.C) {
    70  	fwname := gce.GlobalFirewallName(s.Env)
    71  	_, err := s.Env.Ports()
    72  	c.Assert(err, jc.ErrorIsNil)
    73  
    74  	c.Check(s.FakeConn.Calls, gc.HasLen, 1)
    75  	c.Check(s.FakeConn.Calls[0].FuncName, gc.Equals, "Ports")
    76  	c.Check(s.FakeConn.Calls[0].FirewallName, gc.Equals, fwname)
    77  }