← Back to branch summary

~nskaggs/+junk/xenial-test

« back to all changes in this revision

Viewing changes to src/github.com/juju/juju/provider/ec2/environ_test.go

  • Committer: Nicholas Skaggs
  • Date: 2016-10-24 20:56:05 UTC
  • Revision ID: nicholas.skaggs@canonical.com-20161024205605-z8lta0uvuhtxwzwl
Initi with beta15

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/github.com/juju/juju/provider/ec2/environ_test.go
 
1
// Copyright 2011, 2012, 2013 Canonical Ltd.
 
2
// Licensed under the AGPLv3, see LICENCE file for details.
 
3
 
 
4
package ec2
 
5
 
 
6
import (
 
7
        amzec2 "gopkg.in/amz.v3/ec2"
 
8
        gc "gopkg.in/check.v1"
 
9
 
 
10
        "github.com/juju/juju/constraints"
 
11
        "github.com/juju/juju/environs"
 
12
        "github.com/juju/juju/environs/simplestreams"
 
13
        "github.com/juju/juju/instance"
 
14
        "github.com/juju/juju/network"
 
15
        "github.com/juju/juju/state"
 
16
)
 
17
 
 
18
// Ensure EC2 provider supports the expected interfaces,
 
19
var (
 
20
        _ environs.NetworkingEnviron = (*environ)(nil)
 
21
        _ simplestreams.HasRegion    = (*environ)(nil)
 
22
        _ state.Prechecker           = (*environ)(nil)
 
23
        _ instance.Distributor       = (*environ)(nil)
 
24
)
 
25
 
 
26
type Suite struct{}
 
27
 
 
28
var _ = gc.Suite(&Suite{})
 
29
 
 
30
type RootDiskTest struct {
 
31
        series     string
 
32
        name       string
 
33
        constraint *uint64
 
34
        device     amzec2.BlockDeviceMapping
 
35
}
 
36
 
 
37
var commonInstanceStoreDisks = []amzec2.BlockDeviceMapping{{
 
38
        DeviceName:  "/dev/sdb",
 
39
        VirtualName: "ephemeral0",
 
40
}, {
 
41
        DeviceName:  "/dev/sdc",
 
42
        VirtualName: "ephemeral1",
 
43
}, {
 
44
        DeviceName:  "/dev/sdd",
 
45
        VirtualName: "ephemeral2",
 
46
}, {
 
47
        DeviceName:  "/dev/sde",
 
48
        VirtualName: "ephemeral3",
 
49
}}
 
50
 
 
51
func (*Suite) TestRootDiskBlockDeviceMapping(c *gc.C) {
 
52
        var rootDiskTests = []RootDiskTest{{
 
53
                "trusty",
 
54
                "nil constraint ubuntu",
 
55
                nil,
 
56
                amzec2.BlockDeviceMapping{VolumeSize: 8, DeviceName: "/dev/sda1"},
 
57
        }, {
 
58
                "trusty",
 
59
                "too small constraint ubuntu",
 
60
                pInt(4000),
 
61
                amzec2.BlockDeviceMapping{VolumeSize: 8, DeviceName: "/dev/sda1"},
 
62
        }, {
 
63
                "trusty",
 
64
                "big constraint ubuntu",
 
65
                pInt(20 * 1024),
 
66
                amzec2.BlockDeviceMapping{VolumeSize: 20, DeviceName: "/dev/sda1"},
 
67
        }, {
 
68
                "trusty",
 
69
                "round up constraint ubuntu",
 
70
                pInt(20*1024 + 1),
 
71
                amzec2.BlockDeviceMapping{VolumeSize: 21, DeviceName: "/dev/sda1"},
 
72
        }, {
 
73
                "win2012r2",
 
74
                "nil constraint windows",
 
75
                nil,
 
76
                amzec2.BlockDeviceMapping{VolumeSize: 40, DeviceName: "/dev/sda1"},
 
77
        }, {
 
78
                "win2012r2",
 
79
                "too small constraint windows",
 
80
                pInt(30 * 1024),
 
81
                amzec2.BlockDeviceMapping{VolumeSize: 40, DeviceName: "/dev/sda1"},
 
82
        }, {
 
83
                "win2012r2",
 
84
                "big constraint windows",
 
85
                pInt(50 * 1024),
 
86
                amzec2.BlockDeviceMapping{VolumeSize: 50, DeviceName: "/dev/sda1"},
 
87
        }, {
 
88
                "win2012r2",
 
89
                "round up constraint windows",
 
90
                pInt(50*1024 + 1),
 
91
                amzec2.BlockDeviceMapping{VolumeSize: 51, DeviceName: "/dev/sda1"},
 
92
        }}
 
93
 
 
94
        for _, t := range rootDiskTests {
 
95
                c.Logf("Test %s", t.name)
 
96
                cons := constraints.Value{RootDisk: t.constraint}
 
97
                mappings := getBlockDeviceMappings(cons, t.series)
 
98
                expected := append([]amzec2.BlockDeviceMapping{t.device}, commonInstanceStoreDisks...)
 
99
                c.Assert(mappings, gc.DeepEquals, expected)
 
100
        }
 
101
}
 
102
 
 
103
func pInt(i uint64) *uint64 {
 
104
        return &i
 
105
}
 
106
 
 
107
func (*Suite) TestPortsToIPPerms(c *gc.C) {
 
108
        testCases := []struct {
 
109
                about    string
 
110
                ports    []network.PortRange
 
111
                expected []amzec2.IPPerm
 
112
        }{{
 
113
                about: "single port",
 
114
                ports: []network.PortRange{{
 
115
                        FromPort: 80,
 
116
                        ToPort:   80,
 
117
                        Protocol: "tcp",
 
118
                }},
 
119
                expected: []amzec2.IPPerm{{
 
120
                        Protocol:  "tcp",
 
121
                        FromPort:  80,
 
122
                        ToPort:    80,
 
123
                        SourceIPs: []string{"0.0.0.0/0"},
 
124
                }},
 
125
        }, {
 
126
                about: "multiple ports",
 
127
                ports: []network.PortRange{{
 
128
                        FromPort: 80,
 
129
                        ToPort:   82,
 
130
                        Protocol: "tcp",
 
131
                }},
 
132
                expected: []amzec2.IPPerm{{
 
133
                        Protocol:  "tcp",
 
134
                        FromPort:  80,
 
135
                        ToPort:    82,
 
136
                        SourceIPs: []string{"0.0.0.0/0"},
 
137
                }},
 
138
        }, {
 
139
                about: "multiple port ranges",
 
140
                ports: []network.PortRange{{
 
141
                        FromPort: 80,
 
142
                        ToPort:   82,
 
143
                        Protocol: "tcp",
 
144
                }, {
 
145
                        FromPort: 100,
 
146
                        ToPort:   120,
 
147
                        Protocol: "tcp",
 
148
                }},
 
149
                expected: []amzec2.IPPerm{{
 
150
                        Protocol:  "tcp",
 
151
                        FromPort:  80,
 
152
                        ToPort:    82,
 
153
                        SourceIPs: []string{"0.0.0.0/0"},
 
154
                }, {
 
155
                        Protocol:  "tcp",
 
156
                        FromPort:  100,
 
157
                        ToPort:    120,
 
158
                        SourceIPs: []string{"0.0.0.0/0"},
 
159
                }},
 
160
        }}
 
161
 
 
162
        for i, t := range testCases {
 
163
                c.Logf("test %d: %s", i, t.about)
 
164
                ipperms := portsToIPPerms(t.ports)
 
165
                c.Assert(ipperms, gc.DeepEquals, t.expected)
 
166
        }
 
167
}