github.com/juju/juju@v0.0.0-20240430160146-1752b71fcf00/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  	jc "github.com/juju/testing/checkers"
     9  	"github.com/juju/utils/v3"
    10  	gc "gopkg.in/check.v1"
    11  
    12  	"github.com/juju/juju/cloudconfig/cloudinit/cloudinittest"
    13  	"github.com/juju/juju/core/os/ostype"
    14  	"github.com/juju/juju/provider/ec2"
    15  	"github.com/juju/juju/testing"
    16  )
    17  
    18  type UserdataSuite struct {
    19  	testing.BaseSuite
    20  }
    21  
    22  var _ = gc.Suite(&UserdataSuite{})
    23  
    24  func (s *UserdataSuite) TestAmazonUnix(c *gc.C) {
    25  	renderer := ec2.AmazonRenderer{}
    26  	cloudcfg := &cloudinittest.CloudConfig{YAML: []byte("yaml")}
    27  
    28  	result, err := renderer.Render(cloudcfg, ostype.Ubuntu)
    29  	c.Assert(err, jc.ErrorIsNil)
    30  	c.Assert(result, jc.DeepEquals, utils.Gzip(cloudcfg.YAML))
    31  
    32  	result, err = renderer.Render(cloudcfg, ostype.CentOS)
    33  	c.Assert(err, jc.ErrorIsNil)
    34  	c.Assert(result, jc.DeepEquals, utils.Gzip(cloudcfg.YAML))
    35  }
    36  
    37  func (s *UserdataSuite) TestAmazonUnknownOS(c *gc.C) {
    38  	renderer := ec2.AmazonRenderer{}
    39  	cloudcfg := &cloudinittest.CloudConfig{}
    40  	result, err := renderer.Render(cloudcfg, ostype.GenericLinux)
    41  	c.Assert(result, gc.IsNil)
    42  	c.Assert(err, gc.ErrorMatches, "Cannot encode userdata for OS: GenericLinux")
    43  }