dht test fix (net)
Juan Batiz-Benet committed
Oct 22, 2014 at 05:31 UTC
a8adf2ff2c997c11cc18fe8a0190f4599103d931
1 file changed
+10
-10
routing/dht/dht_test.go
+10
-10
@@ -98,8 +98,8 @@ func TestPing(t *testing.T) {
98
99
defer dhtA.Halt()
100
defer dhtB.Halt()
101
- defer dhtA.network.Close()
102
- defer dhtB.network.Close()
101
+ defer dhtA.dialer.(inet.Network).Close()
102
+ defer dhtB.dialer.(inet.Network).Close()
103
104
_, err = dhtA.Connect(ctx, peerB)
105
if err != nil {
@@ -142,8 +142,8 @@ func TestValueGetSet(t *testing.T) {
142
143
defer dhtA.Halt()
144
defer dhtB.Halt()
145
- defer dhtA.network.Close()
146
- defer dhtB.network.Close()
145
+ defer dhtA.dialer.(inet.Network).Close()
146
+ defer dhtB.dialer.(inet.Network).Close()
147
148
_, err = dhtA.Connect(ctx, peerB)
149
if err != nil {
@@ -184,7 +184,7 @@ func TestProvides(t *testing.T) {
184
defer func() {
185
for i := 0; i < 4; i++ {
186
dhts[i].Halt()
187
- defer dhts[i].network.Close()
187
+ defer dhts[i].dialer.(inet.Network).Close()
188
}
189
}()
190
@@ -244,7 +244,7 @@ func TestProvidesAsync(t *testing.T) {
244
defer func() {
245
for i := 0; i < 4; i++ {
246
dhts[i].Halt()
247
- defer dhts[i].network.Close()
247
+ defer dhts[i].dialer.(inet.Network).Close()
248
}
249
}()
250
@@ -301,7 +301,7 @@ func TestLayeredGet(t *testing.T) {
301
defer func() {
302
for i := 0; i < 4; i++ {
303
dhts[i].Halt()
304
- defer dhts[i].network.Close()
304
+ defer dhts[i].dialer.(inet.Network).Close()
305
}
306
}()
307
@@ -354,7 +354,7 @@ func TestFindPeer(t *testing.T) {
354
defer func() {
355
for i := 0; i < 4; i++ {
356
dhts[i].Halt()
357
- dhts[i].network.Close()
357
+ dhts[i].dialer.(inet.Network).Close()
358
}
359
}()
360
@@ -443,8 +443,8 @@ func TestConnectCollision(t *testing.T) {
443
444
dhtA.Halt()
445
dhtB.Halt()
446
- dhtA.network.Close()
447
- dhtB.network.Close()
446
+ dhtA.dialer.(inet.Network).Close()
447
+ dhtB.dialer.(inet.Network).Close()
448
449
<-time.After(200 * time.Millisecond)
450
}