@@ -87,20 +87,20 @@ func TestGetServices(t *testing.T) {
87
87
server2 := CreateTestServer (t )
88
88
defer server2 .Stop ()
89
89
90
- server1 .JoinWAN (server2 .LANAddr )
90
+ server1 .JoinWAN (t , server2 .LANAddr )
91
91
92
92
// create client
93
93
consul := ClientAtServer (server1 )
94
94
consul .config .Tag = "marathon"
95
95
96
96
// given
97
97
// register services in both servers
98
- server1 .AddService ("serviceA" , "passing" , []string {"public" , "marathon" })
99
- server1 .AddService ("serviceB" , "passing" , []string {"marathon" })
100
- server1 .AddService ("serviceC" , "passing" , []string {"marathon" })
98
+ server1 .AddService (t , "serviceA" , "passing" , []string {"public" , "marathon" })
99
+ server1 .AddService (t , "serviceB" , "passing" , []string {"marathon" })
100
+ server1 .AddService (t , "serviceC" , "passing" , []string {"marathon" })
101
101
102
- server2 .AddService ("serviceA" , "passing" , []string {"private" , "marathon" })
103
- server2 .AddService ("serviceB" , "passing" , []string {"zookeeper" })
102
+ server2 .AddService (t , "serviceA" , "passing" , []string {"private" , "marathon" })
103
+ server2 .AddService (t , "serviceB" , "passing" , []string {"zookeeper" })
104
104
105
105
// when
106
106
services , err := consul .GetServices ("serviceA" )
@@ -171,8 +171,8 @@ func TestGetServices_RemovingFailingAgentsAndRetrying(t *testing.T) {
171
171
consul .config .RequestRetries = 10
172
172
173
173
// given
174
- server1 .AddService ("serviceA" , "passing" , []string {"public" , "marathon" })
175
- server1 .AddService ("serviceB" , "passing" , []string {"marathon" })
174
+ server1 .AddService (t , "serviceA" , "passing" , []string {"public" , "marathon" })
175
+ server1 .AddService (t , "serviceB" , "passing" , []string {"marathon" })
176
176
177
177
// add failing clients
178
178
for i := uint32 (2 ); i < consul .config .RequestRetries ; i ++ {
@@ -196,20 +196,20 @@ func TestGetServices_SelectOnlyTaggedServices(t *testing.T) {
196
196
server2 := CreateTestServer (t )
197
197
defer server2 .Stop ()
198
198
199
- server1 .JoinWAN (server2 .LANAddr )
199
+ server1 .JoinWAN (t , server2 .LANAddr )
200
200
201
201
// create client
202
202
consul := ClientAtServer (server1 )
203
203
consul .config .Tag = "marathon-mycluster"
204
204
205
205
// given
206
206
// register services in both servers
207
- server1 .AddService ("serviceA" , "passing" , []string {"public" , "marathon-mycluster" })
208
- server1 .AddService ("serviceB" , "passing" , []string {"marathon" })
209
- server1 .AddService ("serviceC" , "passing" , []string {"marathon" })
207
+ server1 .AddService (t , "serviceA" , "passing" , []string {"public" , "marathon-mycluster" })
208
+ server1 .AddService (t , "serviceB" , "passing" , []string {"marathon" })
209
+ server1 .AddService (t , "serviceC" , "passing" , []string {"marathon" })
210
210
211
- server2 .AddService ("serviceA" , "passing" , []string {"private" , "marathon" })
212
- server2 .AddService ("serviceB" , "passing" , []string {"zookeeper" })
211
+ server2 .AddService (t , "serviceA" , "passing" , []string {"private" , "marathon" })
212
+ server2 .AddService (t , "serviceB" , "passing" , []string {"zookeeper" })
213
213
214
214
// when
215
215
services , err := consul .GetServices ("serviceA" )
@@ -236,20 +236,20 @@ func TestGetAllServices(t *testing.T) {
236
236
server2 := CreateTestServer (t )
237
237
defer server2 .Stop ()
238
238
239
- server1 .JoinWAN (server2 .LANAddr )
239
+ server1 .JoinWAN (t , server2 .LANAddr )
240
240
241
241
// create client
242
242
consul := ClientAtServer (server1 )
243
243
consul .config .Tag = "marathon"
244
244
245
245
// given
246
246
// register services in both servers
247
- server1 .AddService ("serviceA" , "passing" , []string {"public" , "marathon" })
248
- server1 .AddService ("serviceB" , "passing" , []string {"marathon" })
249
- server1 .AddService ("serviceC" , "passing" , []string {"zookeeper" })
247
+ server1 .AddService (t , "serviceA" , "passing" , []string {"public" , "marathon" })
248
+ server1 .AddService (t , "serviceB" , "passing" , []string {"marathon" })
249
+ server1 .AddService (t , "serviceC" , "passing" , []string {"zookeeper" })
250
250
251
- server2 .AddService ("serviceA" , "passing" , []string {"private" , "marathon" })
252
- server2 .AddService ("serviceB" , "passing" , []string {"zookeeper" })
251
+ server2 .AddService (t , "serviceA" , "passing" , []string {"private" , "marathon" })
252
+ server2 .AddService (t , "serviceB" , "passing" , []string {"zookeeper" })
253
253
254
254
// when
255
255
services , err := consul .GetAllServices ()
@@ -319,7 +319,7 @@ func TestGetAllServices_RemovingFailingAgentsAndRetrying(t *testing.T) {
319
319
server2 := CreateTestServer (t )
320
320
defer server2 .Stop ()
321
321
322
- server1 .JoinWAN (server2 .LANAddr )
322
+ server1 .JoinWAN (t , server2 .LANAddr )
323
323
324
324
// create client
325
325
consul := ClientAtServer (server1 )
@@ -333,12 +333,12 @@ func TestGetAllServices_RemovingFailingAgentsAndRetrying(t *testing.T) {
333
333
334
334
// given
335
335
// register services in both servers
336
- server1 .AddService ("serviceA" , "passing" , []string {"public" , "marathon" })
337
- server1 .AddService ("serviceB" , "passing" , []string {"marathon" })
338
- server1 .AddService ("serviceC" , "passing" , []string {"zookeeper" })
336
+ server1 .AddService (t , "serviceA" , "passing" , []string {"public" , "marathon" })
337
+ server1 .AddService (t , "serviceB" , "passing" , []string {"marathon" })
338
+ server1 .AddService (t , "serviceC" , "passing" , []string {"zookeeper" })
339
339
340
- server2 .AddService ("serviceA" , "passing" , []string {"private" , "marathon" })
341
- server2 .AddService ("serviceB" , "passing" , []string {"zookeeper" })
340
+ server2 .AddService (t , "serviceA" , "passing" , []string {"private" , "marathon" })
341
+ server2 .AddService (t , "serviceB" , "passing" , []string {"zookeeper" })
342
342
343
343
// when
344
344
services , err := consul .GetAllServices ()
@@ -537,8 +537,8 @@ func TestDeregisterServices(t *testing.T) {
537
537
consul .config .Tag = "marathon"
538
538
539
539
// given
540
- server .AddService ("serviceA" , "passing" , []string {"marathon" })
541
- server .AddService ("serviceB" , "passing" , []string {"marathon" })
540
+ server .AddService (t , "serviceA" , "passing" , []string {"marathon" })
541
+ server .AddService (t , "serviceB" , "passing" , []string {"marathon" })
542
542
services , _ := consul .GetAllServices ()
543
543
assert .Len (t , services , 2 )
544
544
@@ -560,7 +560,7 @@ func TestDeregisterServices_shouldReturnErrorOnFailure(t *testing.T) {
560
560
consul .config .Tag = "marathon"
561
561
562
562
// given
563
- server .AddService ("serviceA" , "passing" , []string {"marathon" })
563
+ server .AddService (t , "serviceA" , "passing" , []string {"marathon" })
564
564
565
565
// when
566
566
servicesA , _ := consul .GetServices ("serviceA" )
@@ -583,8 +583,8 @@ func TestDeregisterServicesByTask(t *testing.T) {
583
583
app := utils .ConsulApp ("serviceA" , 1 )
584
584
task := app .Tasks [0 ]
585
585
586
- server .AddService ("serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
587
- server .AddService ("serviceB" , "passing" , []string {"marathon" , service .MarathonTaskTag (apps .TaskID ("other" ))})
586
+ server .AddService (t , "serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
587
+ server .AddService (t , "serviceB" , "passing" , []string {"marathon" , service .MarathonTaskTag (apps .TaskID ("other" ))})
588
588
services , _ := consul .GetAllServices ()
589
589
assert .Len (t , services , 2 )
590
590
@@ -608,7 +608,7 @@ func TestDeregisterServicesByTask_shouldReturnErrorOnFailure(t *testing.T) {
608
608
app := utils .ConsulApp ("serviceA" , 1 )
609
609
task := app .Tasks [0 ]
610
610
611
- server .AddService ("serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
611
+ server .AddService (t , "serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
612
612
613
613
// when
614
614
server .Stop ()
@@ -630,8 +630,8 @@ func TestDeregisterServicesByTask_shouldReturnErrorOnServiceMatchingTaskNotFound
630
630
app := utils .ConsulApp ("serviceA" , 1 )
631
631
task := app .Tasks [0 ]
632
632
633
- server .AddService ("serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
634
- server .AddService ("serviceB" , "passing" , []string {"marathon" , service .MarathonTaskTag (apps .TaskID ("other" ))})
633
+ server .AddService (t , "serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
634
+ server .AddService (t , "serviceB" , "passing" , []string {"marathon" , service .MarathonTaskTag (apps .TaskID ("other" ))})
635
635
services , _ := consul .GetAllServices ()
636
636
assert .Len (t , services , 2 )
637
637
@@ -656,9 +656,9 @@ func TestDeregisterServicesByTask_shouldDeregisterAllMatchingServicesWhenMultipl
656
656
app := utils .ConsulApp ("serviceA" , 1 )
657
657
task := app .Tasks [0 ]
658
658
659
- server .AddService ("serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
660
- server .AddService ("serviceA-bis" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
661
- server .AddService ("serviceB" , "passing" , []string {"marathon" , service .MarathonTaskTag (apps .TaskID ("other" ))})
659
+ server .AddService (t , "serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
660
+ server .AddService (t , "serviceA-bis" , "passing" , []string {"marathon" , service .MarathonTaskTag (task .ID )})
661
+ server .AddService (t , "serviceB" , "passing" , []string {"marathon" , service .MarathonTaskTag (apps .TaskID ("other" ))})
662
662
services , _ := consul .GetAllServices ()
663
663
assert .Len (t , services , 3 )
664
664
@@ -687,7 +687,7 @@ func TestAddAgentsFromApp(t *testing.T) {
687
687
// given
688
688
app := utils .ConsulApp ("serviceA" , 1 )
689
689
app .Tasks [0 ].Host = server .Config .Bind
690
- server .AddService ("serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (app .Tasks [0 ].ID )})
690
+ server .AddService (t , "serviceA" , "passing" , []string {"marathon" , service .MarathonTaskTag (app .Tasks [0 ].ID )})
691
691
692
692
// when
693
693
consul .AddAgentsFromApps ([]* apps.App {app , utils .NonConsulApp ("nonConsulApp" , 1 )})
0 commit comments