mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-23 23:36:56 +00:00
d2379608ad
* Implement upgrade mechanism for p2p * Remove dependencies from flowcontext to v3 * Add p2p v4 * Add Ready flow * Remove copy paste code of v3 * Register SendAddresses flow at the top level * Add option to set protocol version from CLI and add TestAddressExchangeV3V4 * Send ready message on minimal net adapter * Rename defaultMaxProtocolVersion->maxAcceptableProtocolVersion
52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
package testing
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/app/protocol/flows/v3/addressexchange"
|
|
peerpkg "github.com/kaspanet/kaspad/app/protocol/peer"
|
|
"github.com/kaspanet/kaspad/domain/consensus"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/testutils"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/addressmanager"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
)
|
|
|
|
type fakeReceiveAddressesContext struct{}
|
|
|
|
func (f fakeReceiveAddressesContext) AddressManager() *addressmanager.AddressManager {
|
|
return nil
|
|
}
|
|
|
|
func TestReceiveAddressesErrors(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
incomingRoute := router.NewRoute("incoming")
|
|
outgoingRoute := router.NewRoute("outgoing")
|
|
peer := peerpkg.New(nil)
|
|
errChan := make(chan error)
|
|
go func() {
|
|
errChan <- addressexchange.ReceiveAddresses(fakeReceiveAddressesContext{}, incomingRoute, outgoingRoute, peer)
|
|
}()
|
|
|
|
_, err := outgoingRoute.DequeueWithTimeout(time.Second)
|
|
if err != nil {
|
|
t.Fatalf("DequeueWithTimeout: %+v", err)
|
|
}
|
|
|
|
// Sending addressmanager.GetAddressesMax+1 addresses should trigger a ban
|
|
err = incomingRoute.Enqueue(appmessage.NewMsgAddresses(make([]*appmessage.NetAddress,
|
|
addressmanager.GetAddressesMax+1)))
|
|
if err != nil {
|
|
t.Fatalf("Enqueue: %+v", err)
|
|
}
|
|
|
|
select {
|
|
case err := <-errChan:
|
|
checkFlowError(t, err, true, true, "address count exceeded")
|
|
case <-time.After(time.Second):
|
|
t.Fatalf("timed out after %s", time.Second)
|
|
}
|
|
})
|
|
}
|