mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-11-27 03:58:06 +00:00
21efccd300
Two changes being done here, because they require a lot of updates to tests. Now we're back into version 0 and we only have one type of transaction. It also removes GetType and GetScript interops, both are obsolete in Neo 3.
104 lines
2.8 KiB
Go
104 lines
2.8 KiB
Go
package integration
|
|
|
|
import (
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/config"
|
|
"github.com/nspcc-dev/neo-go/pkg/core"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/storage"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
|
"github.com/nspcc-dev/neo-go/pkg/network"
|
|
"github.com/nspcc-dev/neo-go/pkg/wallet"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap/zaptest"
|
|
)
|
|
|
|
// Benchmark test to measure number of processed TX.
|
|
// Same benchmark made on reference C# node https://github.com/neo-project/neo/issues/1321.
|
|
func BenchmarkTXPerformanceTest(t *testing.B) {
|
|
net := config.ModeUnitTestNet
|
|
configPath := "../config"
|
|
cfg, err := config.Load(configPath, net)
|
|
require.NoError(t, err, "could not load config")
|
|
|
|
logger := zaptest.NewLogger(t)
|
|
memoryStore := storage.NewMemoryStore()
|
|
chain, err := core.NewBlockchain(memoryStore, cfg.ProtocolConfiguration, logger)
|
|
require.NoError(t, err, "could not create chain")
|
|
|
|
go chain.Run()
|
|
|
|
serverConfig := network.NewServerConfig(cfg)
|
|
server, err := network.NewServer(serverConfig, chain, logger)
|
|
require.NoError(t, err, "could not create server")
|
|
data := prepareData(t)
|
|
t.ResetTimer()
|
|
|
|
for n := 0; n < t.N; n++ {
|
|
assert.Equal(t, network.RelaySucceed, server.RelayTxn(data[n]))
|
|
assert.Equal(t, network.RelayAlreadyExists, server.RelayTxn(data[n]))
|
|
}
|
|
chain.Close()
|
|
}
|
|
|
|
func prepareData(t *testing.B) []*transaction.Transaction {
|
|
var data []*transaction.Transaction
|
|
|
|
wif := getWif(t)
|
|
acc, err := wallet.NewAccountFromWIF(wif.S)
|
|
require.NoError(t, err)
|
|
|
|
for n := 0; n < t.N; n++ {
|
|
tx := getTX(t, wif)
|
|
require.NoError(t, acc.SignTx(tx))
|
|
data = append(data, tx)
|
|
}
|
|
return data
|
|
}
|
|
|
|
// getWif returns Wif.
|
|
func getWif(t *testing.B) *keys.WIF {
|
|
var (
|
|
wifEncoded = "KxhEDBQyyEFymvfJD96q8stMbJMbZUb6D1PmXqBWZDU2WvbvVs9o"
|
|
version = byte(0x00)
|
|
)
|
|
wif, err := keys.WIFDecode(wifEncoded, version)
|
|
require.NoError(t, err)
|
|
return wif
|
|
}
|
|
|
|
// getTX returns Invocation transaction with some random attributes in order to have different hashes.
|
|
func getTX(t *testing.B, wif *keys.WIF) *transaction.Transaction {
|
|
fromAddress := wif.PrivateKey.Address()
|
|
fromAddressHash, err := address.StringToUint160(fromAddress)
|
|
require.NoError(t, err)
|
|
|
|
tx := transaction.New([]byte{0x51}, 1)
|
|
tx.Version = 0
|
|
tx.Sender = fromAddressHash
|
|
tx.Attributes = append(tx.Attributes,
|
|
transaction.Attribute{
|
|
Usage: transaction.Description,
|
|
Data: []byte(randString(10)),
|
|
})
|
|
return tx
|
|
}
|
|
|
|
// String returns a random string with the n as its length.
|
|
func randString(n int) string {
|
|
b := make([]byte, n)
|
|
for i := range b {
|
|
b[i] = byte(Int(65, 90))
|
|
}
|
|
|
|
return string(b)
|
|
}
|
|
|
|
// Int returns a random integer in [min,max).
|
|
func Int(min, max int) int {
|
|
return min + rand.Intn(max-min)
|
|
}
|