package smartContract

import (
	"bytes"
	"encoding/hex"
	"errors"
	"fmt"
	"io/ioutil"
	"math/big"
	"strings"
	"testing"
	"time"

	"github.com/ElrondNetwork/elrond-go-core/core"
	"github.com/ElrondNetwork/elrond-go-core/core/check"
	"github.com/ElrondNetwork/elrond-go-core/data/block"
	"github.com/ElrondNetwork/elrond-go-core/data/transaction"
	vmData "github.com/ElrondNetwork/elrond-go-core/data/vm"
	logger "github.com/ElrondNetwork/elrond-go-logger"
	"github.com/ElrondNetwork/elrond-go/integrationTests"
	"github.com/ElrondNetwork/elrond-go/integrationTests/vm"
	"github.com/ElrondNetwork/elrond-go/process"
	"github.com/ElrondNetwork/elrond-go/process/factory"
	"github.com/ElrondNetwork/elrond-go/state"
	systemVm "github.com/ElrondNetwork/elrond-go/vm"
	vmcommon "github.com/ElrondNetwork/elrond-vm-common"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

var log = logger.GetOrCreate("integrationtests/multishard/smartcontract")

func TestSCCallingIntraShard(t *testing.T) {
	if testing.Short() {
		t.Skip("this is not a short test")
	}

	numOfShards := 1
	nodesPerShard := 2
	numMetachainNodes := 0

	nodes := integrationTests.CreateNodes(
		numOfShards,
		nodesPerShard,
		numMetachainNodes,
	)

	idxProposers := make([]int, numOfShards+1)
	for i := 0; i < numOfShards; i++ {
		idxProposers[i] = i * nodesPerShard
	}
	idxProposers[numOfShards] = numOfShards * nodesPerShard

	integrationTests.DisplayAndStartNodes(nodes)

	defer func() {
		for _, n := range nodes {
			n.Close()
		}
	}()

	initialVal := big.NewInt(10000000000000)
	initialVal.Mul(initialVal, initialVal)
	fmt.Printf("Initial minted sum: %s\n", initialVal.String())
	integrationTests.MintAllNodes(nodes, initialVal)

	round := uint64(0)
	nonce := uint64(0)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++

	// mint smart contract holders
	firstSCOwner := []byte("12345678901234567890123456789000")
	secondSCOwner := []byte("99945678901234567890123456789001")

	mintPubKey(firstSCOwner, initialVal, nodes)
	mintPubKey(secondSCOwner, initialVal, nodes)

	// deploy the smart contracts
	firstSCAddress := putDeploySCToDataPool(
		"./testdata/first/first.wasm",
		firstSCOwner,
		0,
		big.NewInt(50),
		"",
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)
	//000000000000000005005d3d53b5d0fcf07d222170978932166ee9f3972d3030
	secondSCAddress := putDeploySCToDataPool(
		"./testdata/second/second.wasm",
		secondSCOwner,
		0,
		big.NewInt(50),
		"",
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)
	//00000000000000000500017cc09151c48b99e2a1522fb70a5118ad4cb26c3031

	// Run two rounds, so the two SmartContracts get deployed.
	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 2, nonce, round, idxProposers)

	time.Sleep(time.Second)

	// Create transactions that invoke "doSomething" from the second SC, which
	// will execute an "asyncCall" to a method in the first SC which counts how
	// many times it has been called. There will be as many transactions as there
	// are nodes.
	for _, node := range nodes {
		txData := "doSomething"
		integrationTests.CreateAndSendTransaction(node, nodes, big.NewInt(50), secondSCAddress, txData, integrationTests.AdditionalGasLimit)
	}
	time.Sleep(time.Second)

	_, _ = integrationTests.WaitOperationToBeDone(t, nodes, 3, nonce, round, idxProposers)

	// verify how many times was the first SC called
	for index, node := range nodes {
		numCalled := vm.GetIntValueFromSC(nil, node.AccntState, firstSCAddress, "numCalled", nil)
		assert.NotNil(t, numCalled)
		if numCalled != nil {
			assert.Equal(t, uint64(len(nodes)), numCalled.Uint64(), fmt.Sprintf("Node %d", index))
		}
	}
}

