141 lines
4.2 KiB
Go
141 lines
4.2 KiB
Go
package types
|
|
|
|
import (
|
|
"testing"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPoolEqual(t *testing.T) {
|
|
p1 := InitialPool()
|
|
p2 := InitialPool()
|
|
|
|
ok := p1.Equal(p2)
|
|
require.True(t, ok)
|
|
|
|
p2.BondedTokens = 3
|
|
p2.BondedShares = sdk.NewRat(10)
|
|
|
|
ok = p1.Equal(p2)
|
|
require.False(t, ok)
|
|
}
|
|
|
|
func TestBondedRatio(t *testing.T) {
|
|
pool := InitialPool()
|
|
pool.LooseTokens = 1
|
|
pool.BondedTokens = 2
|
|
|
|
// bonded pool / total supply
|
|
require.Equal(t, pool.BondedRatio(), sdk.NewRat(2).Quo(sdk.NewRat(3)))
|
|
|
|
// avoids divide-by-zero
|
|
pool.LooseTokens = 0
|
|
pool.BondedTokens = 0
|
|
require.Equal(t, pool.BondedRatio(), sdk.ZeroRat())
|
|
}
|
|
|
|
func TestBondedShareExRate(t *testing.T) {
|
|
pool := InitialPool()
|
|
pool.BondedTokens = 3
|
|
pool.BondedShares = sdk.NewRat(10)
|
|
|
|
// bonded pool / bonded shares
|
|
require.Equal(t, pool.BondedShareExRate(), sdk.NewRat(3).Quo(sdk.NewRat(10)))
|
|
pool.BondedShares = sdk.ZeroRat()
|
|
|
|
// avoids divide-by-zero
|
|
require.Equal(t, pool.BondedShareExRate(), sdk.OneRat())
|
|
}
|
|
|
|
func TestUnbondingShareExRate(t *testing.T) {
|
|
pool := InitialPool()
|
|
pool.UnbondingTokens = 3
|
|
pool.UnbondingShares = sdk.NewRat(10)
|
|
|
|
// unbonding pool / unbonding shares
|
|
require.Equal(t, pool.UnbondingShareExRate(), sdk.NewRat(3).Quo(sdk.NewRat(10)))
|
|
pool.UnbondingShares = sdk.ZeroRat()
|
|
|
|
// avoids divide-by-zero
|
|
require.Equal(t, pool.UnbondingShareExRate(), sdk.OneRat())
|
|
}
|
|
|
|
func TestUnbondedShareExRate(t *testing.T) {
|
|
pool := InitialPool()
|
|
pool.UnbondedTokens = 3
|
|
pool.UnbondedShares = sdk.NewRat(10)
|
|
|
|
// unbonded pool / unbonded shares
|
|
require.Equal(t, pool.UnbondedShareExRate(), sdk.NewRat(3).Quo(sdk.NewRat(10)))
|
|
pool.UnbondedShares = sdk.ZeroRat()
|
|
|
|
// avoids divide-by-zero
|
|
require.Equal(t, pool.UnbondedShareExRate(), sdk.OneRat())
|
|
}
|
|
|
|
func TestAddTokensBonded(t *testing.T) {
|
|
poolA := InitialPool()
|
|
poolA.LooseTokens = 10
|
|
require.Equal(t, poolA.BondedShareExRate(), sdk.OneRat())
|
|
|
|
poolB, sharesB := poolA.addTokensBonded(10)
|
|
require.Equal(t, poolB.BondedShareExRate(), sdk.OneRat())
|
|
|
|
// correct changes to bonded shares and bonded pool
|
|
require.Equal(t, poolB.BondedShares, poolA.BondedShares.Add(sharesB.Amount))
|
|
require.Equal(t, poolB.BondedTokens, poolA.BondedTokens+10)
|
|
|
|
// same number of bonded shares / tokens when exchange rate is one
|
|
require.True(t, poolB.BondedShares.Equal(sdk.NewRat(poolB.BondedTokens)))
|
|
}
|
|
|
|
func TestRemoveSharesBonded(t *testing.T) {
|
|
poolA := InitialPool()
|
|
poolA.LooseTokens = 10
|
|
require.Equal(t, poolA.BondedShareExRate(), sdk.OneRat())
|
|
|
|
poolB, tokensB := poolA.removeSharesBonded(sdk.NewRat(10))
|
|
require.Equal(t, poolB.BondedShareExRate(), sdk.OneRat())
|
|
|
|
// correct changes to bonded shares and bonded pool
|
|
require.Equal(t, poolB.BondedShares, poolA.BondedShares.Sub(sdk.NewRat(10)))
|
|
require.Equal(t, poolB.BondedTokens, poolA.BondedTokens-tokensB)
|
|
|
|
// same number of bonded shares / tokens when exchange rate is one
|
|
require.True(t, poolB.BondedShares.Equal(sdk.NewRat(poolB.BondedTokens)))
|
|
}
|
|
|
|
func TestAddTokensUnbonded(t *testing.T) {
|
|
poolA := InitialPool()
|
|
poolA.LooseTokens = 10
|
|
require.Equal(t, poolA.UnbondedShareExRate(), sdk.OneRat())
|
|
|
|
poolB, sharesB := poolA.addTokensUnbonded(10)
|
|
require.Equal(t, poolB.UnbondedShareExRate(), sdk.OneRat())
|
|
|
|
// correct changes to unbonded shares and unbonded pool
|
|
require.Equal(t, poolB.UnbondedShares, poolA.UnbondedShares.Add(sharesB.Amount))
|
|
require.Equal(t, poolB.UnbondedTokens, poolA.UnbondedTokens+10)
|
|
|
|
// same number of unbonded shares / tokens when exchange rate is one
|
|
require.True(t, poolB.UnbondedShares.Equal(sdk.NewRat(poolB.UnbondedTokens)))
|
|
}
|
|
|
|
func TestRemoveSharesUnbonded(t *testing.T) {
|
|
poolA := InitialPool()
|
|
poolA.UnbondedTokens = 10
|
|
poolA.UnbondedShares = sdk.NewRat(10)
|
|
require.Equal(t, poolA.UnbondedShareExRate(), sdk.OneRat())
|
|
|
|
poolB, tokensB := poolA.removeSharesUnbonded(sdk.NewRat(10))
|
|
require.Equal(t, poolB.UnbondedShareExRate(), sdk.OneRat())
|
|
|
|
// correct changes to unbonded shares and bonded pool
|
|
require.Equal(t, poolB.UnbondedShares, poolA.UnbondedShares.Sub(sdk.NewRat(10)))
|
|
require.Equal(t, poolB.UnbondedTokens, poolA.UnbondedTokens-tokensB)
|
|
|
|
// same number of unbonded shares / tokens when exchange rate is one
|
|
require.True(t, poolB.UnbondedShares.Equal(sdk.NewRat(poolB.UnbondedTokens)))
|
|
}
|