Skip to content

Commit

Permalink
test(router_validation)_: Added full test cases for validateInputData
Browse files Browse the repository at this point in the history
  • Loading branch information
@Samyoul
Samyoul committed Jul 3, 2024
1 parent f6f1b56 commit 79d0d08
Showing 1 changed file with 231 additions and 0 deletions.
231 changes: 231 additions & 0 deletions services/wallet/router/router_v2_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ package router
import (
"context"
"database/sql"
"errors"
"math/big"
"testing"

Expand Down Expand Up @@ -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())
}
})
}
}

0 comments on commit 79d0d08

Please sign in to comment.