func TestScDeployAndChangeScOwner(t *testing.T) {
	if testing.Short() {
		t.Skip("this is not a short test")
	}

	sleepDuration := time.Second
	numShards := 2
	nodesPerShard := 2
	numMetachainNodes := 2

	nodes := integrationTests.CreateNodes(
		numShards,
		nodesPerShard,
		numMetachainNodes,
	)

	idxProposers := make([]int, numShards+1)
	for i := 0; i < numShards; i++ {
		idxProposers[i] = i * nodesPerShard
	}
	idxProposers[numShards] = numShards * nodesPerShard

	integrationTests.DisplayAndStartNodes(nodes)

	defer func() {
		for _, n := range nodes {
			n.Close()
		}
	}()

	initialVal := big.NewInt(1000000000)
	integrationTests.MintAllNodes(nodes, initialVal)

	firstSCOwner := nodes[0].OwnAccount.Address

	// deploy the smart contracts
	firstSCAddress := putDeploySCToDataPool(
		"../../vm/arwen/testdata/counter/counter.wasm",
		firstSCOwner,
		0,
		big.NewInt(50),
		"",
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)

	round := uint64(0)
	nonce := uint64(0)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++
	integrationTests.ProposeBlock(nodes, idxProposers, round, nonce)
	integrationTests.SyncBlock(t, nodes, idxProposers, round)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++

	// make smart contract call to shard 1 which will do in shard 0
	for _, node := range nodes {
		txData := "increment"
		for i := 0; i < 10; i++ {
			integrationTests.CreateAndSendTransaction(node, nodes, big.NewInt(0), firstSCAddress, txData, integrationTests.AdditionalGasLimit)
		}
	}

	time.Sleep(sleepDuration)

	numRoundsToPropagateMultiShard := 15
	for i := 0; i < numRoundsToPropagateMultiShard; i++ {
		integrationTests.UpdateRound(nodes, round)
		integrationTests.AddSelfNotarizedHeaderByMetachain(nodes)
		integrationTests.ProposeBlock(nodes, idxProposers, round, nonce)
		integrationTests.SyncBlock(t, nodes, idxProposers, round)
		round = integrationTests.IncrementAndPrintRound(round)
		nonce++
	}

	shId := nodes[0].ShardCoordinator.ComputeId(firstSCAddress)
	for _, node := range nodes {
		if node.ShardCoordinator.SelfId() != shId {
			continue
		}

		numCalled := vm.GetIntValueFromSC(nil, node.AccntState, firstSCAddress, "get", nil)
		require.NotNil(t, numCalled)
	}

	account := getAccountFromAddrBytes(nodes[0].AccntState, nodes[0].OwnAccount.Address)
	require.Equal(t, big.NewInt(0), account.GetDeveloperReward())

	newOwnerAddress := []byte("12345678123456781234567812345678")
	txData := "ChangeOwnerAddress" + "@" + hex.EncodeToString(newOwnerAddress)
	integrationTests.CreateAndSendTransaction(nodes[0], nodes, big.NewInt(0), firstSCAddress, txData, integrationTests.AdditionalGasLimit)
	time.Sleep(sleepDuration)

	for i := 0; i < numRoundsToPropagateMultiShard; i++ {
		integrationTests.UpdateRound(nodes, round)
		integrationTests.AddSelfNotarizedHeaderByMetachain(nodes)
		integrationTests.ProposeBlock(nodes, idxProposers, round, nonce)
		integrationTests.SyncBlock(t, nodes, idxProposers, round)
		round = integrationTests.IncrementAndPrintRound(round)
		nonce++
	}

	// check new owner address is set
	account = getAccountFromAddrBytes(nodes[0].AccntState, firstSCAddress)
	require.Equal(t, newOwnerAddress, account.GetOwnerAddress())
	require.True(t, account.GetDeveloperReward().Cmp(big.NewInt(0)) == 1)
}

