From 6cdb4c23b42735bfa976c2a546a9e0bbcd4c3167 Mon Sep 17 00:00:00 2001
From: Brian Tiger Chow <brian.holderchow@gmail.com>
Date: Tue, 23 Dec 2014 18:18:32 -0500
Subject: [PATCH] pass peer into testnet adapter method

---
 exchange/bitswap/testnet/network.go      | 11 ++++++-----
 exchange/bitswap/testnet/network_test.go | 18 +++++++++---------
 exchange/bitswap/testutils.go            |  2 +-
 3 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/exchange/bitswap/testnet/network.go b/exchange/bitswap/testnet/network.go
index f45202630..26566bf7e 100644
--- a/exchange/bitswap/testnet/network.go
+++ b/exchange/bitswap/testnet/network.go
@@ -12,10 +12,11 @@ import (
 	mockrouting "github.com/jbenet/go-ipfs/routing/mock"
 	util "github.com/jbenet/go-ipfs/util"
 	delay "github.com/jbenet/go-ipfs/util/delay"
+	testutil "github.com/jbenet/go-ipfs/util/testutil"
 )
 
 type Network interface {
-	Adapter(peer.ID) bsnet.BitSwapNetwork
+	Adapter(testutil.Peer) bsnet.BitSwapNetwork
 
 	HasPeer(peer.ID) bool
 }
@@ -36,13 +37,13 @@ type network struct {
 	delay         delay.D
 }
 
-func (n *network) Adapter(p peer.ID) bsnet.BitSwapNetwork {
+func (n *network) Adapter(p testutil.Peer) bsnet.BitSwapNetwork {
 	client := &networkClient{
-		local:   p,
+		local:   p.ID(),
 		network: n,
-		routing: n.routingserver.Client(peer.PeerInfo{ID: p}),
+		routing: n.routingserver.Client(peer.PeerInfo{ID: p.ID()}),
 	}
-	n.clients[p] = client
+	n.clients[p.ID()] = client
 	return client
 }
 
diff --git a/exchange/bitswap/testnet/network_test.go b/exchange/bitswap/testnet/network_test.go
index 1418497f0..08f4ff500 100644
--- a/exchange/bitswap/testnet/network_test.go
+++ b/exchange/bitswap/testnet/network_test.go
@@ -5,24 +5,24 @@ import (
 	"testing"
 
 	context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context"
-
 	blocks "github.com/jbenet/go-ipfs/blocks"
 	bsmsg "github.com/jbenet/go-ipfs/exchange/bitswap/message"
 	bsnet "github.com/jbenet/go-ipfs/exchange/bitswap/network"
 	peer "github.com/jbenet/go-ipfs/peer"
 	mockrouting "github.com/jbenet/go-ipfs/routing/mock"
 	delay "github.com/jbenet/go-ipfs/util/delay"
+	testutil "github.com/jbenet/go-ipfs/util/testutil"
 )
 
 func TestSendRequestToCooperativePeer(t *testing.T) {
 	net := VirtualNetwork(mockrouting.NewServer(), delay.Fixed(0))
 
-	idOfRecipient := peer.ID("recipient")
+	recipientPeer := testutil.RandPeerOrFatal(t)
 
 	t.Log("Get two network adapters")
 
-	initiator := net.Adapter(peer.ID("initiator"))
-	recipient := net.Adapter(idOfRecipient)
+	initiator := net.Adapter(testutil.RandPeerOrFatal(t))
+	recipient := net.Adapter(recipientPeer)
 
 	expectedStr := "response from recipient"
 	recipient.SetDelegate(lambda(func(
@@ -46,7 +46,7 @@ func TestSendRequestToCooperativePeer(t *testing.T) {
 	message := bsmsg.New()
 	message.AddBlock(blocks.NewBlock([]byte("data")))
 	response, err := initiator.SendRequest(
-		context.Background(), idOfRecipient, message)
+		context.Background(), recipientPeer.ID(), message)
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -67,9 +67,9 @@ func TestSendRequestToCooperativePeer(t *testing.T) {
 
 func TestSendMessageAsyncButWaitForResponse(t *testing.T) {
 	net := VirtualNetwork(mockrouting.NewServer(), delay.Fixed(0))
-	idOfResponder := peer.ID("responder")
-	waiter := net.Adapter(peer.ID("waiter"))
-	responder := net.Adapter(idOfResponder)
+	responderPeer := testutil.RandPeerOrFatal(t)
+	waiter := net.Adapter(testutil.RandPeerOrFatal(t))
+	responder := net.Adapter(responderPeer)
 
 	var wg sync.WaitGroup
 
@@ -114,7 +114,7 @@ func TestSendMessageAsyncButWaitForResponse(t *testing.T) {
 	messageSentAsync := bsmsg.New()
 	messageSentAsync.AddBlock(blocks.NewBlock([]byte("data")))
 	errSending := waiter.SendMessage(
-		context.Background(), idOfResponder, messageSentAsync)
+		context.Background(), responderPeer.ID(), messageSentAsync)
 	if errSending != nil {
 		t.Fatal(errSending)
 	}
diff --git a/exchange/bitswap/testutils.go b/exchange/bitswap/testutils.go
index f636eddd6..728c2ba3b 100644
--- a/exchange/bitswap/testutils.go
+++ b/exchange/bitswap/testutils.go
@@ -80,7 +80,7 @@ func (i *Instance) SetBlockstoreLatency(t time.Duration) time.Duration {
 // just a much better idea.
 func session(ctx context.Context, net tn.Network, p testutil.Peer) Instance {
 
-	adapter := net.Adapter(p.ID())
+	adapter := net.Adapter(p)
 
 	bsdelay := delay.Fixed(0)
 	const kWriteCacheElems = 100