github.com/cloud-green/juju@v0.0.0-20151002100041-a00291338d3d/provider/joyent/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 joyent_test
     6  
     7  import (
     8  	jc "github.com/juju/testing/checkers"
     9  	gc "gopkg.in/check.v1"
    10  
    11  	"github.com/juju/juju/provider/joyent"
    12  	"github.com/juju/juju/testing"
    13  	"github.com/juju/utils/os"
    14  )
    15  
    16  type UserdataSuite struct {
    17  	testing.BaseSuite
    18  }
    19  
    20  var _ = gc.Suite(&UserdataSuite{})
    21  
    22  func (s *UserdataSuite) TestJoyentUnix(c *gc.C) {
    23  	renderer := joyent.JoyentRenderer{}
    24  	data := []byte("test")
    25  	result, err := renderer.EncodeUserdata(data, os.Ubuntu)
    26  	c.Assert(err, jc.ErrorIsNil)
    27  	c.Assert(result, jc.DeepEquals, data)
    28  
    29  	data = []byte("test")
    30  	result, err = renderer.EncodeUserdata(data, os.CentOS)
    31  	c.Assert(err, jc.ErrorIsNil)
    32  	c.Assert(result, jc.DeepEquals, data)
    33  }
    34  
    35  func (s *UserdataSuite) TestJoyentUnknownOS(c *gc.C) {
    36  	renderer := joyent.JoyentRenderer{}
    37  	result, err := renderer.EncodeUserdata(nil, os.Windows)
    38  	c.Assert(result, gc.IsNil)
    39  	c.Assert(err, gc.ErrorMatches, "Cannot encode userdata for OS: Windows")
    40  }