cli/pkg/vm/airdrop_test.go

// Copyright (C) 2022, Lux Industries Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package vm

import (
	"math/big"
	"testing"

	"github.com/luxfi/cli/v2/internal/mocks"
	"github.com/luxfi/cli/v2/pkg/application"
	"github.com/luxfi/cli/v2/pkg/statemachine"
	"github.com/luxfi/geth/common"
	"github.com/stretchr/testify/mock"
)

var testAirdropAddress = common.HexToAddress("0x098B69E43b1720Bd12378225519d74e5F3aD0eA5")

func TestGetAllocationCustomUnits(t *testing.T) {
	require := setupTest(t)
	app := application.New()
	mockPrompt := &mocks.Prompter{}
	app.Prompt = mockPrompt

	airdropInputAmount := new(big.Int)
	airdropInputAmount.SetString("1000000", 10)

	expectedAmount := new(big.Int)
	expectedAmount.SetString(defaultEvmAirdropAmount, 10)

	mockPrompt.On("CaptureList", mock.Anything, mock.Anything).Return(customAirdrop, nil)
	mockPrompt.On("CaptureAddress", mock.Anything).Return(testAirdropAddress, nil)
	mockPrompt.On("CapturePositiveBigInt", mock.Anything).Return(airdropInputAmount, nil)
	mockPrompt.On("CaptureNoYes", mock.Anything).Return(false, nil)

	alloc, direction, err := getEVMAllocation(app)
	require.NoError(err)
	require.Equal(direction, statemachine.Forward)

	require.Equal(alloc[testAirdropAddress].Balance, expectedAmount)
}