4bytes_db_update_test.go

package main

import (
	"testing"

	"github.com/AstraProtocol/astra-indexing/internal/evm"
	"github.com/stretchr/testify/assert"
)

func TestUpdate4Bytes(t *testing.T) {
	assert := assert.New(t)

	util, err := evm.NewEvmUtils()
	if err != nil {
		panic(err)
	}

	util.UpdateSignature("0x18cbafe5", "swapExactTokensForETH(uint256, uint256, address[], address, uint256)")
	util.UpdateSignature("0xcea08621", "changeDailyLimit(uint256)")
	util.UpdateSignature("0x2fafb7c2", "changeRewardBudget(uint256)")

	res, err := util.GetSignature("0xa07aea1c")
	if err != nil {
		panic(err)
	}
	assert.Equal(res, "addOperators(address[])")

	input := "0x18cbafe5"
	assert.Equal(util.GetMethodNameFromMethodId(input[2:10]), "swapExactTokensForETH")
	assert.Equal(util.GetMethodNameFromMethodId("0x095ea7b3"), "approve")
	assert.Equal(util.GetMethodNameFromMethodId("0xe2bbb158"), "deposit")
	assert.Equal(util.GetMethodNameFromMethodId("0xded9382a"), "removeLiquidityETHWithPermit")
	assert.Equal(util.GetMethodNameFromMethodId("0xa9059cbb"), "transfer")
	assert.Equal(util.GetMethodNameFromMethodId("0x441a3e70"), "withdraw")
	assert.Equal(util.GetMethodNameFromMethodId("0x71679309"), "zapInEth")
	assert.Equal(util.GetMethodNameFromMethodId("0xf305d719"), "addLiquidityETH")
	assert.Equal(util.GetMethodNameFromMethodId("0x2195995c"), "removeLiquidityWithPermit")
	assert.Equal(util.GetMethodNameFromMethodId("0xb6f9de95"), "swapExactETHForTokensSupportingFeeOnTransferTokens")
	assert.Equal(util.GetMethodNameFromMethodId("0x38ed1739"), "swapExactTokensForTokens")
	assert.Equal(util.GetMethodNameFromMethodId("0xe8e33700"), "addLiquidity")
	assert.Equal(util.GetMethodNameFromMethodId("0x42842e0e"), "safeTransferFrom")
	assert.Equal(util.GetMethodNameFromMethodId("0xecd5ff33"), "setProgramTime")
	assert.Equal(util.GetMethodNameFromMethodId("0xbe739356"), "sendReward")
	assert.Equal(util.GetMethodNameFromMethodId("0x743cec2d"), "exchangeWithValue")
	assert.Equal(util.GetMethodNameFromMethodId("0x1482dcda"), "mintCoupons")
	assert.Equal(util.GetMethodNameFromMethodId("0xcea08621"), "changeDailyLimit")
	assert.Equal(util.GetMethodNameFromMethodId("0x2fafb7c2"), "changeRewardBudget")

	data := "0q7S1gAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABgAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAGSWzhIAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAoAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACcAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAQVbZvQpCuU8VUVDSeyAd243bdr5ljiPA3bK7aUm/VuQVdhajtudaHAI7ltLe1GHQeZchiXxaXAq2Ez+6ErCpJRscAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=="
	signature := util.GetMethodNameFromData(data)
	assert.Equal(signature, "exchange")
}