package transactions

import (
	"testing"

	"github.com/ethereum/go-ethereum/common/hexutil"
	"github.com/stretchr/testify/assert"
)

func TestSendTxArgsValidity(t *testing.T) {
	// 1. If only data fields is set, valid and return data

	bytes1 := hexutil.Bytes([]byte{0xAA, 0xBB, 0xCC, 0xDD})
	bytes2 := hexutil.Bytes([]byte{0x00, 0x01, 0x02})

	bytesEmpty := hexutil.Bytes([]byte{})

	doSendTxValidityTest(t, SendTxArgs{}, true, nil)
	doSendTxValidityTest(t, SendTxArgs{Input: bytes1}, true, bytes1)
	doSendTxValidityTest(t, SendTxArgs{Data: bytes1}, true, bytes1)
	doSendTxValidityTest(t, SendTxArgs{Input: bytes1, Data: bytes1}, true, bytes1)
	doSendTxValidityTest(t, SendTxArgs{Input: bytes1, Data: bytes2}, false, nil)
	doSendTxValidityTest(t, SendTxArgs{Input: bytes1, Data: bytesEmpty}, true, bytes1)
	doSendTxValidityTest(t, SendTxArgs{Input: bytesEmpty, Data: bytes1}, true, bytes1)
	doSendTxValidityTest(t, SendTxArgs{Input: bytesEmpty, Data: bytesEmpty}, true, bytesEmpty)
}

func doSendTxValidityTest(t *testing.T, args SendTxArgs, expectValid bool, expectValue hexutil.Bytes) {
	assert.Equal(t, expectValid, args.Valid(), "Valid() returned unexpected value")
	if expectValid {
		assert.Equal(t, expectValue, args.GetInput(), "GetInput() returned unexpected value")
	}
}