github.com/makyo/juju@v0.0.0-20160425123129-2608902037e9/provider/lxd/environ_broker_test.go (about)

     1  // Copyright 2015 Canonical Ltd.
     2  // Licensed under the AGPLv3, see LICENCE file for details.
     3  
     4  // +build go1.3
     5  
     6  package lxd_test
     7  
     8  import (
     9  	gitjujutesting "github.com/juju/testing"
    10  	jc "github.com/juju/testing/checkers"
    11  	"github.com/juju/utils/arch"
    12  	gc "gopkg.in/check.v1"
    13  
    14  	"github.com/juju/juju/provider/lxd"
    15  )
    16  
    17  type environBrokerSuite struct {
    18  	lxd.BaseSuite
    19  }
    20  
    21  var _ = gc.Suite(&environBrokerSuite{})
    22  
    23  func (s *environBrokerSuite) SetUpTest(c *gc.C) {
    24  	s.BaseSuite.SetUpTest(c)
    25  }
    26  
    27  func (s *environBrokerSuite) TestStartInstance(c *gc.C) {
    28  	s.Client.Inst = s.RawInstance
    29  
    30  	// Patch the host's arch, so the broker will filter tools.
    31  	s.PatchValue(&arch.HostArch, func() string { return arch.ARM64 })
    32  
    33  	result, err := s.Env.StartInstance(s.StartInstArgs)
    34  
    35  	c.Assert(err, jc.ErrorIsNil)
    36  	c.Check(result.Instance, gc.DeepEquals, s.Instance)
    37  	c.Check(result.Hardware, gc.DeepEquals, s.HWC)
    38  	c.Assert(s.StartInstArgs.InstanceConfig.AgentVersion().Arch, gc.Equals, arch.ARM64)
    39  }
    40  
    41  func (s *environBrokerSuite) TestStartInstanceNoTools(c *gc.C) {
    42  	s.Client.Inst = s.RawInstance
    43  
    44  	// Patch the host's arch, so the broker will filter tools.
    45  	s.PatchValue(&arch.HostArch, func() string { return arch.PPC64EL })
    46  
    47  	_, err := s.Env.StartInstance(s.StartInstArgs)
    48  	c.Assert(err, gc.ErrorMatches, "no matching tools available")
    49  }
    50  
    51  func (s *environBrokerSuite) TestStopInstances(c *gc.C) {
    52  	err := s.Env.StopInstances(s.Instance.Id())
    53  	c.Assert(err, jc.ErrorIsNil)
    54  
    55  	s.Stub.CheckCalls(c, []gitjujutesting.StubCall{{
    56  		FuncName: "RemoveInstances",
    57  		Args: []interface{}{
    58  			"juju-2d02eeac-9dbb-11e4-89d3-123b93f75cba-machine-",
    59  			[]string{"spam"},
    60  		},
    61  	}})
    62  }
    63  
    64  func (s *environBrokerSuite) TestImageMetadataURL(c *gc.C) {
    65  	s.UpdateConfig(c, map[string]interface{}{
    66  		"image-metadata-url": "https://my-test.com/images/",
    67  	})
    68  	s.checkSources(c, []string{
    69  		"https://my-test.com/images/",
    70  		"https://streams.canonical.com/juju/images/releases/",
    71  		"https://cloud-images.ubuntu.com/releases/",
    72  	})
    73  }
    74  
    75  func (s *environBrokerSuite) TestImageMetadataURLMungesHTTP(c *gc.C) {
    76  	// LXD requires 'https://' hosts for simplestreams data.
    77  	// https://github.com/lxc/lxd/issues/1763
    78  	s.UpdateConfig(c, map[string]interface{}{
    79  		"image-metadata-url": "http://my-test.com/images/",
    80  	})
    81  	s.checkSources(c, []string{
    82  		"https://my-test.com/images/",
    83  		"https://streams.canonical.com/juju/images/releases/",
    84  		"https://cloud-images.ubuntu.com/releases/",
    85  	})
    86  }
    87  
    88  func (s *environBrokerSuite) checkSources(c *gc.C, expectedURLs []string) {
    89  	sources, err := lxd.GetImageSources(s.Env)
    90  	c.Assert(err, jc.ErrorIsNil)
    91  	var sourceURLs []string
    92  	for _, source := range sources {
    93  		sourceURLs = append(sourceURLs, source.Host)
    94  	}
    95  	c.Check(sourceURLs, gc.DeepEquals, expectedURLs)
    96  }
    97  
    98  func (s *environBrokerSuite) checkSourcesFromStream(c *gc.C, stream string, expectedURLs []string) {
    99  	if stream != "" {
   100  		s.UpdateConfig(c, map[string]interface{}{"image-stream": stream})
   101  	}
   102  	s.checkSources(c, expectedURLs)
   103  }
   104  
   105  func (s *environBrokerSuite) TestImageStreamDefault(c *gc.C) {
   106  	s.checkSourcesFromStream(c, "", []string{
   107  		"https://streams.canonical.com/juju/images/releases/",
   108  		"https://cloud-images.ubuntu.com/releases/",
   109  	})
   110  }
   111  
   112  func (s *environBrokerSuite) TestImageStreamReleased(c *gc.C) {
   113  	s.checkSourcesFromStream(c, "released", []string{
   114  		"https://streams.canonical.com/juju/images/releases/",
   115  		"https://cloud-images.ubuntu.com/releases/",
   116  	})
   117  }
   118  
   119  func (s *environBrokerSuite) TestImageStreamDaily(c *gc.C) {
   120  	s.checkSourcesFromStream(c, "daily", []string{
   121  		"https://streams.canonical.com/juju/images/daily/",
   122  		"https://cloud-images.ubuntu.com/daily/",
   123  	})
   124  }