mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 03:55:19 +00:00
73 lines
2.0 KiB
Go
73 lines
2.0 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
tmprototypes "github.com/cometbft/cometbft/proto/tendermint/types"
|
|
tmtime "github.com/cometbft/cometbft/types/time"
|
|
|
|
"github.com/0glabs/0g-chain/app"
|
|
"github.com/0glabs/0g-chain/x/pricefeed/keeper"
|
|
"github.com/0glabs/0g-chain/x/pricefeed/types"
|
|
)
|
|
|
|
type KeeperTestSuite struct {
|
|
suite.Suite
|
|
|
|
keeper keeper.Keeper
|
|
addrs []sdk.AccAddress
|
|
ctx sdk.Context
|
|
}
|
|
|
|
func (suite *KeeperTestSuite) SetupTest() {
|
|
tApp := app.NewTestApp()
|
|
ctx := tApp.NewContext(true, tmprototypes.Header{Height: 1, Time: tmtime.Now()})
|
|
tApp.InitializeFromGenesisStates(
|
|
NewPricefeedGenStateMulti(),
|
|
)
|
|
suite.keeper = tApp.GetPriceFeedKeeper()
|
|
suite.ctx = ctx
|
|
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(10)
|
|
suite.addrs = addrs
|
|
}
|
|
|
|
func (suite *KeeperTestSuite) TestGetSetOracles() {
|
|
params := suite.keeper.GetParams(suite.ctx)
|
|
suite.Equal([]sdk.AccAddress(nil), params.Markets[0].Oracles)
|
|
|
|
params.Markets[0].Oracles = suite.addrs
|
|
suite.NotPanics(func() { suite.keeper.SetParams(suite.ctx, params) })
|
|
params = suite.keeper.GetParams(suite.ctx)
|
|
suite.Equal(suite.addrs, params.Markets[0].Oracles)
|
|
|
|
addr, err := suite.keeper.GetOracle(suite.ctx, params.Markets[0].MarketID, suite.addrs[0])
|
|
suite.NoError(err)
|
|
suite.Equal(suite.addrs[0], addr)
|
|
}
|
|
|
|
func (suite *KeeperTestSuite) TestGetAuthorizedAddresses() {
|
|
_, oracles := app.GeneratePrivKeyAddressPairs(5)
|
|
|
|
params := types.Params{
|
|
Markets: []types.Market{
|
|
{MarketID: "btc:usd", BaseAsset: "btc", QuoteAsset: "usd", Oracles: oracles[:3], Active: true},
|
|
{MarketID: "xrp:usd", BaseAsset: "xrp", QuoteAsset: "usd", Oracles: oracles[2:], Active: true},
|
|
{MarketID: "xrp:usd:30", BaseAsset: "xrp", QuoteAsset: "usd", Oracles: nil, Active: true},
|
|
},
|
|
}
|
|
suite.keeper.SetParams(suite.ctx, params)
|
|
|
|
actualOracles := suite.keeper.GetAuthorizedAddresses(suite.ctx)
|
|
|
|
suite.Require().ElementsMatch(oracles, actualOracles)
|
|
}
|
|
|
|
func TestKeeperTestSuite(t *testing.T) {
|
|
suite.Run(t, new(KeeperTestSuite))
|
|
}
|