func TestScDeployAndClaimSmartContractDeveloperRewards(t *testing.T) {
	// TODO: Fix this test and enable it.
	t.Skip()

	if testing.Short() {
		t.Skip("this is not a short test")
	}

	numShards := 1
	nodesPerShard := 5
	numMetachainNodes := 1

	nodes := integrationTests.CreateNodes(
		numShards,
		nodesPerShard,
		numMetachainNodes,
	)

	idxProposers := make([]int, numShards+1)
	for i := 0; i < numShards; i++ {
		idxProposers[i] = i * nodesPerShard
	}
	idxProposers[numShards] = numShards * nodesPerShard

	integrationTests.DisplayAndStartNodes(nodes)

	defer func() {
		for _, n := range nodes {
			n.Close()
		}
	}()

	initialVal, _ := big.NewInt(0).SetString("100000000000000000000000", 10)
	integrationTests.MintAllNodes(nodes, initialVal)

	firstSCOwner := nodes[0].OwnAccount.Address
	nodes[0].OwnAccount.Nonce += 1
	// deploy the smart contracts
	firstSCAddress := putDeploySCToDataPool(
		"../../vm/arwen/testdata/counter/counter.wasm",
		firstSCOwner,
		0,
		big.NewInt(50),
		"",
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)
	nodes[0].OwnAccount.Nonce += 1

	time.Sleep(time.Second)

	round := uint64(0)
	nonce := uint64(0)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++
	integrationTests.ProposeBlock(nodes, idxProposers, round, nonce)
	integrationTests.SyncBlock(t, nodes, idxProposers, round)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++

	// make smart contract call to shard 1 which will do in shard 0
	numTxsPerNode := 10
	for _, node := range nodes {
		txData := "increment"
		for i := 0; i < numTxsPerNode; i++ {
			integrationTests.CreateAndSendTransaction(node, nodes, big.NewInt(0), firstSCAddress, txData, 50000)
		}
	}

	time.Sleep(time.Second)

	for i := 0; i < 5; i++ {
		integrationTests.UpdateRound(nodes, round)
		integrationTests.AddSelfNotarizedHeaderByMetachain(nodes)
		integrationTests.ProposeBlock(nodes, idxProposers, round, nonce)
		integrationTests.SyncBlock(t, nodes, idxProposers, round)
		round = integrationTests.IncrementAndPrintRound(round)
		nonce++
	}

	shId := nodes[0].ShardCoordinator.ComputeId(firstSCAddress)
	for _, node := range nodes {
		if node.ShardCoordinator.SelfId() != shId {
			continue
		}

		numCalled := vm.GetIntValueFromSC(nil, node.AccntState, firstSCAddress, "get", nil)
		require.NotNil(t, numCalled)
		require.Equal(t, numCalled.Uint64(), uint64(len(nodes)*numTxsPerNode)+1)
	}

	account := getAccountFromAddrBytes(nodes[0].AccntState, nodes[0].OwnAccount.Address)
	require.Equal(t, big.NewInt(0), account.GetDeveloperReward())
	fmt.Println("smart contract owner before claim", account.GetBalance())
	oldOwnerBalance := big.NewInt(0).Set(account.GetBalance())

	account = getAccountFromAddrBytes(nodes[0].AccntState, firstSCAddress)
	fmt.Println("smart contract rewards balance", account.GetDeveloperReward())

	for _, node := range nodes {
		node.EconomicsData.SetGasPerDataByte(0)
		node.EconomicsData.SetMinGasLimit(0)
		node.EconomicsData.SetMinGasPrice(0)
	}

	txData := "ClaimDeveloperRewards"
	integrationTests.CreateAndSendTransaction(nodes[0], nodes, big.NewInt(0), firstSCAddress, txData, integrationTests.AdditionalGasLimit)
	time.Sleep(time.Second)

	for i := 0; i < 3; i++ {
		integrationTests.UpdateRound(nodes, round)
		integrationTests.AddSelfNotarizedHeaderByMetachain(nodes)
		integrationTests.ProposeBlock(nodes, idxProposers, round, nonce)
		integrationTests.SyncBlock(t, nodes, idxProposers, round)
		round = integrationTests.IncrementAndPrintRound(round)
		nonce++
	}

	time.Sleep(time.Second)
	account = getAccountFromAddrBytes(nodes[0].AccntState, nodes[0].OwnAccount.Address)
	fmt.Println("smart contract owner after claim", account.GetBalance())
	require.True(t, account.GetBalance().Cmp(oldOwnerBalance) == 1)
}

func getAccountFromAddrBytes(accState state.AccountsAdapter, address []byte) state.UserAccountHandler {
	sndrAcc, _ := accState.GetExistingAccount(address)

	sndAccSt, _ := sndrAcc.(state.UserAccountHandler)

	return sndAccSt
}

func TestSCCallingInCrossShard(t *testing.T) {
	if testing.Short() {
		t.Skip("this is not a short test")
	}

	numOfShards := 2
	nodesPerShard := 1
	numMetachainNodes := 1

	nodes := integrationTests.CreateNodes(
		numOfShards,
		nodesPerShard,
		numMetachainNodes,
	)

	idxProposers := make([]int, numOfShards+1)
	for i := 0; i < numOfShards; i++ {
		idxProposers[i] = i * nodesPerShard
	}
	idxProposers[numOfShards] = numOfShards * nodesPerShard

	integrationTests.DisplayAndStartNodes(nodes)

	defer func() {
		for _, n := range nodes {
			n.Close()
		}
	}()

	initialVal := big.NewInt(10000000000000)
	initialVal.Mul(initialVal, initialVal)
	fmt.Printf("Initial minted sum: %s\n", initialVal.String())
	integrationTests.MintAllNodes(nodes, initialVal)

	round := uint64(0)
	nonce := uint64(0)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++

	// mint smart contract holders
	firstSCOwner := []byte("12345678901234567890123456789000")
	secondSCOwner := []byte("99945678901234567890123456789001")

	mintPubKey(firstSCOwner, initialVal, nodes)
	mintPubKey(secondSCOwner, initialVal, nodes)

	// deploy the smart contracts
	firstSCAddress := putDeploySCToDataPool(
		"./testdata/first/first.wasm",
		firstSCOwner,
		0,
		big.NewInt(50),
		"",
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)
	//000000000000000005005d3d53b5d0fcf07d222170978932166ee9f3972d3030
	secondSCAddress := putDeploySCToDataPool(
		"./testdata/second/second.wasm",
		secondSCOwner,
		0,
		big.NewInt(50),
		"",
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)
	//00000000000000000500017cc09151c48b99e2a1522fb70a5118ad4cb26c3031

	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 1, nonce, round, idxProposers)

	// make smart contract call to shard 1 which will do in shard 0
	for _, node := range nodes {
		txData := "doSomething"
		integrationTests.PlayerSendsTransaction(
			nodes,
			node.OwnAccount,
			secondSCAddress,
			big.NewInt(50),
			txData,
			500000,
		)
	}

	time.Sleep(time.Second)

	nrRoundsToPropagateMultiShard := 10
	_, _ = integrationTests.WaitOperationToBeDone(t, nodes, nrRoundsToPropagateMultiShard, nonce, round, idxProposers)

	// verify how many times was shard 0 and shard 1 called
	shId := nodes[0].ShardCoordinator.ComputeId(firstSCAddress)
	for index, node := range nodes {
		if node.ShardCoordinator.SelfId() != shId {
			continue
		}

		numCalled := vm.GetIntValueFromSC(nil, node.AccntState, firstSCAddress, "numCalled", nil)
		assert.NotNil(t, numCalled)
		if numCalled != nil {
			assert.Equal(t, uint64(len(nodes)), numCalled.Uint64(), fmt.Sprintf("Node %d, Shard %d", index, shId))
		}
	}
}

func TestSCCallingBuiltinAndFails(t *testing.T) {
	if testing.Short() {
		t.Skip("this is not a short test")
	}

	numOfShards := 2
	nodesPerShard := 1
	numMetachainNodes := 1

	testBuiltinFunc := &integrationTests.TestBuiltinFunction{}
	testBuiltinFunc.Function = func(acntSnd, acntDst vmcommon.UserAccountHandler, vmInput *vmcommon.ContractCallInput) (*vmcommon.VMOutput, error) {
		if !check.IfNil(acntSnd) {
			fmt.Println("builtin snd", hex.EncodeToString(acntSnd.AddressBytes()))
		}
		if !check.IfNil(acntDst) {
			fmt.Println("builtin dst", hex.EncodeToString(acntDst.AddressBytes()))
			return nil, errors.New("some error")
		}

		vmOutput := &vmcommon.VMOutput{}
		vmOutput.ReturnCode = vmcommon.Ok
		vmOutput.OutputAccounts = make(map[string]*vmcommon.OutputAccount)
		outTransfer := vmcommon.OutputTransfer{
			Value:     big.NewInt(0),
			GasLimit:  200000,
			GasLocked: vmInput.GasLocked,
			Data:      []byte("testfunc@01"),
			CallType:  vmData.AsynchronousCall,
		}
		vmOutput.OutputAccounts[string(vmInput.RecipientAddr)] = &vmcommon.OutputAccount{
			Address:         vmInput.RecipientAddr,
			OutputTransfers: []vmcommon.OutputTransfer{outTransfer},
		}
		vmOutput.GasRemaining = vmInput.GasProvided - outTransfer.GasLimit - vmInput.GasLocked

		fmt.Println("OutputAccount recipient", hex.EncodeToString(vmInput.RecipientAddr))

		return vmOutput, nil
	}

	integrationTests.TestBuiltinFunctions["testfunc"] = testBuiltinFunc

	nodes := integrationTests.CreateNodes(
		numOfShards,
		nodesPerShard,
		numMetachainNodes,
	)

	idxProposers := make([]int, numOfShards+1)
	for i := 0; i < numOfShards; i++ {
		idxProposers[i] = i * nodesPerShard
	}
	idxProposers[numOfShards] = numOfShards * nodesPerShard

	integrationTests.DisplayAndStartNodes(nodes)

	defer func() {
		for _, n := range nodes {
			n.Close()
		}
	}()

	initialVal := big.NewInt(10000000000000)
	initialVal.Mul(initialVal, initialVal)
	fmt.Printf("Initial minted sum: %s\n", initialVal.String())
	integrationTests.MintAllNodes(nodes, initialVal)

	round := uint64(0)
	nonce := uint64(0)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++

	require.Equal(t, uint32(0), nodes[0].ShardCoordinator.SelfId())
	require.Equal(t, uint32(1), nodes[1].ShardCoordinator.SelfId())

	scOwner := []byte("12345678901234567890123456789000")
	mintPubKey(scOwner, initialVal, nodes)
	scAddress := putDeploySCToDataPool(
		"./testdata/callBuiltin/output/callBuiltin.wasm",
		scOwner,
		0,
		big.NewInt(0),
		"",
		[]*integrationTests.TestProcessorNode{nodes[0]},
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)

	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 1, nonce, round, idxProposers)

	sender := nodes[0]
	receiver := nodes[1]

	fmt.Println("nodes[0]", hex.EncodeToString(sender.OwnAccount.Address))
	fmt.Println("nodes[1]", hex.EncodeToString(receiver.OwnAccount.Address))
	fmt.Println("scAddress", hex.EncodeToString(scAddress))

	integrationTests.CreateAndSendTransaction(
		sender,
		nodes,
		big.NewInt(0),
		scAddress,
		"callBuiltin@"+hex.EncodeToString(receiver.OwnAccount.Address),
		1500000,
	)

	time.Sleep(time.Second)
	_, _ = integrationTests.WaitOperationToBeDone(t, nodes, 10, nonce, round, idxProposers)

	testValue1 := vm.GetIntValueFromSC(nil, sender.AccntState, scAddress, "testValue1", nil)
	require.NotNil(t, testValue1)
	require.Equal(t, uint64(255), testValue1.Uint64())

	testValue2 := vm.GetIntValueFromSC(nil, sender.AccntState, scAddress, "testValue2", nil)
	require.NotNil(t, testValue2)
	require.Equal(t, uint64(254), testValue2.Uint64())
}

func TestSCCallingInCrossShardDelegationMock(t *testing.T) {
	if testing.Short() {
		t.Skip("this is not a short test")
	}

	numOfShards := 2
	nodesPerShard := 3
	numMetachainNodes := 3
	shardConsensusGroupSize := 2
	metaConsensusGroupSize := 2

	nodesMap := integrationTests.CreateNodesWithNodesCoordinator(
		nodesPerShard,
		numMetachainNodes,
		numOfShards,
		shardConsensusGroupSize,
		metaConsensusGroupSize,
	)

	nodes := make([]*integrationTests.TestProcessorNode, 0)
	idxProposers := make([]int, numOfShards+1)

	for _, nds := range nodesMap {
		nodes = append(nodes, nds...)
	}

	for _, nds := range nodesMap {
		idx, err := getNodeIndex(nodes, nds[0])
		assert.Nil(t, err)

		idxProposers = append(idxProposers, idx)
	}

	integrationTests.DisplayAndStartNodes(nodes)

	defer func() {
		for _, n := range nodes {
			n.Close()
		}
	}()

	initialVal := big.NewInt(1000000000)
	integrationTests.MintAllNodes(nodes, initialVal)

	round := uint64(0)
	nonce := uint64(0)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++

	// mint smart contract holders
	delegateSCOwner := []byte("12345678901234567890123456789002")
	stakerBLSKey, _ := hex.DecodeString(strings.Repeat("a", 256))

	mintPubKey(delegateSCOwner, initialVal, nodes)

	// deploy the smart contracts
	delegateSCAddress := putDeploySCToDataPool(
		"./testdata/delegate-mock/delegate.wasm",
		delegateSCOwner,
		0,
		big.NewInt(50),
		"",
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)

	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 1, nonce, round, idxProposers)

	// one node calls to stake all the money from the delegation - that's how the contract is :D
	node := nodes[0]
	genesisBlock := node.GenesisBlocks[core.MetachainShardId]
	metaBlock := genesisBlock.(*block.MetaBlock)
	nodePrice := big.NewInt(0).Set(metaBlock.EpochStart.Economics.NodePrice)
	txData := "sendToStaking"
	integrationTests.PlayerSendsTransaction(nodes, node.OwnAccount, delegateSCAddress, nodePrice, txData, 500000)

	time.Sleep(time.Second)

	nrRoundsToPropagateMultiShard := 10
	_, _ = integrationTests.WaitOperationToBeDone(t, nodes, nrRoundsToPropagateMultiShard, nonce, round, idxProposers)

	time.Sleep(time.Second)
	// verify system smart contract has the value
	for _, n := range nodes {
		if n.ShardCoordinator.SelfId() != core.MetachainShardId {
			continue
		}
		scQuery := &process.SCQuery{
			ScAddress: systemVm.StakingSCAddress,
			FuncName:  "isStaked",
			Arguments: [][]byte{stakerBLSKey},
		}
		vmOutput, _ := n.SCQueryService.ExecuteQuery(scQuery)

		assert.NotNil(t, vmOutput)
		if vmOutput != nil {
			assert.Equal(t, vmOutput.ReturnCode, vmcommon.Ok)
		}
	}
}

