github.com/cloud-green/juju@v0.0.0-20151002100041-a00291338d3d/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" 10 "github.com/juju/utils/os" 11 gc "gopkg.in/check.v1" 12 13 "github.com/juju/juju/cloudconfig/providerinit/renderers" 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 data := []byte("test") 27 result, err := renderer.EncodeUserdata(data, os.Ubuntu) 28 c.Assert(err, jc.ErrorIsNil) 29 c.Assert(result, jc.DeepEquals, utils.Gzip(data)) 30 31 data = []byte("test") 32 result, err = renderer.EncodeUserdata(data, os.CentOS) 33 c.Assert(err, jc.ErrorIsNil) 34 c.Assert(result, jc.DeepEquals, utils.Gzip(data)) 35 } 36 37 func (s *UserdataSuite) TestAmazonWindows(c *gc.C) { 38 renderer := ec2.AmazonRenderer{} 39 data := []byte("test") 40 result, err := renderer.EncodeUserdata(data, os.Windows) 41 c.Assert(err, jc.ErrorIsNil) 42 expected := []byte(`<powershell>` + 43 string(renderers.WinEmbedInScript(data)) + 44 `</powershell>`) 45 c.Assert(result, jc.DeepEquals, expected) 46 } 47 48 func (s *UserdataSuite) TestAmazonUnknownOS(c *gc.C) { 49 renderer := ec2.AmazonRenderer{} 50 result, err := renderer.EncodeUserdata(nil, os.Arch) 51 c.Assert(result, gc.IsNil) 52 c.Assert(err, gc.ErrorMatches, "Cannot encode userdata for OS: Arch") 53 }