mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-05-23 15:26:42 +00:00
ab89efe3dc
* [NOD-1344] relaytransactions: simple unit tests * [NOD-1344] Add mid-complexity unit tests for relaytransactions * Improve TestHandleRelayedTransactionssub tests * Improve TestHandleRequestedTransactions sub tests * [NOD-1344] Fix Simple call test * [NOD-1344] Fix tests after redesign * Divide transactionrelay_test.go to 2 separated tests and updates the tests. * Changes due to review:change the test files name and the test function names, adds new comments and fix typo. * Delete an unnecessary comparison to True in the if statement condition. * Update the branch to v0.11.0-dev. Co-authored-by: karim1king <karimkaspersky@yahoo.com> Co-authored-by: tal <tal@daglabs.com> Co-authored-by: Svarog <feanorr@gmail.com>
89 lines
3.6 KiB
Go
89 lines
3.6 KiB
Go
package transactionrelay_test
|
|
|
|
import (
|
|
"github.com/kaspanet/kaspad/app/appmessage"
|
|
"github.com/kaspanet/kaspad/app/protocol/flows/transactionrelay"
|
|
"github.com/kaspanet/kaspad/domain"
|
|
"github.com/kaspanet/kaspad/domain/consensus"
|
|
"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
|
|
"github.com/kaspanet/kaspad/domain/consensus/utils/testutils"
|
|
"github.com/kaspanet/kaspad/infrastructure/config"
|
|
"github.com/kaspanet/kaspad/infrastructure/logger"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter"
|
|
"github.com/kaspanet/kaspad/infrastructure/network/netadapter/router"
|
|
"github.com/kaspanet/kaspad/util/panics"
|
|
"github.com/pkg/errors"
|
|
"testing"
|
|
)
|
|
|
|
// TestHandleRequestedTransactionsNotFound tests the flow of HandleRequestedTransactions
|
|
// when the requested transactions don't found in the mempool.
|
|
func TestHandleRequestedTransactionsNotFound(t *testing.T) {
|
|
testutils.ForAllNets(t, true, func(t *testing.T, consensusConfig *consensus.Config) {
|
|
var log = logger.RegisterSubSystem("PROT")
|
|
var spawn = panics.GoroutineWrapperFunc(log)
|
|
factory := consensus.NewFactory()
|
|
tc, teardown, err := factory.NewTestConsensus(consensusConfig, "TestHandleRequestedTransactionsNotFound")
|
|
if err != nil {
|
|
t.Fatalf("Error setting up test Consensus: %+v", err)
|
|
}
|
|
defer teardown(false)
|
|
|
|
sharedRequestedTransactions := transactionrelay.NewSharedRequestedTransactions()
|
|
adapter, err := netadapter.NewNetAdapter(config.DefaultConfig())
|
|
if err != nil {
|
|
t.Fatalf("Failed to create a NetAdapter: %v", err)
|
|
}
|
|
domainInstance, err := domain.New(consensusConfig, tc.Database())
|
|
if err != nil {
|
|
t.Fatalf("Failed to set up a domain Instance: %v", err)
|
|
}
|
|
context := &mocTransactionsRelayContext{
|
|
netAdapter: adapter,
|
|
domain: domainInstance,
|
|
sharedRequestedTransactions: sharedRequestedTransactions,
|
|
}
|
|
incomingRoute := router.NewRoute()
|
|
outgoingRoute := router.NewRoute()
|
|
defer outgoingRoute.Close()
|
|
|
|
txID1 := externalapi.NewDomainTransactionIDFromByteArray(&[externalapi.DomainHashSize]byte{
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01})
|
|
txID2 := externalapi.NewDomainTransactionIDFromByteArray(&[externalapi.DomainHashSize]byte{
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02})
|
|
txIDs := []*externalapi.DomainTransactionID{txID1, txID2}
|
|
msg := appmessage.NewMsgRequestTransactions(txIDs)
|
|
err = incomingRoute.Enqueue(msg)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error from incomingRoute.Enqueue: %v", err)
|
|
}
|
|
// The goroutine is representing the peer's actions.
|
|
spawn("peerResponseToTheTransactionsMessages", func() {
|
|
for i, id := range txIDs {
|
|
msg, err := outgoingRoute.Dequeue()
|
|
if err != nil {
|
|
t.Fatalf("Dequeue: %s", err)
|
|
}
|
|
outMsg := msg.(*appmessage.MsgTransactionNotFound)
|
|
if txIDs[i].String() != outMsg.ID.String() {
|
|
t.Fatalf("TestHandleRelayedTransactions: expected equal txID: expected %s, but got %s", txIDs[i].String(), id.String())
|
|
}
|
|
}
|
|
// Closed the incomingRoute for stop the infinity loop.
|
|
incomingRoute.Close()
|
|
})
|
|
|
|
err = transactionrelay.HandleRequestedTransactions(context, incomingRoute, outgoingRoute)
|
|
// Make sure the error is due to the closed route.
|
|
if err == nil || !errors.Is(err, router.ErrRouteClosed) {
|
|
t.Fatalf("Unexpected error: expected: %v, got : %v", router.ErrRouteClosed, err)
|
|
}
|
|
})
|
|
}
|