func TestSCCallingInCrossShardDelegation(t *testing.T) {
	if testing.Short() {
		t.Skip("this is not a short test")
	}

	numOfShards := 2
	nodesPerShard := 3
	numMetachainNodes := 3
	shardConsensusGroupSize := 2
	metaConsensusGroupSize := 2

	nodesMap := integrationTests.CreateNodesWithNodesCoordinator(
		nodesPerShard,
		numMetachainNodes,
		numOfShards,
		shardConsensusGroupSize,
		metaConsensusGroupSize,
	)

	nodes := make([]*integrationTests.TestProcessorNode, 0)
	idxProposers := make([]int, numOfShards+1)

	for _, nds := range nodesMap {
		nodes = append(nodes, nds...)
	}

	for _, nds := range nodesMap {
		idx, err := getNodeIndex(nodes, nds[0])
		assert.Nil(t, err)

		idxProposers = append(idxProposers, idx)
	}

	integrationTests.DisplayAndStartNodes(nodes)

	defer func() {
		for _, n := range nodes {
			n.Close()
		}
	}()

	initialVal := big.NewInt(1000000000)
	integrationTests.MintAllNodes(nodes, initialVal)

	round := uint64(0)
	nonce := uint64(0)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++

	// mint smart contract holders
	shardNode := findAnyShardNode(nodes)
	delegateSCOwner := shardNode.OwnAccount.Address
	node := nodes[0]
	genesisBlock := node.GenesisBlocks[core.MetachainShardId]
	metaBlock := genesisBlock.(*block.MetaBlock)
	nodePrice := big.NewInt(0).Set(metaBlock.EpochStart.Economics.NodePrice)
	totalStake := nodePrice // 1 node only in this test
	serviceFeePer10000 := 3000
	blocksBeforeForceUnstake := 50
	blocksBeforeUnBond := 60
	stakerBLSKey, _ := hex.DecodeString(strings.Repeat("a", 96*2))
	stakerBLSSignature, _ := hex.DecodeString(strings.Repeat("c", 32*2))

	delegateSCAddress := putDeploySCToDataPool(
		"./testdata/delegate/delegation.wasm",
		delegateSCOwner,
		0,
		big.NewInt(0),
		"@"+hex.EncodeToString(systemVm.ValidatorSCAddress)+"@"+core.ConvertToEvenHex(serviceFeePer10000)+"@"+core.ConvertToEvenHex(blocksBeforeForceUnstake)+"@"+core.ConvertToEvenHex(blocksBeforeUnBond),
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)
	shardNode.OwnAccount.Nonce++

	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 1, nonce, round, idxProposers)

	// check that the version is the expected one
	scQueryVersion := &process.SCQuery{
		ScAddress: delegateSCAddress,
		FuncName:  "version",
		Arguments: [][]byte{},
	}
	vmOutputVersion, _ := shardNode.SCQueryService.ExecuteQuery(scQueryVersion)
	assert.NotNil(t, vmOutputVersion)
	assert.Equal(t, len(vmOutputVersion.ReturnData), 1)
	require.True(t, bytes.Contains(vmOutputVersion.ReturnData[0], []byte("0.3.")))
	log.Info("SC deployed", "version", string(vmOutputVersion.ReturnData[0]))

	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 1, nonce, round, idxProposers)

	// set stake per node
	setStakePerNodeTxData := "setStakePerNode@" + core.ConvertToEvenHexBigInt(nodePrice)
	integrationTests.CreateAndSendTransaction(shardNode, nodes, big.NewInt(0), delegateSCAddress, setStakePerNodeTxData, integrationTests.AdditionalGasLimit)

	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 1, nonce, round, idxProposers)

	// add node
	addNodesTxData := fmt.Sprintf("addNodes@%s@%s",
		hex.EncodeToString(stakerBLSKey),
		hex.EncodeToString(stakerBLSSignature))
	integrationTests.CreateAndSendTransaction(shardNode, nodes, big.NewInt(0), delegateSCAddress, addNodesTxData, integrationTests.AdditionalGasLimit)

	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 1, nonce, round, idxProposers)

	// stake some coin!
	// here the node account fills all the required stake
	stakeTxData := "stake"
	integrationTests.CreateAndSendTransaction(shardNode, nodes, totalStake, delegateSCAddress, stakeTxData, integrationTests.AdditionalGasLimit)

	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 1, nonce, round, idxProposers)

	// activate the delegation, this involves an async call to validatorSC
	stakeAllAvailableTxData := "stakeAllAvailable"
	integrationTests.CreateAndSendTransaction(shardNode, nodes, big.NewInt(0), delegateSCAddress, stakeAllAvailableTxData, integrationTests.AdditionalGasLimit)

	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 1, nonce, round, idxProposers)

	time.Sleep(time.Second)

	nrRoundsToPropagateMultiShard := 10
	_, _ = integrationTests.WaitOperationToBeDone(t, nodes, nrRoundsToPropagateMultiShard, nonce, round, idxProposers)

	time.Sleep(time.Second)

	// check that delegation contract was correctly initialized by checking number of nodes added
	scQuery1 := &process.SCQuery{
		ScAddress: delegateSCAddress,
		FuncName:  "getNumNodes",
		Arguments: [][]byte{},
	}
	vmOutput1, _ := shardNode.SCQueryService.ExecuteQuery(scQuery1)
	require.NotNil(t, vmOutput1)
	require.Equal(t, len(vmOutput1.ReturnData), 1)
	require.True(t, bytes.Equal(vmOutput1.ReturnData[0], []byte{1}))

	// check that node/signature are correctly set
	scQuery2 := &process.SCQuery{
		ScAddress: delegateSCAddress,
		FuncName:  "getNodeSignature",
		Arguments: [][]byte{stakerBLSKey},
	}
	vmOutput2, _ := shardNode.SCQueryService.ExecuteQuery(scQuery2)
	require.NotNil(t, vmOutput2)
	require.Equal(t, len(vmOutput2.ReturnData), 1)
	require.True(t, bytes.Equal(stakerBLSSignature, vmOutput2.ReturnData[0]))

	// check that the stake got registered in delegation
	scQuery3 := &process.SCQuery{
		ScAddress: delegateSCAddress,
		FuncName:  "getUserStake",
		Arguments: [][]byte{delegateSCOwner},
	}
	vmOutput3, _ := shardNode.SCQueryService.ExecuteQuery(scQuery3)
	require.NotNil(t, vmOutput3)
	require.Equal(t, len(vmOutput3.ReturnData), 1)
	require.True(t, totalStake.Cmp(big.NewInt(0).SetBytes(vmOutput3.ReturnData[0])) == 0)

	// check that the stake got activated
	scQuery4 := &process.SCQuery{
		ScAddress: delegateSCAddress,
		FuncName:  "getUserActiveStake",
		Arguments: [][]byte{delegateSCOwner},
	}
	vmOutput4, _ := shardNode.SCQueryService.ExecuteQuery(scQuery4)
	require.NotNil(t, vmOutput4)
	require.Equal(t, len(vmOutput4.ReturnData), 1)
	require.True(t, totalStake.Cmp(big.NewInt(0).SetBytes(vmOutput4.ReturnData[0])) == 0)

	// check that the staking system smart contract has the value
	for _, n := range nodes {
		if n.ShardCoordinator.SelfId() != core.MetachainShardId {
			continue
		}
		scQuery := &process.SCQuery{
			ScAddress: systemVm.StakingSCAddress,
			FuncName:  "isStaked",
			Arguments: [][]byte{stakerBLSKey},
		}
		vmOutput, _ := n.SCQueryService.ExecuteQuery(scQuery)

		assert.NotNil(t, vmOutput)
		if vmOutput != nil {
			assert.Equal(t, vmOutput.ReturnCode, vmcommon.Ok)
		}
	}
}

