status-go/services/wallet/transfer/multi_transaction_db_test.go

package transfer

import (
	"math/big"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/common/hexutil"
	wallet_common "github.com/status-im/status-go/services/wallet/common"
	"github.com/status-im/status-go/t/helpers"
	"github.com/status-im/status-go/walletdatabase"
)

func setupTestMultiTransactionDB(t *testing.T) (*MultiTransactionDB, func()) {
	db, err := helpers.SetupTestMemorySQLDB(walletdatabase.DbInitializer{})
	require.NoError(t, err)
	SetMultiTransactionIDGenerator(StaticIDCounter()) // to have different multi-transaction IDs even with fast execution
	return NewMultiTransactionDB(db), func() {
		require.NoError(t, db.Close())
	}
}
func TestCreateMultiTransaction(t *testing.T) {
	mtDB, cleanup := setupTestMultiTransactionDB(t)
	defer cleanup()

	tr := generateTestTransfer(0)
	multiTransaction := GenerateTestSendMultiTransaction(tr)

	err := mtDB.CreateMultiTransaction(&multiTransaction)
	require.NoError(t, err)

	// Add assertions here to verify the result of the CreateMultiTransaction method
	details := NewMultiTxDetails()
	details.IDs = []wallet_common.MultiTransactionIDType{multiTransaction.ID}
	mtx, err := mtDB.ReadMultiTransactions(details)
	require.NoError(t, err)
	require.Len(t, mtx, 1)
	require.True(t, areMultiTransactionsEqual(&multiTransaction, mtx[0]))
}
func TestReadMultiTransactions(t *testing.T) {
	mtDB, cleanup := setupTestMultiTransactionDB(t)
	defer cleanup()

	// Create test multi transactions
	tr := generateTestTransfer(0)
	mt1 := GenerateTestSendMultiTransaction(tr)
	tr20 := generateTestTransfer(1)
	tr21 := generateTestTransfer(2)
	mt2 := GenerateTestBridgeMultiTransaction(tr20, tr21)
	tr3 := generateTestTransfer(3)
	mt3 := GenerateTestSwapMultiTransaction(tr3, "SNT", 100)

	require.NotEqual(t, mt1.ID, mt2.ID)
	require.NotEqual(t, mt1.ID, mt3.ID)

	err := mtDB.CreateMultiTransaction(&mt1)
	require.NoError(t, err)
	err = mtDB.CreateMultiTransaction(&mt2)
	require.NoError(t, err)
	err = mtDB.CreateMultiTransaction(&mt3)
	require.NoError(t, err)

	// Read multi transactions
	details := NewMultiTxDetails()
	details.IDs = []wallet_common.MultiTransactionIDType{mt1.ID, mt2.ID, mt3.ID}
	mtx, err := mtDB.ReadMultiTransactions(details)
	require.NoError(t, err)
	require.Len(t, mtx, 3)
	require.True(t, areMultiTransactionsEqual(&mt1, mtx[0]))
	require.True(t, areMultiTransactionsEqual(&mt2, mtx[1]))
	require.True(t, areMultiTransactionsEqual(&mt3, mtx[2]))
}

func TestUpdateMultiTransaction(t *testing.T) {
	mtDB, cleanup := setupTestMultiTransactionDB(t)
	defer cleanup()

	// Create test multi transaction
	tr := generateTestTransfer(0)
	multiTransaction := GenerateTestSendMultiTransaction(tr)

	err := mtDB.CreateMultiTransaction(&multiTransaction)
	require.NoError(t, err)

	// Update the multi transaction
	multiTransaction.FromNetworkID = 1
	multiTransaction.FromTxHash = common.Hash{1}
	multiTransaction.FromAddress = common.Address{2}
	multiTransaction.FromAsset = "fromAsset1"
	multiTransaction.FromAmount = (*hexutil.Big)(big.NewInt(234))
	multiTransaction.ToNetworkID = 2
	multiTransaction.ToTxHash = common.Hash{3}
	multiTransaction.ToAddress = common.Address{4}
	multiTransaction.ToAsset = "toAsset1"
	multiTransaction.ToAmount = (*hexutil.Big)(big.NewInt(345))
	multiTransaction.Type = MultiTransactionBridge
	multiTransaction.CrossTxID = "crossTxD2"

	err = mtDB.UpdateMultiTransaction(&multiTransaction)
	require.NoError(t, err)

	// Read the updated multi transaction
	details := NewMultiTxDetails()
	details.IDs = []wallet_common.MultiTransactionIDType{multiTransaction.ID}
	mtx, err := mtDB.ReadMultiTransactions(details)
	require.NoError(t, err)
	require.Len(t, mtx, 1)
	require.True(t, areMultiTransactionsEqual(&multiTransaction, mtx[0]))
}

func TestDeleteMultiTransaction(t *testing.T) {
	mtDB, cleanup := setupTestMultiTransactionDB(t)
	defer cleanup()

	// Create test multi transaction
	tr := generateTestTransfer(0)
	multiTransaction := GenerateTestSendMultiTransaction(tr)

	err := mtDB.CreateMultiTransaction(&multiTransaction)
	require.NoError(t, err)

	// Delete the multi transaction
	err = mtDB.DeleteMultiTransaction(multiTransaction.ID)
	require.NoError(t, err)

	// Read the deleted multi transaction
	mtx, err := mtDB.ReadMultiTransactions(&MultiTxDetails{
		IDs: []wallet_common.MultiTransactionIDType{multiTransaction.ID}})
	require.NoError(t, err)
	require.Len(t, mtx, 0)
}