~rogpeppe/juju-core/azure

« back to all changes in this revision

Viewing changes to environs/azure/customdata_test.go

[r=jtv] Unify providers' user-data composition functions.

All the "physical" providers (EC2, OpenStack, MAAS, Azure) were generating
user data in essentially the same way. The MAAS one looked different but
actually just added an optional bit that was still completely generic. And so
this branch unifies the implementations.

Construction of the cloudinit.MachineConfig was lifted out of the function.
That may seem arbitrary, but it's part of a greater plan: in a later step of
the ongoing refactoring the construction can be lifted out of the providers'
internalStartInstance methods as well, doing away with the EC2 and OpenStack
providers' startInstanceParams structs. This actually shortens some of the
data flows and reduces cognitive load, which can only be good for maintenance.

After that step we can have a fresh look at sanitizing NewMachineConfig (which
arguably should take more parameters, or not exist at all, but is in an
intermediate state that facilitates the changes happening now). We may also
look into re-introducing some sort of dedicated Parameter Object for
internalStartInstance, but by then the cluster of methods Bootstrap /
StartInstance / internalStartInstance will look a bit different and we can
find the most maintainable solution for that new situation.

Show diffs side-by-side

added added

removed removed

Lines of Context:
8
8
 
9
9
        gc "launchpad.net/gocheck"
10
10
 
 
11
        "launchpad.net/juju-core/environs"
11
12
        "launchpad.net/juju-core/environs/cloudinit"
12
13
        "launchpad.net/juju-core/state"
13
14
        "launchpad.net/juju-core/state/api"
14
15
        "launchpad.net/juju-core/testing"
15
 
        "launchpad.net/juju-core/utils"
16
16
)
17
17
 
18
18
type CustomDataSuite struct{}
49
49
        return &cloudinit.MachineConfig{}
50
50
}
51
51
 
52
 
func (*CustomDataSuite) TestUserDataFailsOnBadData(c *gc.C) {
53
 
        _, err := userData(makeBadMachineConfig())
54
 
        c.Assert(err, gc.NotNil)
55
 
        c.Check(err, gc.ErrorMatches, "problem with cloudinit config: .*")
56
 
}
57
 
 
58
 
func (*CustomDataSuite) TestUserDataProducesZipFile(c *gc.C) {
59
 
        marker := "recognizable_text"
60
 
        cfg := makeMachineConfig(c)
61
 
        cfg.MachineNonce = marker
62
 
 
63
 
        data, err := userData(cfg)
64
 
        c.Assert(err, gc.IsNil)
65
 
 
66
 
        content, err := utils.Gunzip(data)
67
 
        c.Assert(err, gc.IsNil)
68
 
        // The regex syntax is weird here.  It's a trick to work around
69
 
        // gocheck's matcher prefixing "^" to the regex and suffixing "$".
70
 
        // With this trick, we can scan for a pattern in a multi-line input.
71
 
        c.Check(string(content), gc.Matches, "(?s).*"+marker+".*")
72
 
}
73
 
 
74
52
func (*CustomDataSuite) TestMakeCustomDataPropagatesError(c *gc.C) {
75
53
        _, err := makeCustomData(makeBadMachineConfig())
76
54
        c.Assert(err, gc.NotNil)
77
 
        c.Check(err, gc.ErrorMatches, "failure while generating custom data: problem with cloudinit config: .*")
 
55
        c.Check(err, gc.ErrorMatches, "failure while generating custom data: invalid machine configuration: invalid machine id")
78
56
}
79
57
 
80
58
func (*CustomDataSuite) TestMakeCustomDataEncodesUserData(c *gc.C) {
85
63
 
86
64
        data, err := base64.StdEncoding.DecodeString(encodedData)
87
65
        c.Assert(err, gc.IsNil)
88
 
        reference, err := userData(cfg)
 
66
        reference, err := environs.ComposeUserData(cfg)
89
67
        c.Assert(err, gc.IsNil)
90
68
        c.Check(data, gc.DeepEquals, reference)
91
69
}