test(router_validation)_: Added full test cases for validateInputData
This commit is contained in:
Samuel Hawksby-Robinson
parent
f6f1b56cf7
commit
79d0d08cfd
|
|
@ -3,6 +3,7 @@ package router
|
|||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"errors"
|
||||
"math/big"
|
||||
"testing"
|
||||
|
||||
|
|
@ -2599,3 +2600,233 @@ func TestNoBalanceForTheBestRouteRouterV2(t *testing.T) {
|
|||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestValidateInputData(t *testing.T) {
|
||||
testCases := []struct {
|
||||
name string
|
||||
input *RouteInputParams
|
||||
expectedError error
|
||||
}{
|
||||
{
|
||||
name: "ENSRegister valid data on testnet",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSRegister,
|
||||
Username: "validusername.eth",
|
||||
PublicKey: "validpublickey",
|
||||
TokenID: pathprocessor.SttSymbol,
|
||||
testnetMode: true,
|
||||
},
|
||||
expectedError: nil,
|
||||
},
|
||||
{
|
||||
name: "ENSRegister valid data on mainnet",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSRegister,
|
||||
Username: "validusername.eth",
|
||||
PublicKey: "validpublickey",
|
||||
TokenID: pathprocessor.SntSymbol,
|
||||
},
|
||||
expectedError: nil,
|
||||
},
|
||||
{
|
||||
name: "ENSRegister missing username",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSRegister,
|
||||
PublicKey: "validpublickey",
|
||||
TokenID: pathprocessor.SttSymbol,
|
||||
testnetMode: true,
|
||||
},
|
||||
expectedError: errors.New("username and public key are required for ENSRegister"),
|
||||
},
|
||||
{
|
||||
name: "ENSRegister missing public key",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSRegister,
|
||||
Username: "validusername.eth",
|
||||
TokenID: pathprocessor.SttSymbol,
|
||||
testnetMode: true,
|
||||
},
|
||||
expectedError: errors.New("username and public key are required for ENSRegister"),
|
||||
},
|
||||
{
|
||||
name: "ENSRegister invalid token on testnet",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSRegister,
|
||||
Username: "validusername.eth",
|
||||
PublicKey: "validpublickey",
|
||||
TokenID: "invalidtoken",
|
||||
testnetMode: true,
|
||||
},
|
||||
expectedError: errors.New("only STT is supported for ENSRegister on testnet"),
|
||||
},
|
||||
{
|
||||
name: "ENSRegister invalid token on mainnet",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSRegister,
|
||||
Username: "validusername.eth",
|
||||
PublicKey: "validpublickey",
|
||||
TokenID: "invalidtoken",
|
||||
},
|
||||
expectedError: errors.New("only SNT is supported for ENSRegister"),
|
||||
},
|
||||
{
|
||||
name: "ENSRelease valid data",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSRelease,
|
||||
Username: "validusername.eth",
|
||||
},
|
||||
expectedError: nil,
|
||||
},
|
||||
{
|
||||
name: "ENSRelease missing username",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSRelease,
|
||||
},
|
||||
expectedError: errors.New("username is required for ENSRelease"),
|
||||
},
|
||||
{
|
||||
name: "ENSSetPubKey valid data",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSSetPubKey,
|
||||
Username: "validusername.eth",
|
||||
PublicKey: "validpublickey",
|
||||
},
|
||||
expectedError: nil,
|
||||
},
|
||||
{
|
||||
name: "ENSSetPubKey missing username",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSSetPubKey,
|
||||
PublicKey: "validpublickey",
|
||||
},
|
||||
expectedError: errors.New("username and public key are required for ENSSetPubKey"),
|
||||
},
|
||||
{
|
||||
name: "ENSSetPubKey missing public key",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSSetPubKey,
|
||||
Username: "validusername",
|
||||
},
|
||||
expectedError: errors.New("username and public key are required for ENSSetPubKey"),
|
||||
},
|
||||
{
|
||||
name: "ENSSetPubKey invalid ENS username",
|
||||
input: &RouteInputParams{
|
||||
SendType: ENSSetPubKey,
|
||||
Username: "invalidusername",
|
||||
PublicKey: "validpublickey",
|
||||
},
|
||||
expectedError: errors.New("username and public key are required for ENSSetPubKey"),
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with supported network on testnet",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
walletCommon.EthereumSepolia: (*hexutil.Big)(big.NewInt(10)),
|
||||
},
|
||||
testnetMode: true,
|
||||
},
|
||||
expectedError: nil,
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with supported network on mainnet",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
walletCommon.EthereumMainnet: (*hexutil.Big)(big.NewInt(10)),
|
||||
},
|
||||
},
|
||||
expectedError: nil,
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with supported mainnet network while in test mode",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
walletCommon.EthereumMainnet: (*hexutil.Big)(big.NewInt(10)),
|
||||
},
|
||||
testnetMode: true,
|
||||
},
|
||||
expectedError: errors.New("locked amount is not supported for the selected network"),
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with unsupported network on testnet",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
999: (*hexutil.Big)(big.NewInt(10)),
|
||||
},
|
||||
testnetMode: true,
|
||||
},
|
||||
expectedError: errors.New("locked amount is not supported for the selected network"),
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with unsupported network on mainnet",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
999: (*hexutil.Big)(big.NewInt(10)),
|
||||
},
|
||||
},
|
||||
expectedError: errors.New("locked amount is not supported for the selected network"),
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with negative amount",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
walletCommon.EthereumMainnet: (*hexutil.Big)(big.NewInt(-10)),
|
||||
},
|
||||
},
|
||||
expectedError: errors.New("locked amount must not be negative"),
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with zero amount",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
walletCommon.EthereumMainnet: (*hexutil.Big)(big.NewInt(0)),
|
||||
},
|
||||
},
|
||||
expectedError: nil,
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with zero amounts",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
walletCommon.EthereumMainnet: (*hexutil.Big)(big.NewInt(0)),
|
||||
walletCommon.OptimismMainnet: (*hexutil.Big)(big.NewInt(0)),
|
||||
},
|
||||
},
|
||||
expectedError: nil,
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with all supported networks with zero amount",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
walletCommon.EthereumMainnet: (*hexutil.Big)(big.NewInt(0)),
|
||||
walletCommon.OptimismMainnet: (*hexutil.Big)(big.NewInt(0)),
|
||||
walletCommon.ArbitrumMainnet: (*hexutil.Big)(big.NewInt(0)),
|
||||
},
|
||||
},
|
||||
expectedError: errors.New("all supported chains are excluded, routing impossible"),
|
||||
},
|
||||
{
|
||||
name: "fromLockedAmount with all supported test networks with zero amount",
|
||||
input: &RouteInputParams{
|
||||
FromLockedAmount: map[uint64]*hexutil.Big{
|
||||
walletCommon.EthereumSepolia: (*hexutil.Big)(big.NewInt(0)),
|
||||
walletCommon.OptimismSepolia: (*hexutil.Big)(big.NewInt(0)),
|
||||
walletCommon.ArbitrumSepolia: (*hexutil.Big)(big.NewInt(0)),
|
||||
},
|
||||
testnetMode: true,
|
||||
},
|
||||
expectedError: errors.New("all supported chains are excluded, routing impossible"),
|
||||
},
|
||||
}
|
||||
|
||||
for _, tc := range testCases {
|
||||
t.Run(tc.name, func(t *testing.T) {
|
||||
err := validateInputData(tc.input)
|
||||
if tc.expectedError == nil {
|
||||
assert.NoError(t, err)
|
||||
} else {
|
||||
assert.EqualError(t, err, tc.expectedError.Error())
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue