0g-chain/x/pricefeed/keeper/keeper_test.go
Draco 614d4e40fe
Update cosmos-sdk to v0.47.7 (#1811)
* Update cometbft, cosmos, ethermint, and ibc-go

* Replace github.com/tendermint/tendermint by github.com/cometbft/cometbft

* Replace github.com/tendermint/tm-db by github.com/cometbft/cometbft-db

* Replace gogo/protobuf with cosmos/gogoproto & simapp replacement

* Replace cosmos-sdk/simapp/helpers with cosmos-sdk/testutil/sims

* Remove no longer used simulations

* Replace ibchost with ibcexported
See https://github.com/cosmos/ibc-go/blob/v7.2.2/docs/migrations/v6-to-v7.md#ibc-module-constants

* Add new consensus params keeper

* Add consensus keeper to blockers

* Fix keeper and module issues in app.go

* Add IsSendEnabledCoins and update SetParams interface changes

* Fix protobuf build for cosmos 47 (#1800)

* fix cp errors by using -f; fix lint by only linting our proto dir;
and use proofs.proto directly from ics23 for ibc-go v7

* run proto-all; commit updated third party deps and swagger changes

* regenerate proto files

* use correct gocosmos build plugin for buf

* re-gen all protobuf files to update paths for new gocosmos plugin

* update protoc and buf to latest versions

* fix staking keeper issues in app.go

* update tally handler for gov changes

* chain id fix and flag fixes

* update deps for cometbft 47.7 upgrade

* remove all module legacy queriers

* update stakingKeeper to pointer

* Replace ModuleCdc from govv1beta1 to govcodec

* remove simulations

* abci.LastCommitInfo → abci.CommitInfo

* Remove unused code in keys.go

* simapp.MakeTestEncodingConfig -> moduletestutil.MakeTestEncodingConfi

* Fix chain id issues in tests

* Fix remaining unit test issues

* Update changelog for upgrade

* Fix e2e tests using updated kvtool

* Update protonet to v47 compatible genesis

* Bump cometbft-db to v0.9.1-kava.1

* Update kvtool

* Remove extra changelog

* Fix merged rocksdb issues

* go mod cleanup

* Bump cometbft-db to v9 and go to 1.21

* Bump rocksdb version to v8.10.0

* Update kvtool to latest version

* Update gin to v1.9.0

* Use ibctm.ModuleName in app_test

* Fallback to genesis chain id instead of client toml

* Remove all simulations

* Fix cdp migrations issue with v47

* Update dependencies to correct tags

---------

Co-authored-by: Nick DeLuca <nickdeluca08@gmail.com>
2024-02-06 17:54:10 -05:00

239 lines
6.5 KiB
Go

package keeper_test
import (
"testing"
"time"
"github.com/stretchr/testify/require"
sdk "github.com/cosmos/cosmos-sdk/types"
tmprototypes "github.com/cometbft/cometbft/proto/tendermint/types"
"github.com/kava-labs/kava/app"
"github.com/kava-labs/kava/x/pricefeed/types"
)
// TestKeeper_SetGetMarket tests adding markets to the pricefeed, getting markets from the store
func TestKeeper_SetGetMarket(t *testing.T) {
tApp := app.NewTestApp()
ctx := tApp.NewContext(true, tmprototypes.Header{})
keeper := tApp.GetPriceFeedKeeper()
mp := types.Params{
Markets: []types.Market{
{MarketID: "tstusd", BaseAsset: "tst", QuoteAsset: "usd", Oracles: []sdk.AccAddress{}, Active: true},
},
}
keeper.SetParams(ctx, mp)
markets := keeper.GetMarkets(ctx)
require.Equal(t, len(markets), 1)
require.Equal(t, markets[0].MarketID, "tstusd")
_, found := keeper.GetMarket(ctx, "tstusd")
require.True(t, found, "market should be found")
_, found = keeper.GetMarket(ctx, "invalidmarket")
require.False(t, found, "invalidmarket should not be found")
mp = types.Params{
Markets: []types.Market{
{MarketID: "tstusd", BaseAsset: "tst", QuoteAsset: "usd", Oracles: []sdk.AccAddress{}, Active: true},
{MarketID: "tst2usd", BaseAsset: "tst2", QuoteAsset: "usd", Oracles: []sdk.AccAddress{}, Active: true},
},
}
keeper.SetParams(ctx, mp)
markets = keeper.GetMarkets(ctx)
require.Equal(t, len(markets), 2)
require.Equal(t, markets[0].MarketID, "tstusd")
require.Equal(t, markets[1].MarketID, "tst2usd")
_, found = keeper.GetMarket(ctx, "nan")
require.Equal(t, found, false)
}
// TestKeeper_GetSetPrice Test Posting the price by an oracle
func TestKeeper_GetSetPrice(t *testing.T) {
_, addrs := app.GeneratePrivKeyAddressPairs(2)
tApp := app.NewTestApp()
ctx := tApp.NewContext(true, tmprototypes.Header{})
keeper := tApp.GetPriceFeedKeeper()
mp := types.Params{
Markets: []types.Market{
{MarketID: "tstusd", BaseAsset: "tst", QuoteAsset: "usd", Oracles: []sdk.AccAddress{}, Active: true},
},
}
keeper.SetParams(ctx, mp)
prices := []struct {
oracle sdk.AccAddress
marketID string
price sdk.Dec
total int
}{
{addrs[0], "tstusd", sdk.MustNewDecFromStr("0.33"), 1},
{addrs[1], "tstusd", sdk.MustNewDecFromStr("0.35"), 2},
{addrs[0], "tstusd", sdk.MustNewDecFromStr("0.37"), 2},
}
for _, p := range prices {
// Set price by oracle 1
pp, err := keeper.SetPrice(
ctx,
p.oracle,
p.marketID,
p.price,
time.Now().UTC().Add(1*time.Hour),
)
require.NoError(t, err)
// Get raw prices
rawPrices := keeper.GetRawPrices(ctx, "tstusd")
require.Equal(t, p.total, len(rawPrices))
require.Contains(t, rawPrices, pp)
// Find the oracle and require price to be same
for _, rp := range rawPrices {
if p.oracle.Equals(rp.OracleAddress) {
require.Equal(t, p.price, rp.Price)
}
}
}
}
// TestKeeper_GetSetCurrentPrice Test Setting the median price of an Asset
func TestKeeper_GetSetCurrentPrice(t *testing.T) {
_, addrs := app.GeneratePrivKeyAddressPairs(5)
tApp := app.NewTestApp()
ctx := tApp.NewContext(true, tmprototypes.Header{}).
WithBlockTime(time.Now().UTC())
keeper := tApp.GetPriceFeedKeeper()
mp := types.Params{
Markets: []types.Market{
{MarketID: "tstusd", BaseAsset: "tst", QuoteAsset: "usd", Oracles: []sdk.AccAddress{}, Active: true},
},
}
keeper.SetParams(ctx, mp)
_, err := keeper.SetPrice(
ctx, addrs[0], "tstusd",
sdk.MustNewDecFromStr("0.33"),
time.Now().Add(time.Hour*1))
require.NoError(t, err)
_, err = keeper.SetPrice(
ctx, addrs[1], "tstusd",
sdk.MustNewDecFromStr("0.35"),
time.Now().Add(time.Hour*1))
require.NoError(t, err)
_, err = keeper.SetPrice(
ctx, addrs[2], "tstusd",
sdk.MustNewDecFromStr("0.34"),
time.Now().Add(time.Hour*1))
require.NoError(t, err)
// Add an expired one which should fail
_, err = keeper.SetPrice(
ctx, addrs[3], "tstusd",
sdk.MustNewDecFromStr("0.9"),
ctx.BlockTime().Add(-time.Hour*1))
require.Error(t, err)
// Add a non-expired price, but will not be counted when BlockTime is changed
_, err = keeper.SetPrice(
ctx, addrs[3], "tstusd",
sdk.MustNewDecFromStr("0.9"),
time.Now().Add(time.Minute*30))
require.NoError(t, err)
// Update block time such that first 3 prices valid but last one is expired
ctx = ctx.WithBlockTime(time.Now().Add(time.Minute * 45))
// Set current price
err = keeper.SetCurrentPrices(ctx, "tstusd")
require.NoError(t, err)
// Get current price
price, err := keeper.GetCurrentPrice(ctx, "tstusd")
require.Nil(t, err)
expCurPrice := sdk.MustNewDecFromStr("0.34")
require.Truef(
t,
price.Price.Equal(expCurPrice),
"expected current price to equal %v, actual %v",
expCurPrice, price.Price,
)
// Even number of oracles
_, err = keeper.SetPrice(
ctx, addrs[4], "tstusd",
sdk.MustNewDecFromStr("0.36"),
time.Now().Add(time.Hour*1))
require.NoError(t, err)
err = keeper.SetCurrentPrices(ctx, "tstusd")
require.NoError(t, err)
price, err = keeper.GetCurrentPrice(ctx, "tstusd")
require.Nil(t, err)
exp := sdk.MustNewDecFromStr("0.345")
require.Truef(t, price.Price.Equal(exp),
"current price %s should be %s",
price.Price.String(),
exp.String(),
)
prices := keeper.GetCurrentPrices(ctx)
require.Equal(t, 1, len(prices))
require.Equal(t, price, prices[0])
}
func TestKeeper_ExpiredSetCurrentPrices(t *testing.T) {
_, addrs := app.GeneratePrivKeyAddressPairs(5)
tApp := app.NewTestApp()
ctx := tApp.NewContext(true, tmprototypes.Header{}).
WithBlockTime(time.Now().UTC())
keeper := tApp.GetPriceFeedKeeper()
mp := types.Params{
Markets: []types.Market{
{MarketID: "tstusd", BaseAsset: "tst", QuoteAsset: "usd", Oracles: []sdk.AccAddress{}, Active: true},
},
}
keeper.SetParams(ctx, mp)
_, err := keeper.SetPrice(
ctx, addrs[0], "tstusd",
sdk.MustNewDecFromStr("0.33"),
time.Now().Add(time.Hour*1))
require.NoError(t, err)
_, err = keeper.SetPrice(
ctx, addrs[1], "tstusd",
sdk.MustNewDecFromStr("0.35"),
time.Now().Add(time.Hour*1))
require.NoError(t, err)
_, err = keeper.SetPrice(
ctx, addrs[2], "tstusd",
sdk.MustNewDecFromStr("0.34"),
time.Now().Add(time.Hour*1))
require.NoError(t, err)
// Update block time such that all prices expire
ctx = ctx.WithBlockTime(time.Now().UTC().Add(time.Hour * 2))
err = keeper.SetCurrentPrices(ctx, "tstusd")
require.ErrorIs(t, types.ErrNoValidPrice, err, "there should be no valid prices to be set")
_, err = keeper.GetCurrentPrice(ctx, "tstusd")
require.ErrorIs(t, types.ErrNoValidPrice, err, "current prices should be invalid")
}