func TestSCNonPayableIntraShardErrorShouldProcessBlock(t *testing.T) {
	if testing.Short() {
		t.Skip("this is not a short test")
	}

	numOfShards := 1
	nodesPerShard := 3
	numMetachainNodes := 0

	nodes := integrationTests.CreateNodes(
		numOfShards,
		nodesPerShard,
		numMetachainNodes,
	)

	idxProposers := make([]int, numOfShards+1)
	for i := 0; i < numOfShards; i++ {
		idxProposers[i] = i * nodesPerShard
	}
	idxProposers[numOfShards] = numOfShards * nodesPerShard

	integrationTests.DisplayAndStartNodes(nodes)

	defer func() {
		for _, n := range nodes {
			n.Close()
		}
	}()

	initialVal := big.NewInt(10000000000000)
	initialVal.Mul(initialVal, initialVal)
	fmt.Printf("Initial minted sum: %s\n", initialVal.String())
	integrationTests.MintAllNodes(nodes, initialVal)

	round := uint64(0)
	nonce := uint64(0)
	round = integrationTests.IncrementAndPrintRound(round)
	nonce++

	// mint smart contract holders
	firstSCOwner := []byte("12345678901234567890123456789000")
	secondSCOwner := []byte("99945678901234567890123456789001")

	mintPubKey(firstSCOwner, initialVal, nodes)
	mintPubKey(secondSCOwner, initialVal, nodes)

	// deploy the smart contracts
	_ = putDeploySCToDataPool(
		"./testdata/first/first.wasm",
		firstSCOwner,
		0,
		big.NewInt(50),
		"",
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)
	//000000000000000005005d3d53b5d0fcf07d222170978932166ee9f3972d3030
	secondSCAddress := putDeploySCToDataPool(
		"./testdata/second/second.wasm",
		secondSCOwner,
		0,
		big.NewInt(50),
		"",
		nodes,
		nodes[0].EconomicsData.MaxGasLimitPerBlock(0)-1,
	)
	//00000000000000000500017cc09151c48b99e2a1522fb70a5118ad4cb26c3031

	// Run two rounds, so the two SmartContracts get deployed.
	nonce, round = integrationTests.WaitOperationToBeDone(t, nodes, 2, nonce, round, idxProposers)

	time.Sleep(time.Second)

	// Create transactions that invoke "doSomething" from the second SC, which
	// will execute an "asyncCall" to a method in the first SC which counts how
	// many times it has been called. There will be as many transactions as there
	// are nodes.
	for _, node := range nodes {
		txData := "doSomething@WRONG"
		integrationTests.CreateAndSendTransaction(node, nodes, big.NewInt(50), secondSCAddress, txData, integrationTests.AdditionalGasLimit)
	}
	time.Sleep(time.Second)

	_, _ = integrationTests.WaitOperationToBeDone(t, nodes, 3, nonce, round, idxProposers)

	for _, node := range nodes {
		assert.Equal(t, uint64(5), node.BlockChain.GetCurrentBlockHeader().GetNonce())
	}
}

