opensource/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2025-09-10 05:52:20 +08:00
Code Issues Projects Releases Wiki Activity

improve cleaning up in dht tests.

Browse Source
This commit is contained in:
Jeromy
2014-09-15 20:45:36 +00:00
parent ab9baf9146
commit b5a1b1c658
1 changed files with 19 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
routing/dht/dht_test.go
View File

@ -107,6 +107,8 @@ func TestPing(t *testing.T) {
dhtA.Start()
dhtB.Start()
defer dhtA.Halt()
defer dhtB.Halt()
_, err = dhtA.Connect(addrB)
if err != nil {
@ -118,9 +120,6 @@ func TestPing(t *testing.T) {
if err != nil {
t.Fatal(err)
}
dhtA.Halt()
dhtB.Halt()
}
func TestValueGetSet(t *testing.T) {
@ -153,6 +152,8 @@ func TestValueGetSet(t *testing.T) {
dhtA.Start()
dhtB.Start()
defer dhtA.Halt()
defer dhtB.Halt()
errsa := dhtA.network.GetErrChan()
errsb := dhtB.network.GetErrChan()
@ -187,6 +188,11 @@ func TestProvides(t *testing.T) {
u.Debug = false
addrs, _, dhts := setupDHTS(4, t)
defer func() {
for i := 0; i < 4; i++ {
dhts[i].Halt()
}
}()
_, err := dhts[0].Connect(addrs[1])
if err != nil {
@ -228,15 +234,16 @@ func TestProvides(t *testing.T) {
if len(provs) != 1 {
t.Fatal("Didnt get back providers")
}
for i := 0; i < 4; i++ {
dhts[i].Halt()
}
}
func TestLayeredGet(t *testing.T) {
u.Debug = false
addrs, _, dhts := setupDHTS(4, t)
defer func() {
for i := 0; i < 4; i++ {
dhts[i].Halt()
}
}()
_, err := dhts[0].Connect(addrs[1])
if err != nil {
@ -274,15 +281,17 @@ func TestLayeredGet(t *testing.T) {
t.Fatal("Got incorrect value.")
}
for i := 0; i < 4; i++ {
dhts[i].Halt()
}
}
func TestFindPeer(t *testing.T) {
u.Debug = false
addrs, peers, dhts := setupDHTS(4, t)
go func() {
for i := 0; i < 4; i++ {
dhts[i].Halt()
}
}()
_, err := dhts[0].Connect(addrs[1])
if err != nil {
@ -311,8 +320,4 @@ func TestFindPeer(t *testing.T) {
if !p.ID.Equal(peers[2].ID) {
t.Fatal("Didnt find expected peer.")
}
for i := 0; i < 4; i++ {
dhts[i].Halt()
}
}