fix(routing:dht) add ctx args
Brian Tiger Chow committed
Sep 22, 2014 at 03:41 UTC
197046c9d2fb0e0cc7e6cb582cd67a50c0fa281f
1 file changed
+19
-18
routing/dht/dht_test.go
+19
-18
@@ -109,13 +109,14 @@ func TestPing(t *testing.T) {
109
defer dhtA.network.Close()
110
defer dhtB.network.Close()
111
112
- _, err = dhtA.Connect(peerB)
112
+ _, err = dhtA.Connect(context.Background(), peerB)
113
if err != nil {
114
t.Fatal(err)
115
}
116
117
//Test that we can ping the node
118
- err = dhtA.Ping(peerB, time.Second*2)
118
+ ctx, _ := context.WithTimeout(context.Background(), 2*time.Millisecond)
119
+ err = dhtA.Ping(ctx, peerB)
120
if err != nil {
121
t.Fatal(err)
122
}
@@ -145,7 +146,7 @@ func TestValueGetSet(t *testing.T) {
146
defer dhtA.network.Close()
147
defer dhtB.network.Close()
148
148
- _, err = dhtA.Connect(peerB)
149
+ _, err = dhtA.Connect(context.Background(), peerB)
150
if err != nil {
151
t.Fatal(err)
152
}
@@ -178,17 +179,17 @@ func TestProvides(t *testing.T) {
179
}
180
}()
181
181
- _, err := dhts[0].Connect(peers[1])
182
+ _, err := dhts[0].Connect(context.Background(), peers[1])
183
if err != nil {
184
t.Fatal(err)
185
}
186
186
- _, err = dhts[1].Connect(peers[2])
187
+ _, err = dhts[1].Connect(context.Background(), peers[2])
188
if err != nil {
189
t.Fatal(err)
190
}
191
191
- _, err = dhts[1].Connect(peers[3])
192
+ _, err = dhts[1].Connect(context.Background(), peers[3])
193
if err != nil {
194
t.Fatal(err)
195
}
@@ -203,7 +204,7 @@ func TestProvides(t *testing.T) {
204
t.Fatal(err)
205
}
206
206
- err = dhts[3].Provide(u.Key("hello"))
207
+ err = dhts[3].Provide(context.Background(), u.Key("hello"))
208
if err != nil {
209
t.Fatal(err)
210
}
@@ -234,17 +235,17 @@ func TestProvidesAsync(t *testing.T) {
235
}
236
}()
237
237
- _, err := dhts[0].Connect(peers[1])
238
+ _, err := dhts[0].Connect(context.Background(), peers[1])
239
if err != nil {
240
t.Fatal(err)
241
}
242
242
- _, err = dhts[1].Connect(peers[2])
243
+ _, err = dhts[1].Connect(context.Background(), peers[2])
244
if err != nil {
245
t.Fatal(err)
246
}
247
247
- _, err = dhts[1].Connect(peers[3])
248
+ _, err = dhts[1].Connect(context.Background(), peers[3])
249
if err != nil {
250
t.Fatal(err)
251
}
@@ -259,7 +260,7 @@ func TestProvidesAsync(t *testing.T) {
260
t.Fatal(err)
261
}
262
262
- err = dhts[3].Provide(u.Key("hello"))
263
+ err = dhts[3].Provide(context.Background(), u.Key("hello"))
264
if err != nil {
265
t.Fatal(err)
266
}
@@ -290,17 +291,17 @@ func TestLayeredGet(t *testing.T) {
291
}
292
}()
293
293
- _, err := dhts[0].Connect(peers[1])
294
+ _, err := dhts[0].Connect(context.Background(), peers[1])
295
if err != nil {
296
t.Fatalf("Failed to connect: %s", err)
297
}
298
298
- _, err = dhts[1].Connect(peers[2])
299
+ _, err = dhts[1].Connect(context.Background(), peers[2])
300
if err != nil {
301
t.Fatal(err)
302
}
303
303
- _, err = dhts[1].Connect(peers[3])
304
+ _, err = dhts[1].Connect(context.Background(), peers[3])
305
if err != nil {
306
t.Fatal(err)
307
}
@@ -310,7 +311,7 @@ func TestLayeredGet(t *testing.T) {
311
t.Fatal(err)
312
}
313
313
- err = dhts[3].Provide(u.Key("hello"))
314
+ err = dhts[3].Provide(context.Background(), u.Key("hello"))
315
if err != nil {
316
t.Fatal(err)
317
}
@@ -342,17 +343,17 @@ func TestFindPeer(t *testing.T) {
343
}
344
}()
345
345
- _, err := dhts[0].Connect(peers[1])
346
+ _, err := dhts[0].Connect(context.Background(), peers[1])
347
if err != nil {
348
t.Fatal(err)
349
}
350
350
- _, err = dhts[1].Connect(peers[2])
351
+ _, err = dhts[1].Connect(context.Background(), peers[2])
352
if err != nil {
353
t.Fatal(err)
354
}
355
355
- _, err = dhts[1].Connect(peers[3])
356
+ _, err = dhts[1].Connect(context.Background(), peers[3])
357
if err != nil {
358
t.Fatal(err)
359
}