@@ -18,13 +18,13 @@ package cloud_test
18
18
19
19
import (
20
20
"fmt"
21
- metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
22
21
23
22
capiv1 "sigs.k8s.io/cluster-api/api/v1beta1"
24
23
25
24
"github.com/apache/cloudstack-go/v2/cloudstack"
26
25
infrav1 "github.com/aws/cluster-api-provider-cloudstack/api/v1beta1"
27
26
"github.com/aws/cluster-api-provider-cloudstack/pkg/cloud"
27
+ dummies "github.com/aws/cluster-api-provider-cloudstack/pkg/cloud/test_dummies"
28
28
"github.com/golang/mock/gomock"
29
29
. "github.com/onsi/ginkgo"
30
30
. "github.com/onsi/ginkgo/extensions/table"
@@ -58,17 +58,7 @@ var _ = Describe("Instance", func() {
58
58
ts = mockClient .Template .(* cloudstack.MockTemplateServiceIface )
59
59
client = cloud .NewClientFromCSAPIClient (mockClient )
60
60
61
- csMachine = & infrav1.CloudStackMachine {
62
- Spec : infrav1.CloudStackMachineSpec {
63
- InstanceID : pointer .StringPtr ("instance-id" ),
64
- Offering : "service-offering-name" ,
65
- Template : "template-name" }}
66
- csMachine .Name = "instance-name"
67
- csCluster = & infrav1.CloudStackCluster {
68
- Spec : infrav1.CloudStackClusterSpec {},
69
- Status : infrav1.CloudStackClusterStatus {ZoneID : "zone-id" },
70
- ObjectMeta : metav1.ObjectMeta {UID : "0" }}
71
- machine = & capiv1.Machine {}
61
+ dummies .SetDummyVars ()
72
62
})
73
63
74
64
AfterEach (func () {
@@ -158,7 +148,7 @@ var _ = Describe("Instance", func() {
158
148
vms .EXPECT ().GetVirtualMachinesMetricByID (* csMachine .Spec .InstanceID ).Return (nil , - 1 , notFoundError )
159
149
vms .EXPECT ().GetVirtualMachinesMetricByName (csMachine .Name ).Return (nil , - 1 , notFoundError )
160
150
sos .EXPECT ().GetServiceOfferingID (csMachine .Spec .Offering ).Return (serviceOfferingID , 1 , nil )
161
- ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , csCluster . Status . ZoneID ).
151
+ ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , dummies . Zone1 ).
162
152
Return ("" , - 1 , unknownError )
163
153
ts .EXPECT ().GetTemplateByID (csMachine .Spec .Template , allFilter ).Return (nil , - 1 , unknownError )
164
154
Ω (client .GetOrCreateVMInstance (csMachine , machine , csCluster , "" )).ShouldNot (Succeed ())
@@ -168,7 +158,7 @@ var _ = Describe("Instance", func() {
168
158
vms .EXPECT ().GetVirtualMachinesMetricByID (* csMachine .Spec .InstanceID ).Return (nil , - 1 , notFoundError )
169
159
vms .EXPECT ().GetVirtualMachinesMetricByName (csMachine .Name ).Return (nil , - 1 , notFoundError )
170
160
sos .EXPECT ().GetServiceOfferingID (csMachine .Spec .Offering ).Return (serviceOfferingID , 1 , nil )
171
- ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , csCluster . Status . ZoneID ).Return ("" , 2 , nil )
161
+ ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , dummies . Zone1 ).Return ("" , 2 , nil )
172
162
ts .EXPECT ().GetTemplateByID (csMachine .Spec .Template , allFilter ).Return (nil , - 1 , unknownError )
173
163
Ω (client .GetOrCreateVMInstance (csMachine , machine , csCluster , "" )).ShouldNot (Succeed ())
174
164
})
@@ -177,9 +167,9 @@ var _ = Describe("Instance", func() {
177
167
vms .EXPECT ().GetVirtualMachinesMetricByID (* csMachine .Spec .InstanceID ).Return (nil , - 1 , notFoundError )
178
168
vms .EXPECT ().GetVirtualMachinesMetricByName (csMachine .Name ).Return (nil , - 1 , notFoundError )
179
169
sos .EXPECT ().GetServiceOfferingID (csMachine .Spec .Offering ).Return (serviceOfferingID , 1 , nil )
180
- ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , csCluster . Status . ZoneID ).
170
+ ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , dummies . Zone1 ).
181
171
Return (templateID , 1 , nil )
182
- vms .EXPECT ().NewDeployVirtualMachineParams (serviceOfferingID , templateID , csCluster . Status . ZoneID ).
172
+ vms .EXPECT ().NewDeployVirtualMachineParams (serviceOfferingID , templateID , dummies . Zone1 ).
183
173
Return (& cloudstack.DeployVirtualMachineParams {})
184
174
vms .EXPECT ().DeployVirtualMachine (gomock .Any ()).Return (nil , unknownError )
185
175
@@ -199,11 +189,11 @@ var _ = Describe("Instance", func() {
199
189
csCluster .Status .DomainID = domainID
200
190
vms .EXPECT ().GetVirtualMachinesMetricByID (* csMachine .Spec .InstanceID ).Return (nil , - 1 , notFoundError )
201
191
sos .EXPECT ().GetServiceOfferingID (csMachine .Spec .Offering ).Return (serviceOfferingID , 1 , nil )
202
- ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , csCluster . Status . ZoneID ).
192
+ ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , dummies . Zone1 ).
203
193
Return (templateID , 1 , nil )
204
194
vms .EXPECT ().GetVirtualMachinesMetricByID (* csMachine .Spec .InstanceID ).
205
195
Return (& cloudstack.VirtualMachinesMetric {}, 1 , nil )
206
- vms .EXPECT ().NewDeployVirtualMachineParams (serviceOfferingID , templateID , csCluster . Status . ZoneID ).
196
+ vms .EXPECT ().NewDeployVirtualMachineParams (serviceOfferingID , templateID , dummies . Zone1 ).
207
197
Return (& cloudstack.DeployVirtualMachineParams {})
208
198
vms .EXPECT ().DeployVirtualMachine (ParamMatch (And (AccountEquals (account ), DomainIDEquals (domainID )))).
209
199
Return (& cloudstack.DeployVirtualMachineResponse {Id : * csMachine .Spec .InstanceID }, nil )
@@ -227,7 +217,7 @@ var _ = Describe("Instance", func() {
227
217
})
228
218
229
219
ActionAndAssert := func () {
230
- vms .EXPECT ().NewDeployVirtualMachineParams (serviceOfferingID , templateID , csCluster . Status . ZoneID ).
220
+ vms .EXPECT ().NewDeployVirtualMachineParams (serviceOfferingID , templateID , dummies . Zone1 ).
231
221
Return (& cloudstack.DeployVirtualMachineParams {})
232
222
233
223
deploymentResp := & cloudstack.DeployVirtualMachineResponse {Id : * csMachine .Spec .InstanceID }
@@ -238,7 +228,7 @@ var _ = Describe("Instance", func() {
238
228
239
229
It ("works with service offering name and template name" , func () {
240
230
sos .EXPECT ().GetServiceOfferingID (csMachine .Spec .Offering ).Return (serviceOfferingID , 1 , nil )
241
- ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , csCluster . Status . ZoneID ).
231
+ ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , dummies . Zone1 ).
242
232
Return (templateID , 1 , nil )
243
233
244
234
ActionAndAssert ()
@@ -248,7 +238,7 @@ var _ = Describe("Instance", func() {
248
238
csMachine .Spec .Offering = serviceOfferingID
249
239
sos .EXPECT ().GetServiceOfferingID (csMachine .Spec .Offering ).Return ("" , - 1 , notFoundError )
250
240
sos .EXPECT ().GetServiceOfferingByID (csMachine .Spec .Offering ).Return (& cloudstack.ServiceOffering {}, 1 , nil )
251
- ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , csCluster . Status . ZoneID ).
241
+ ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , dummies . Zone1 ).
252
242
Return (templateID , 1 , nil )
253
243
254
244
ActionAndAssert ()
@@ -257,7 +247,7 @@ var _ = Describe("Instance", func() {
257
247
It ("works with service offering name and template ID" , func () {
258
248
csMachine .Spec .Template = templateID
259
249
sos .EXPECT ().GetServiceOfferingID (csMachine .Spec .Offering ).Return (serviceOfferingID , 1 , nil )
260
- ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , csCluster . Status . ZoneID ).
250
+ ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , dummies . Zone1 ).
261
251
Return ("" , - 1 , notFoundError )
262
252
ts .EXPECT ().GetTemplateByID (csMachine .Spec .Template , allFilter ).
263
253
Return (& cloudstack.Template {}, 1 , nil )
@@ -270,7 +260,7 @@ var _ = Describe("Instance", func() {
270
260
csMachine .Spec .Template = templateID
271
261
sos .EXPECT ().GetServiceOfferingID (csMachine .Spec .Offering ).Return ("" , - 1 , notFoundError )
272
262
sos .EXPECT ().GetServiceOfferingByID (csMachine .Spec .Offering ).Return (& cloudstack.ServiceOffering {}, 1 , nil )
273
- ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , csCluster . Status . ZoneID ).
263
+ ts .EXPECT ().GetTemplateID (csMachine .Spec .Template , allFilter , dummies . Zone1 ).
274
264
Return ("" , - 1 , notFoundError )
275
265
ts .EXPECT ().GetTemplateByID (csMachine .Spec .Template , allFilter ).
276
266
Return (& cloudstack.Template {}, 1 , nil )
0 commit comments