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