make a few tests perform operations in two directions instead of one

ipfs · Sep 23, 2014 · 7228786 · 7228786
1 parent 7bc4fbf
commit 7228786
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/routing/dht/dht_test.go b/routing/dht/dht_test.go
@@ -115,11 +115,17 @@ func TestPing(t *testing.T) {
 	}
 
 	//Test that we can ping the node
-	ctx, _ := context.WithTimeout(context.Background(), 2*time.Millisecond)
+	ctx, _ := context.WithTimeout(context.Background(), 5*time.Millisecond)
 	err = dhtA.Ping(ctx, peerB)
 	if err != nil {
 		t.Fatal(err)
 	}
+
+	ctx, _ = context.WithTimeout(context.Background(), 5*time.Millisecond)
+	err = dhtB.Ping(ctx, peerA)
+	if err != nil {
+		t.Fatal(err)
+	}
 }
 
 func TestValueGetSet(t *testing.T) {
@@ -164,6 +170,15 @@ func TestValueGetSet(t *testing.T) {
 		t.Fatalf("Expected 'world' got '%s'", string(val))
 	}
 
+	ctxT, _ = context.WithTimeout(context.Background(), time.Second*2)
+	val, err = dhtB.GetValue(ctxT, "hello")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if string(val) != "world" {
+		t.Fatalf("Expected 'world' got '%s'", string(val))
+	}
 }
 
 func TestProvides(t *testing.T) {