forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-19 18:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

write tests for RequestFromPeers

Browse source
This commit is contained in:
Mark Vujevits 2018-10-19 11:18:07 +02:00
parent b7aa01cfb8
commit 3fae03a6b7
1 changed files with 17 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
swarm/network/stream/delivery_test.go
View file

@ -36,6 +36,7 @@ import (
p2ptest "github.com/ethereum/go-ethereum/p2p/testing" p2ptest "github.com/ethereum/go-ethereum/p2p/testing"
"github.com/ethereum/go-ethereum/swarm/log" "github.com/ethereum/go-ethereum/swarm/log"
"github.com/ethereum/go-ethereum/swarm/network" "github.com/ethereum/go-ethereum/swarm/network"
pq "github.com/ethereum/go-ethereum/swarm/network/priorityqueue"
"github.com/ethereum/go-ethereum/swarm/network/simulation" "github.com/ethereum/go-ethereum/swarm/network/simulation"
"github.com/ethereum/go-ethereum/swarm/state" "github.com/ethereum/go-ethereum/swarm/state"
"github.com/ethereum/go-ethereum/swarm/storage" "github.com/ethereum/go-ethereum/swarm/storage"
@ -241,14 +242,19 @@ func TestRequestFromPeers(t *testing.T) {
}, to) }, to)
to.On(peer) to.On(peer)
r := NewRegistry(addr.ID(), delivery, nil, nil, nil) r := NewRegistry(addr.ID(), delivery, nil, nil, nil)
r.setPeer(NewPeer(protocolsPeer, r))
ctx := context.Background() sp := &Peer{
Peer: protocolsPeer,
pq: pq.New(int(PriorityQueue), PriorityQueueCap),
streamer: r,
}
r.setPeer(sp)
req := network.NewRequest( req := network.NewRequest(
storage.Address(hash0[:]), storage.Address(hash0[:]),
true, true,
&sync.Map{}, &sync.Map{},
) )
ctx := context.Background()
id, _, err := delivery.RequestFromPeers(ctx, req) id, _, err := delivery.RequestFromPeers(ctx, req)
if err != nil { if err != nil {
@ -273,14 +279,20 @@ func TestRequestFromPeersWithLightNode(t *testing.T) {
}, to) }, to)
to.On(peer) to.On(peer)
r := NewRegistry(addr.ID(), delivery, nil, nil, nil) r := NewRegistry(addr.ID(), delivery, nil, nil, nil)
r.setPeer(NewPeer(protocolsPeer, r)) sp := &Peer{
ctx := context.Background() Peer: protocolsPeer,
pq: pq.New(int(PriorityQueue), PriorityQueueCap),
streamer: r,
}
r.setPeer(sp)
req := network.NewRequest( req := network.NewRequest(
storage.Address(hash0[:]), storage.Address(hash0[:]),
true, true,
&sync.Map{}, &sync.Map{},
) )
ctx := context.Background()
_, _, err := delivery.RequestFromPeers(ctx, req) _, _, err := delivery.RequestFromPeers(ctx, req)
expectedError := "no peer found" expectedError := "no peer found"