160 lines
3.5 KiB
Go
160 lines
3.5 KiB
Go
package keeper
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"cosmossdk.io/math"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/protocolpool/types"
|
|
)
|
|
|
|
// TestValidateAmount tests the validateAmount function.
|
|
func TestValidateAmount(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
amount sdk.Coins
|
|
expErr bool
|
|
errMsg string
|
|
}{
|
|
{
|
|
name: "nil amount",
|
|
amount: nil,
|
|
expErr: true,
|
|
errMsg: "amount cannot be nil",
|
|
},
|
|
{
|
|
name: "negative coin amount",
|
|
amount: sdk.Coins{
|
|
{
|
|
Denom: "stake",
|
|
Amount: math.NewInt(-100),
|
|
},
|
|
},
|
|
expErr: true,
|
|
errMsg: "-100",
|
|
},
|
|
{
|
|
name: "valid single coin",
|
|
amount: sdk.NewCoins(sdk.NewCoin("stake", math.NewInt(100))),
|
|
expErr: false,
|
|
},
|
|
{
|
|
name: "multiple valid coins",
|
|
amount: sdk.NewCoins(
|
|
sdk.NewCoin("stake", math.NewInt(100)),
|
|
sdk.NewCoin("token", math.NewInt(200)),
|
|
),
|
|
expErr: false,
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
err := validateAmount(tc.amount)
|
|
if tc.expErr {
|
|
require.Error(t, err, "expected an error but got none")
|
|
require.Contains(t, err.Error(), tc.errMsg)
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestValidateContinuousFund(t *testing.T) {
|
|
now := time.Now()
|
|
future := now.Add(1 * time.Hour)
|
|
past := now.Add(-1 * time.Hour)
|
|
|
|
// Create a context with the current block time.
|
|
ctx := sdk.Context{}.WithBlockTime(now)
|
|
|
|
tests := []struct {
|
|
name string
|
|
msg types.MsgCreateContinuousFund
|
|
expErr bool
|
|
errMsg string
|
|
}{
|
|
{
|
|
name: "zero percentage",
|
|
msg: types.MsgCreateContinuousFund{
|
|
Authority: "authority",
|
|
Recipient: "recipient",
|
|
Percentage: math.LegacyZeroDec(),
|
|
Expiry: &future,
|
|
},
|
|
expErr: true,
|
|
errMsg: "percentage cannot be zero or empty",
|
|
},
|
|
{
|
|
name: "negative percentage",
|
|
msg: types.MsgCreateContinuousFund{
|
|
Authority: "authority",
|
|
Recipient: "recipient",
|
|
Percentage: math.LegacyNewDecFromInt(math.NewInt(-1)),
|
|
Expiry: &future,
|
|
},
|
|
expErr: true,
|
|
errMsg: "percentage cannot be negative",
|
|
},
|
|
{
|
|
name: "percentage greater than one",
|
|
msg: types.MsgCreateContinuousFund{
|
|
Authority: "authority",
|
|
Recipient: "recipient",
|
|
Percentage: math.LegacyMustNewDecFromStr("1.1"),
|
|
Expiry: &future,
|
|
},
|
|
expErr: true,
|
|
errMsg: "percentage cannot be greater than one",
|
|
},
|
|
{
|
|
name: "valid percentage with nil expiry",
|
|
msg: types.MsgCreateContinuousFund{
|
|
Authority: "authority",
|
|
Recipient: "recipient",
|
|
Percentage: math.LegacyNewDecWithPrec(5, 1), // 0.5
|
|
Expiry: nil,
|
|
},
|
|
expErr: false,
|
|
},
|
|
{
|
|
name: "valid percentage with future expiry",
|
|
msg: types.MsgCreateContinuousFund{
|
|
Authority: "authority",
|
|
Recipient: "recipient",
|
|
Percentage: math.LegacyNewDecWithPrec(5, 1), // 0.5
|
|
Expiry: &future,
|
|
},
|
|
expErr: false,
|
|
},
|
|
{
|
|
name: "expiry in past",
|
|
msg: types.MsgCreateContinuousFund{
|
|
Authority: "authority",
|
|
Recipient: "recipient",
|
|
Percentage: math.LegacyNewDecWithPrec(5, 1), // 0.5
|
|
Expiry: &past,
|
|
},
|
|
expErr: true,
|
|
errMsg: "cannot be less than the current block time",
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
err := validateContinuousFund(ctx, tc.msg)
|
|
if tc.expErr {
|
|
require.Error(t, err, "expected an error but got none")
|
|
require.Contains(t, err.Error(), tc.errMsg)
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
})
|
|
}
|
|
}
|