github.com/niedbalski/juju@v0.0.0-20190215020005-8ff100488e47/provider/ec2/userdata_test.go (about)

     1  // Copyright 2015 Canonical Ltd.
     2  // Copyright 2015 Cloudbase Solutions SRL
     3  // Licensed under the AGPLv3, see LICENCE file for details.
     4  
     5  package ec2_test
     6  
     7  import (
     8  	"github.com/juju/os"
     9  	jc "github.com/juju/testing/checkers"
    10  	"github.com/juju/utils"
    11  	gc "gopkg.in/check.v1"
    12  
    13  	"github.com/juju/juju/cloudconfig/cloudinit/cloudinittest"
    14  	"github.com/juju/juju/cloudconfig/providerinit/renderers"
    15  	"github.com/juju/juju/provider/ec2"
    16  	"github.com/juju/juju/testing"
    17  )
    18  
    19  type UserdataSuite struct {
    20  	testing.BaseSuite
    21  }
    22  
    23  var _ = gc.Suite(&UserdataSuite{})
    24  
    25  func (s *UserdataSuite) TestAmazonUnix(c *gc.C) {
    26  	renderer := ec2.AmazonRenderer{}
    27  	cloudcfg := &cloudinittest.CloudConfig{YAML: []byte("yaml")}
    28  
    29  	result, err := renderer.Render(cloudcfg, os.Ubuntu)
    30  	c.Assert(err, jc.ErrorIsNil)
    31  	c.Assert(result, jc.DeepEquals, utils.Gzip(cloudcfg.YAML))
    32  
    33  	result, err = renderer.Render(cloudcfg, os.CentOS)
    34  	c.Assert(err, jc.ErrorIsNil)
    35  	c.Assert(result, jc.DeepEquals, utils.Gzip(cloudcfg.YAML))
    36  }
    37  
    38  func (s *UserdataSuite) TestAmazonWindows(c *gc.C) {
    39  	renderer := ec2.AmazonRenderer{}
    40  	cloudcfg := &cloudinittest.CloudConfig{YAML: []byte("yaml")}
    41  
    42  	result, err := renderer.Render(cloudcfg, os.Windows)
    43  	c.Assert(err, jc.ErrorIsNil)
    44  	expected := []byte(`<powershell>` +
    45  		string(renderers.WinEmbedInScript(cloudcfg.YAML)) +
    46  		`</powershell>`)
    47  	c.Assert(result, jc.DeepEquals, expected)
    48  }
    49  
    50  func (s *UserdataSuite) TestAmazonUnknownOS(c *gc.C) {
    51  	renderer := ec2.AmazonRenderer{}
    52  	cloudcfg := &cloudinittest.CloudConfig{}
    53  	result, err := renderer.Render(cloudcfg, os.GenericLinux)
    54  	c.Assert(result, gc.IsNil)
    55  	c.Assert(err, gc.ErrorMatches, "Cannot encode userdata for OS: GenericLinux")
    56  }