Skip to content

Commit

Permalink
bitswap/server/internal/decision: add more non flaky tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Jorropo committed Feb 17, 2023
1 parent 62cbac4 commit baa748b
Showing 1 changed file with 133 additions and 0 deletions.
133 changes: 133 additions & 0 deletions bitswap/server/internal/decision/engine_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,10 @@ package decision
import (
"bytes"
"context"
"encoding/binary"
"errors"
"fmt"
"math/rand"
"strings"
"sync"
"testing"
Expand All @@ -23,6 +25,7 @@ import (
process "github.com/jbenet/goprocess"
peer "github.com/libp2p/go-libp2p/core/peer"
libp2ptest "github.com/libp2p/go-libp2p/core/test"
mh "github.com/multiformats/go-multihash"
)

type peerTag struct {
Expand Down Expand Up @@ -1051,6 +1054,12 @@ func TestWantlistForPeer(t *testing.T) {
if len(entries) != 4 {
t.Fatal("expected wantlist to contain all wants from parter")
}

e.PeerDisconnected(partner)
entries = e.WantlistForPeer(partner)
if len(entries) != 0 {
t.Fatal("expected wantlist to be empty after disconnect")
}
}

func TestTaskComparator(t *testing.T) {
Expand Down Expand Up @@ -1629,3 +1638,127 @@ func stringsComplement(set, subset []string) []string {
}
return complement
}

func TestWantlistDoesNotGrowPastLimit(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()

const limit = 32
warsaw := newTestEngine(ctx, "warsaw", WithMaxQueuedWantlistEntriesPerPeer(limit))
riga := newTestEngine(ctx, "riga")

// Send in two messages to test reslicing.
for i := 2; i != 0; i-- {
m := message.New(false)
for j := limit * 3 / 4; j != 0; j-- {
m.AddEntry(blocks.NewBlock([]byte(fmt.Sprint(i, j))).Cid(), 0, pb.Message_Wantlist_Block, true)
}
warsaw.Engine.MessageReceived(ctx, riga.Peer, m)
}

if warsaw.Peer == riga.Peer {
t.Fatal("Sanity Check: Peers have same Key!")
}

wl := warsaw.Engine.WantlistForPeer(riga.Peer)
if len(wl) != limit {
t.Fatal("wantlist does not match limit", len(wl))
}
}

func TestWantlistGrowsToLimit(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()

const limit = 32
warsaw := newTestEngine(ctx, "warsaw", WithMaxQueuedWantlistEntriesPerPeer(limit))
riga := newTestEngine(ctx, "riga")

// Send in two messages to test reslicing.
m := message.New(false)
for j := limit; j != 0; j-- {
m.AddEntry(blocks.NewBlock([]byte(fmt.Sprint(j))).Cid(), 0, pb.Message_Wantlist_Block, true)
}
warsaw.Engine.MessageReceived(ctx, riga.Peer, m)

if warsaw.Peer == riga.Peer {
t.Fatal("Sanity Check: Peers have same Key!")
}

wl := warsaw.Engine.WantlistForPeer(riga.Peer)
if len(wl) != limit {
t.Fatal("wantlist does not match limit", len(wl))
}
}

func TestIgnoresCidsAboveLimit(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()

const cidLimit = 64
warsaw := newTestEngine(ctx, "warsaw", WithMaxCidSize(cidLimit))
riga := newTestEngine(ctx, "riga")

// Send in two messages to test reslicing.
m := message.New(true)

m.AddEntry(blocks.NewBlock([]byte("Hæ")).Cid(), 0, pb.Message_Wantlist_Block, true)

var hash mh.Multihash
hash = binary.AppendUvarint(hash, mh.BLAKE3)
hash = binary.AppendUvarint(hash, cidLimit)
startOfDigest := len(hash)
hash = append(hash, make(mh.Multihash, cidLimit)...)
rand.Read(hash[startOfDigest:])
m.AddEntry(cid.NewCidV1(cid.Raw, hash), 0, pb.Message_Wantlist_Block, true)

warsaw.Engine.MessageReceived(ctx, riga.Peer, m)

if warsaw.Peer == riga.Peer {
t.Fatal("Sanity Check: Peers have same Key!")
}

wl := warsaw.Engine.WantlistForPeer(riga.Peer)
if len(wl) != 1 {
t.Fatal("wantlist add a CID too big")
}
}

func TestKillConnectionForInlineCid(t *testing.T) {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()

warsaw := newTestEngine(ctx, "warsaw")
riga := newTestEngine(ctx, "riga")

if warsaw.Peer == riga.Peer {
t.Fatal("Sanity Check: Peers have same Key!")
}

// Send in two messages to test reslicing.
m := message.New(true)

m.AddEntry(blocks.NewBlock([]byte("Hæ")).Cid(), 0, pb.Message_Wantlist_Block, true)

var hash mh.Multihash
hash = binary.AppendUvarint(hash, mh.IDENTITY)
const digestSize = 32
hash = binary.AppendUvarint(hash, digestSize)
startOfDigest := len(hash)
hash = append(hash, make(mh.Multihash, digestSize)...)
rand.Read(hash[startOfDigest:])
m.AddEntry(cid.NewCidV1(cid.Raw, hash), 0, pb.Message_Wantlist_Block, true)

if !warsaw.Engine.MessageReceived(ctx, riga.Peer, m) {
t.Fatal("connection was not killed when receiving inline in cancel")
}

m.Reset(true)

m.AddEntry(blocks.NewBlock([]byte("Hæ")).Cid(), 0, pb.Message_Wantlist_Block, true)
m.Cancel(cid.NewCidV1(cid.Raw, hash))

if !warsaw.Engine.MessageReceived(ctx, riga.Peer, m) {
t.Fatal("connection was not killed when receiving inline in cancel")
}
}

0 comments on commit baa748b

Please sign in to comment.