mirror of
https://github.com/ipfs/kubo.git
synced 2025-09-09 19:32:24 +08:00
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package bitswap
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
|
|
context "github.com/ipfs/go-ipfs/Godeps/_workspace/src/golang.org/x/net/context"
|
|
blocks "github.com/ipfs/go-ipfs/blocks"
|
|
bsmsg "github.com/ipfs/go-ipfs/exchange/bitswap/message"
|
|
bsnet "github.com/ipfs/go-ipfs/exchange/bitswap/network"
|
|
peer "github.com/ipfs/go-ipfs/p2p/peer"
|
|
mockrouting "github.com/ipfs/go-ipfs/routing/mock"
|
|
delay "github.com/ipfs/go-ipfs/thirdparty/delay"
|
|
testutil "github.com/ipfs/go-ipfs/util/testutil"
|
|
)
|
|
|
|
func TestSendMessageAsyncButWaitForResponse(t *testing.T) {
|
|
net := VirtualNetwork(mockrouting.NewServer(), delay.Fixed(0))
|
|
responderPeer := testutil.RandIdentityOrFatal(t)
|
|
waiter := net.Adapter(testutil.RandIdentityOrFatal(t))
|
|
responder := net.Adapter(responderPeer)
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
wg.Add(1)
|
|
|
|
expectedStr := "received async"
|
|
|
|
responder.SetDelegate(lambda(func(
|
|
ctx context.Context,
|
|
fromWaiter peer.ID,
|
|
msgFromWaiter bsmsg.BitSwapMessage) error {
|
|
|
|
msgToWaiter := bsmsg.New()
|
|
msgToWaiter.AddBlock(blocks.NewBlock([]byte(expectedStr)))
|
|
waiter.SendMessage(ctx, fromWaiter, msgToWaiter)
|
|
|
|
return nil
|
|
}))
|
|
|
|
waiter.SetDelegate(lambda(func(
|
|
ctx context.Context,
|
|
fromResponder peer.ID,
|
|
msgFromResponder bsmsg.BitSwapMessage) error {
|
|
|
|
// TODO assert that this came from the correct peer and that the message contents are as expected
|
|
ok := false
|
|
for _, b := range msgFromResponder.Blocks() {
|
|
if string(b.Data) == expectedStr {
|
|
wg.Done()
|
|
ok = true
|
|
}
|
|
}
|
|
|
|
if !ok {
|
|
t.Fatal("Message not received from the responder")
|
|
|
|
}
|
|
return nil
|
|
}))
|
|
|
|
messageSentAsync := bsmsg.New()
|
|
messageSentAsync.AddBlock(blocks.NewBlock([]byte("data")))
|
|
errSending := waiter.SendMessage(
|
|
context.Background(), responderPeer.ID(), messageSentAsync)
|
|
if errSending != nil {
|
|
t.Fatal(errSending)
|
|
}
|
|
|
|
wg.Wait() // until waiter delegate function is executed
|
|
}
|
|
|
|
type receiverFunc func(ctx context.Context, p peer.ID,
|
|
incoming bsmsg.BitSwapMessage) error
|
|
|
|
// lambda returns a Receiver instance given a receiver function
|
|
func lambda(f receiverFunc) bsnet.Receiver {
|
|
return &lambdaImpl{
|
|
f: f,
|
|
}
|
|
}
|
|
|
|
type lambdaImpl struct {
|
|
f func(ctx context.Context, p peer.ID, incoming bsmsg.BitSwapMessage) error
|
|
}
|
|
|
|
func (lam *lambdaImpl) ReceiveMessage(ctx context.Context,
|
|
p peer.ID, incoming bsmsg.BitSwapMessage) error {
|
|
return lam.f(ctx, p, incoming)
|
|
}
|
|
|
|
func (lam *lambdaImpl) ReceiveError(err error) {
|
|
// TODO log error
|
|
}
|
|
|
|
func (lam *lambdaImpl) PeerConnected(p peer.ID) {
|
|
// TODO
|
|
}
|
|
func (lam *lambdaImpl) PeerDisconnected(peer.ID) {
|
|
// TODO
|
|
}
|