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

pass peer into testnet adapter method

Browse Source
This commit is contained in:
Brian Tiger Chow
2014-12-23 18:18:32 -05:00
parent e017d6edc6
commit 6cdb4c23b4
3 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
exchange/bitswap/testnet/network.go
View File

@ -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
}

18
exchange/bitswap/testnet/network_test.go
View File

@ -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)
}

2
exchange/bitswap/testutils.go
View File

@ -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