6
6
"fmt"
7
7
"testing"
8
8
9
- "github.com/aws/aws-cloud-map-mcs-controller-for-k8s/mocks/pkg/cloudmap"
9
+ cloudmapMock "github.com/aws/aws-cloud-map-mcs-controller-for-k8s/mocks/pkg/cloudmap"
10
10
"github.com/aws/aws-cloud-map-mcs-controller-for-k8s/pkg/common"
11
11
"github.com/aws/aws-cloud-map-mcs-controller-for-k8s/pkg/model"
12
12
"github.com/aws/aws-cloud-map-mcs-controller-for-k8s/test"
@@ -27,7 +27,7 @@ func TestServiceDiscoveryApi_ListNamespaces_HappyCase(t *testing.T) {
27
27
mockController := gomock .NewController (t )
28
28
defer mockController .Finish ()
29
29
30
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
30
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
31
31
sdApi := getServiceDiscoveryApi (t , awsFacade )
32
32
33
33
id , name := test .NsId , test .NsName
@@ -48,7 +48,7 @@ func TestServiceDiscoveryApi_ListNamespaces_SkipPublicDNSNotSupported(t *testing
48
48
mockController := gomock .NewController (t )
49
49
defer mockController .Finish ()
50
50
51
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
51
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
52
52
sdApi := getServiceDiscoveryApi (t , awsFacade )
53
53
54
54
id , name := test .NsId , test .NsName
@@ -68,7 +68,7 @@ func TestServiceDiscoveryApi_ListServices_HappyCase(t *testing.T) {
68
68
mockController := gomock .NewController (t )
69
69
defer mockController .Finish ()
70
70
71
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
71
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
72
72
sdApi := getServiceDiscoveryApi (t , awsFacade )
73
73
74
74
filter := types.ServiceFilter {
@@ -91,7 +91,7 @@ func TestServiceDiscoveryApi_DiscoverInstances_HappyCase(t *testing.T) {
91
91
mockController := gomock .NewController (t )
92
92
defer mockController .Finish ()
93
93
94
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
94
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
95
95
sdApi := getServiceDiscoveryApi (t , awsFacade )
96
96
97
97
awsFacade .EXPECT ().DiscoverInstances (context .TODO (),
@@ -119,7 +119,7 @@ func TestServiceDiscoveryApi_ListOperations_HappyCase(t *testing.T) {
119
119
mockController := gomock .NewController (t )
120
120
defer mockController .Finish ()
121
121
122
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
122
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
123
123
sdApi := getServiceDiscoveryApi (t , awsFacade )
124
124
125
125
filters := make ([]types.OperationFilter , 0 )
@@ -139,7 +139,7 @@ func TestServiceDiscoveryApi_GetOperation_HappyCase(t *testing.T) {
139
139
mockController := gomock .NewController (t )
140
140
defer mockController .Finish ()
141
141
142
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
142
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
143
143
sdApi := getServiceDiscoveryApi (t , awsFacade )
144
144
145
145
expectedOp := & types.Operation {Id : aws .String (test .OpId1 ), Status : types .OperationStatusPending }
@@ -155,7 +155,7 @@ func TestServiceDiscoveryApi_CreateHttNamespace_HappyCase(t *testing.T) {
155
155
mockController := gomock .NewController (t )
156
156
defer mockController .Finish ()
157
157
158
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
158
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
159
159
sdApi := getServiceDiscoveryApi (t , awsFacade )
160
160
161
161
awsFacade .EXPECT ().CreateHttpNamespace (context .TODO (), & sd.CreateHttpNamespaceInput {Name : aws .String (test .NsName )}).
@@ -170,7 +170,7 @@ func TestServiceDiscoveryApi_CreateService_CreateForHttpNamespace(t *testing.T)
170
170
mockController := gomock .NewController (t )
171
171
defer mockController .Finish ()
172
172
173
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
173
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
174
174
sdApi := getServiceDiscoveryApi (t , awsFacade )
175
175
176
176
nsId , svcId , svcName := test .NsId , test .SvcId , test .SvcName
@@ -192,7 +192,7 @@ func TestServiceDiscoveryApi_CreateService_CreateForDnsNamespace(t *testing.T) {
192
192
mockController := gomock .NewController (t )
193
193
defer mockController .Finish ()
194
194
195
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
195
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
196
196
sdApi := getServiceDiscoveryApi (t , awsFacade )
197
197
198
198
nsId , svcId , svcName := test .NsId , test .SvcId , test .SvcName
@@ -220,7 +220,7 @@ func TestServiceDiscoveryApi_CreateService_ThrowError(t *testing.T) {
220
220
mockController := gomock .NewController (t )
221
221
defer mockController .Finish ()
222
222
223
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
223
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
224
224
sdApi := getServiceDiscoveryApi (t , awsFacade )
225
225
226
226
nsId , svcName := test .NsId , test .SvcName
@@ -241,7 +241,7 @@ func TestServiceDiscoveryApi_RegisterInstance_HappyCase(t *testing.T) {
241
241
242
242
attrs := map [string ]string {"a" : "b" }
243
243
244
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
244
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
245
245
awsFacade .EXPECT ().RegisterInstance (context .TODO (),
246
246
& sd.RegisterInstanceInput {
247
247
ServiceId : aws .String (test .SvcId ),
@@ -260,7 +260,7 @@ func TestServiceDiscoveryApi_RegisterInstance_Error(t *testing.T) {
260
260
defer mockController .Finish ()
261
261
262
262
sdkErr := errors .New ("fail" )
263
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
263
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
264
264
awsFacade .EXPECT ().RegisterInstance (context .TODO (), gomock .Any ()).Return (nil , sdkErr )
265
265
266
266
sdApi := getServiceDiscoveryApi (t , awsFacade )
@@ -272,7 +272,7 @@ func TestServiceDiscoveryApi_DeregisterInstance_HappyCase(t *testing.T) {
272
272
mockController := gomock .NewController (t )
273
273
defer mockController .Finish ()
274
274
275
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
275
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
276
276
awsFacade .EXPECT ().DeregisterInstance (context .TODO (),
277
277
& sd.DeregisterInstanceInput {
278
278
ServiceId : aws .String (test .SvcId ),
@@ -290,7 +290,7 @@ func TestServiceDiscoveryApi_DeregisterInstance_Error(t *testing.T) {
290
290
defer mockController .Finish ()
291
291
292
292
sdkErr := errors .New ("fail" )
293
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
293
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
294
294
awsFacade .EXPECT ().DeregisterInstance (context .TODO (), gomock .Any ()).Return (nil , sdkErr )
295
295
296
296
sdApi := getServiceDiscoveryApi (t , awsFacade )
@@ -302,7 +302,7 @@ func TestServiceDiscoveryApi_PollNamespaceOperation_HappyCase(t *testing.T) {
302
302
mockController := gomock .NewController (t )
303
303
defer mockController .Finish ()
304
304
305
- awsFacade := cloudmap .NewMockAwsFacade (mockController )
305
+ awsFacade := cloudmapMock .NewMockAwsFacade (mockController )
306
306
awsFacade .EXPECT ().GetOperation (context .TODO (), & sd.GetOperationInput {OperationId : aws .String (test .OpId1 )}).
307
307
Return (& sd.GetOperationOutput {Operation : & types.Operation {Status : types .OperationStatusPending }}, nil )
308
308
@@ -317,7 +317,7 @@ func TestServiceDiscoveryApi_PollNamespaceOperation_HappyCase(t *testing.T) {
317
317
assert .Equal (t , test .NsId , nsId )
318
318
}
319
319
320
- func getServiceDiscoveryApi (t * testing.T , awsFacade * cloudmap .MockAwsFacade ) ServiceDiscoveryApi {
320
+ func getServiceDiscoveryApi (t * testing.T , awsFacade * cloudmapMock .MockAwsFacade ) ServiceDiscoveryApi {
321
321
return & serviceDiscoveryApi {
322
322
log : common .NewLoggerWithLogr (testingLogger.TestLogger {T : t }),
323
323
awsFacade : awsFacade ,
0 commit comments