tidy test genesis states

This commit is contained in:
rhuairahrighairigh 2019-12-07 14:56:40 +00:00
parent 11fd42685b
commit d8347fd49c
4 changed files with 24 additions and 30 deletions

View File

@ -17,15 +17,13 @@ func d(str string) sdk.Dec { return sdk.MustNewDecFromStr(str)
func c(denom string, amount int64) sdk.Coin { return sdk.NewInt64Coin(denom, amount) } func c(denom string, amount int64) sdk.Coin { return sdk.NewInt64Coin(denom, amount) }
func cs(coins ...sdk.Coin) sdk.Coins { return sdk.NewCoins(coins...) } func cs(coins ...sdk.Coin) sdk.Coins { return sdk.NewCoins(coins...) }
func NewPFGenState(asset string, price sdk.Dec) app.GenesisState { func NewPricefeedGenState(asset string, price sdk.Dec) app.GenesisState {
quote := "usd"
ap := pricefeed.Params{
Markets: []pricefeed.Market{
pricefeed.Market{MarketID: asset, BaseAsset: asset, QuoteAsset: quote, Oracles: pricefeed.Oracles{}, Active: true},
},
}
pfGenesis := pricefeed.GenesisState{ pfGenesis := pricefeed.GenesisState{
Params: ap, Params: pricefeed.Params{
Markets: []pricefeed.Market{
pricefeed.Market{MarketID: asset, BaseAsset: asset, QuoteAsset: "usd", Oracles: pricefeed.Oracles{}, Active: true},
},
},
PostedPrices: []pricefeed.PostedPrice{ PostedPrices: []pricefeed.PostedPrice{
pricefeed.PostedPrice{ pricefeed.PostedPrice{
MarketID: asset, MarketID: asset,
@ -56,16 +54,14 @@ func NewCDPGenState(asset string, liquidationRatio sdk.Dec) app.GenesisState {
return app.GenesisState{cdp.ModuleName: cdp.ModuleCdc.MustMarshalJSON(cdpGenesis)} return app.GenesisState{cdp.ModuleName: cdp.ModuleCdc.MustMarshalJSON(cdpGenesis)}
} }
func NewPFGenStateMulti() app.GenesisState { func NewPricefeedGenStateMulti() app.GenesisState {
quote := "usd"
ap := pricefeed.Params{
Markets: []pricefeed.Market{
pricefeed.Market{MarketID: "btc", BaseAsset: "btc", QuoteAsset: quote, Oracles: pricefeed.Oracles{}, Active: true},
pricefeed.Market{MarketID: "xrp", BaseAsset: "xrp", QuoteAsset: quote, Oracles: pricefeed.Oracles{}, Active: true},
},
}
pfGenesis := pricefeed.GenesisState{ pfGenesis := pricefeed.GenesisState{
Params: ap, Params: pricefeed.Params{
Markets: []pricefeed.Market{
pricefeed.Market{MarketID: "btc", BaseAsset: "btc", QuoteAsset: "usd", Oracles: pricefeed.Oracles{}, Active: true},
pricefeed.Market{MarketID: "xrp", BaseAsset: "xrp", QuoteAsset: "usd", Oracles: pricefeed.Oracles{}, Active: true},
},
},
PostedPrices: []pricefeed.PostedPrice{ PostedPrices: []pricefeed.PostedPrice{
pricefeed.PostedPrice{ pricefeed.PostedPrice{
MarketID: "btc", MarketID: "btc",

View File

@ -98,7 +98,7 @@ func TestKeeper_ModifyCDP(t *testing.T) {
// initialize cdp owner account with coins, and collateral with price and params // initialize cdp owner account with coins, and collateral with price and params
tApp.InitializeFromGenesisStates( tApp.InitializeFromGenesisStates(
app.NewAuthGenState([]sdk.AccAddress{ownerAddr}, []sdk.Coins{tc.priorState.OwnerCoins}), app.NewAuthGenState([]sdk.AccAddress{ownerAddr}, []sdk.Coins{tc.priorState.OwnerCoins}),
NewPFGenState("xrp", d(tc.price)), NewPricefeedGenState("xrp", d(tc.price)),
NewCDPGenState("xrp", d("2.0")), NewCDPGenState("xrp", d("2.0")),
) )
// create a context for db access // create a context for db access
@ -151,7 +151,7 @@ func TestKeeper_PartialSeizeCDP(t *testing.T) {
tApp := app.NewTestApp() tApp := app.NewTestApp()
tApp.InitializeFromGenesisStates( tApp.InitializeFromGenesisStates(
app.NewAuthGenState(addrs, []sdk.Coins{cs(c(collateral, 100))}), app.NewAuthGenState(addrs, []sdk.Coins{cs(c(collateral, 100))}),
NewPFGenState(collateral, d("1.00")), NewPricefeedGenState(collateral, d("1.00")),
NewCDPGenState(collateral, d("2.00")), NewCDPGenState(collateral, d("2.00")),
) )
@ -182,7 +182,7 @@ func TestKeeper_PartialSeizeCDP(t *testing.T) {
func TestKeeper_GetCDPs(t *testing.T) { func TestKeeper_GetCDPs(t *testing.T) {
// setup test app // setup test app
tApp := app.NewTestApp().InitializeFromGenesisStates( tApp := app.NewTestApp().InitializeFromGenesisStates(
NewPFGenStateMulti(), // collateral needs to be in pricefeed for cdp InitGenesis to validate NewPricefeedGenStateMulti(), // collateral needs to be in pricefeed for cdp InitGenesis to validate
NewCDPGenStateMulti(), NewCDPGenStateMulti(),
) )
ctx := tApp.NewContext(true, abci.Header{}) ctx := tApp.NewContext(true, abci.Header{})

View File

@ -18,16 +18,14 @@ func d(str string) sdk.Dec { return sdk.MustNewDecFromStr(str)
func c(denom string, amount int64) sdk.Coin { return sdk.NewInt64Coin(denom, amount) } func c(denom string, amount int64) sdk.Coin { return sdk.NewInt64Coin(denom, amount) }
func cs(coins ...sdk.Coin) sdk.Coins { return sdk.NewCoins(coins...) } func cs(coins ...sdk.Coin) sdk.Coins { return sdk.NewCoins(coins...) }
// Default genesis states to initialize test apps // Genesis states to initialize test apps
func NewPFGenState(asset string, price sdk.Dec) app.GenesisState { func NewPricefeedGenState(asset string, price sdk.Dec) app.GenesisState {
quote := "usd"
ap := pricefeed.Params{
Markets: []pricefeed.Market{
pricefeed.Market{MarketID: asset, BaseAsset: asset, QuoteAsset: quote, Oracles: pricefeed.Oracles{}, Active: true}},
}
pfGenesis := pricefeed.GenesisState{ pfGenesis := pricefeed.GenesisState{
Params: ap, Params: pricefeed.Params{
Markets: []pricefeed.Market{
pricefeed.Market{MarketID: asset, BaseAsset: asset, QuoteAsset: "usd", Oracles: pricefeed.Oracles{}, Active: true}},
},
PostedPrices: []pricefeed.PostedPrice{ PostedPrices: []pricefeed.PostedPrice{
pricefeed.PostedPrice{ pricefeed.PostedPrice{
MarketID: asset, MarketID: asset,

View File

@ -20,7 +20,7 @@ func TestKeeper_SeizeAndStartCollateralAuction(t *testing.T) {
tApp := app.NewTestApp() tApp := app.NewTestApp()
tApp.InitializeFromGenesisStates( tApp.InitializeFromGenesisStates(
app.NewAuthGenState(addrs, []sdk.Coins{cs(c("btc", 100))}), app.NewAuthGenState(addrs, []sdk.Coins{cs(c("btc", 100))}),
NewPFGenState("btc", d("8000.00")), NewPricefeedGenState("btc", d("8000.00")),
NewCDPGenState(), NewCDPGenState(),
NewLiquidatorGenState(), NewLiquidatorGenState(),
) )
@ -81,7 +81,7 @@ func TestKeeper_partialSeizeCDP(t *testing.T) {
tApp := app.NewTestApp() tApp := app.NewTestApp()
tApp.InitializeFromGenesisStates( tApp.InitializeFromGenesisStates(
app.NewAuthGenState(addrs, []sdk.Coins{cs(c("btc", 100))}), app.NewAuthGenState(addrs, []sdk.Coins{cs(c("btc", 100))}),
NewPFGenState("btc", d("8000.00")), NewPricefeedGenState("btc", d("8000.00")),
NewCDPGenState(), NewCDPGenState(),
NewLiquidatorGenState(), NewLiquidatorGenState(),
) )