From 414ff34194e275d5b7c7e59cf2eeef2a19c414ee Mon Sep 17 00:00:00 2001
From: Jeromy <jeromyj@gmail.com>
Date: Tue, 23 Sep 2014 11:45:02 -0700
Subject: [PATCH] change back to using Client method

---
 exchange/bitswap/bitswap_test.go         |  3 +--
 exchange/bitswap/testnet/routing.go      |  1 -
 exchange/bitswap/testnet/routing_test.go | 16 +++++++---------
 routing/mock/routing.go                  | 11 ++++++++++-
 4 files changed, 18 insertions(+), 13 deletions(-)
 delete mode 100644 exchange/bitswap/testnet/routing.go

diff --git a/exchange/bitswap/bitswap_test.go b/exchange/bitswap/bitswap_test.go
index fd9808160..a9fc11f82 100644
--- a/exchange/bitswap/bitswap_test.go
+++ b/exchange/bitswap/bitswap_test.go
@@ -281,8 +281,7 @@ func session(net tn.Network, rs mock.RoutingServer, id peer.ID) instance {
 	p := &peer.Peer{ID: id}
 
 	adapter := net.Adapter(p)
-	htc := mock.NewMockRouter(p, nil)
-	htc.SetRoutingServer(rs)
+	htc := rs.Client(p)
 
 	blockstore := bstore.NewBlockstore(ds.NewMapDatastore())
 	const alwaysSendToPeer = true
diff --git a/exchange/bitswap/testnet/routing.go b/exchange/bitswap/testnet/routing.go
deleted file mode 100644
index 67a03afb7..000000000
--- a/exchange/bitswap/testnet/routing.go
+++ /dev/null
@@ -1 +0,0 @@
-package bitswap
diff --git a/exchange/bitswap/testnet/routing_test.go b/exchange/bitswap/testnet/routing_test.go
index 30a573f6f..b3cbd385a 100644
--- a/exchange/bitswap/testnet/routing_test.go
+++ b/exchange/bitswap/testnet/routing_test.go
@@ -43,12 +43,10 @@ func TestSetAndGet(t *testing.T) {
 }
 
 func TestClientFindProviders(t *testing.T) {
-	peer := &peer.Peer{
-		ID: []byte("42"),
-	}
+	peer := &peer.Peer{ID: []byte("42")}
 	rs := mock.VirtualRoutingServer()
-	client := mock.NewMockRouter(peer, nil)
-	client.SetRoutingServer(rs)
+	client := rs.Client(peer)
+
 	k := u.Key("hello")
 	err := client.Provide(context.Background(), k)
 	if err != nil {
@@ -99,8 +97,9 @@ func TestClientOverMax(t *testing.T) {
 	}
 
 	max := 10
-	client := mock.NewMockRouter(&peer.Peer{ID: []byte("TODO")}, nil)
-	client.SetRoutingServer(rs)
+	peer := &peer.Peer{ID: []byte("TODO")}
+	client := rs.Client(peer)
+
 	providersFromClient := client.FindProvidersAsync(context.Background(), k, max)
 	i := 0
 	for _ = range providersFromClient {
@@ -132,8 +131,7 @@ func TestCanceledContext(t *testing.T) {
 	}()
 
 	local := &peer.Peer{ID: []byte("peer id doesn't matter")}
-	client := mock.NewMockRouter(local, nil)
-	client.SetRoutingServer(rs)
+	client := rs.Client(local)
 
 	t.Log("warning: max is finite so this test is non-deterministic")
 	t.Log("context cancellation could simply take lower priority")
diff --git a/routing/mock/routing.go b/routing/mock/routing.go
index c239c634e..43aecf269 100644
--- a/routing/mock/routing.go
+++ b/routing/mock/routing.go
@@ -20,7 +20,7 @@ type MockRouter struct {
 	peer      *peer.Peer
 }
 
-func NewMockRouter(local *peer.Peer, dstore ds.Datastore) *MockRouter {
+func NewMockRouter(local *peer.Peer, dstore ds.Datastore) routing.IpfsRouting {
 	return &MockRouter{
 		datastore: dstore,
 		peer:      local,
@@ -84,6 +84,8 @@ type RoutingServer interface {
 	Announce(*peer.Peer, u.Key) error
 
 	Providers(u.Key) []*peer.Peer
+
+	Client(p *peer.Peer) routing.IpfsRouting
 }
 
 func VirtualRoutingServer() RoutingServer {
@@ -128,3 +130,10 @@ func (rs *hashTable) Providers(k u.Key) []*peer.Peer {
 
 	return ret
 }
+
+func (rs *hashTable) Client(p *peer.Peer) routing.IpfsRouting {
+	return &MockRouter{
+		peer:      p,
+		hashTable: rs,
+	}
+}