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/0glabs/0g-chain/app" "github.com/0glabs/0g-chain/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") }