func getNodeIndex(nodeList []*integrationTests.TestProcessorNode, node *integrationTests.TestProcessorNode) (int, error) {
	for i := range nodeList {
		if node == nodeList[i] {
			return i, nil
		}
	}

	return 0, errors.New("no such node in list")
}

func putDeploySCToDataPool(
	fileName string,
	pubkey []byte,
	nonce uint64,
	transferOnDeploy *big.Int,
	initArgs string,
	nodes []*integrationTests.TestProcessorNode,
	gasLimit uint64,
) []byte {
	scCode, err := ioutil.ReadFile(fileName)
	if err != nil {
		panic(fmt.Sprintf("putDeploySCToDataPool(): %s", err))
	}

	scCodeString := hex.EncodeToString(scCode)
	scCodeMetadataString := "0000"

	blockChainHook := nodes[0].BlockchainHook

	scAddressBytes, _ := blockChainHook.NewAddress(pubkey, nonce, factory.ArwenVirtualMachine)

	tx := &transaction.Transaction{
		Nonce:    nonce,
		Value:    new(big.Int).Set(transferOnDeploy),
		RcvAddr:  make([]byte, 32),
		SndAddr:  pubkey,
		GasPrice: nodes[0].EconomicsData.GetMinGasPrice(),
		GasLimit: gasLimit,
		Data:     []byte(scCodeString + "@" + hex.EncodeToString(factory.ArwenVirtualMachine) + "@" + scCodeMetadataString + initArgs),
		ChainID:  integrationTests.ChainID,
	}
	txHash, _ := core.CalculateHash(integrationTests.TestMarshalizer, integrationTests.TestHasher, tx)

	shId := nodes[0].ShardCoordinator.ComputeId(pubkey)

	for _, node := range nodes {
		if node.ShardCoordinator.SelfId() != shId {
			continue
		}
		strCache := process.ShardCacherIdentifier(shId, shId)
		node.DataPool.Transactions().AddData(txHash, tx, tx.Size(), strCache)
	}

	return scAddressBytes
}

func mintPubKey(
	pubkey []byte,
	initialVal *big.Int,
	nodes []*integrationTests.TestProcessorNode,
) {
	shId := nodes[0].ShardCoordinator.ComputeId(pubkey)
	for _, node := range nodes {
		if node.ShardCoordinator.SelfId() != shId {
			continue
		}
		integrationTests.MintAddress(node.AccntState, pubkey, initialVal)
	}
}

func findAnyShardNode(nodes []*integrationTests.TestProcessorNode) *integrationTests.TestProcessorNode {
	for _, node := range nodes {
		if node.ShardCoordinator.SelfId() != core.MetachainShardId {
			return node
		}
	}
	panic("no shard nodes found in test")
}
