105 lines
3.4 KiB
Go
105 lines
3.4 KiB
Go
package cool
|
|
|
|
import (
|
|
"testing"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
bank "github.com/cosmos/cosmos-sdk/x/bank"
|
|
"github.com/cosmos/cosmos-sdk/x/mock"
|
|
"github.com/stretchr/testify/require"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/crypto"
|
|
)
|
|
|
|
var (
|
|
priv1 = crypto.GenPrivKeyEd25519()
|
|
pubKey = priv1.PubKey()
|
|
addr1 = pubKey.Address()
|
|
|
|
quizMsg1 = MsgQuiz{
|
|
Sender: addr1,
|
|
CoolAnswer: "icecold",
|
|
}
|
|
|
|
quizMsg2 = MsgQuiz{
|
|
Sender: addr1,
|
|
CoolAnswer: "badvibesonly",
|
|
}
|
|
|
|
setTrendMsg1 = MsgSetTrend{
|
|
Sender: addr1,
|
|
Cool: "icecold",
|
|
}
|
|
|
|
setTrendMsg2 = MsgSetTrend{
|
|
Sender: addr1,
|
|
Cool: "badvibesonly",
|
|
}
|
|
|
|
setTrendMsg3 = MsgSetTrend{
|
|
Sender: addr1,
|
|
Cool: "warmandkind",
|
|
}
|
|
)
|
|
|
|
// initialize the mock application for this module
|
|
func getMockApp(t *testing.T) *mock.App {
|
|
mapp := mock.NewApp()
|
|
|
|
RegisterWire(mapp.Cdc)
|
|
keyCool := sdk.NewKVStoreKey("cool")
|
|
coinKeeper := bank.NewKeeper(mapp.AccountMapper)
|
|
keeper := NewKeeper(keyCool, coinKeeper, mapp.RegisterCodespace(DefaultCodespace))
|
|
mapp.Router().AddRoute("cool", NewHandler(keeper))
|
|
|
|
mapp.SetInitChainer(getInitChainer(mapp, keeper, "ice-cold"))
|
|
|
|
require.NoError(t, mapp.CompleteSetup([]*sdk.KVStoreKey{keyCool}))
|
|
return mapp
|
|
}
|
|
|
|
// overwrite the mock init chainer
|
|
func getInitChainer(mapp *mock.App, keeper Keeper, newTrend string) sdk.InitChainer {
|
|
return func(ctx sdk.Context, req abci.RequestInitChain) abci.ResponseInitChain {
|
|
mapp.InitChainer(ctx, req)
|
|
keeper.setTrend(ctx, newTrend)
|
|
|
|
return abci.ResponseInitChain{}
|
|
}
|
|
}
|
|
|
|
func TestMsgQuiz(t *testing.T) {
|
|
mapp := getMockApp(t)
|
|
|
|
// Construct genesis state
|
|
acc1 := &auth.BaseAccount{
|
|
Address: addr1,
|
|
Coins: nil,
|
|
}
|
|
accs := []auth.Account{acc1}
|
|
|
|
// Initialize the chain (nil)
|
|
mock.SetGenesis(mapp, accs)
|
|
|
|
// A checkTx context (true)
|
|
ctxCheck := mapp.BaseApp.NewContext(true, abci.Header{})
|
|
res1 := mapp.AccountMapper.GetAccount(ctxCheck, addr1)
|
|
require.Equal(t, acc1, res1)
|
|
|
|
// Set the trend, submit a really cool quiz and check for reward
|
|
mock.SignCheckDeliver(t, mapp.BaseApp, []sdk.Msg{setTrendMsg1}, []int64{0}, []int64{0}, true, priv1)
|
|
mock.SignCheckDeliver(t, mapp.BaseApp, []sdk.Msg{quizMsg1}, []int64{0}, []int64{1}, true, priv1)
|
|
mock.CheckBalance(t, mapp, addr1, sdk.Coins{{"icecold", sdk.NewInt(69)}})
|
|
mock.SignCheckDeliver(t, mapp.BaseApp, []sdk.Msg{quizMsg2}, []int64{0}, []int64{2}, false, priv1) // result without reward
|
|
mock.CheckBalance(t, mapp, addr1, sdk.Coins{{"icecold", sdk.NewInt(69)}})
|
|
mock.SignCheckDeliver(t, mapp.BaseApp, []sdk.Msg{quizMsg1}, []int64{0}, []int64{3}, true, priv1)
|
|
mock.CheckBalance(t, mapp, addr1, sdk.Coins{{"icecold", sdk.NewInt(138)}})
|
|
mock.SignCheckDeliver(t, mapp.BaseApp, []sdk.Msg{setTrendMsg2}, []int64{0}, []int64{4}, true, priv1) // reset the trend
|
|
mock.SignCheckDeliver(t, mapp.BaseApp, []sdk.Msg{quizMsg1}, []int64{0}, []int64{5}, false, priv1) // the same answer will nolonger do!
|
|
mock.CheckBalance(t, mapp, addr1, sdk.Coins{{"icecold", sdk.NewInt(138)}})
|
|
mock.SignCheckDeliver(t, mapp.BaseApp, []sdk.Msg{quizMsg2}, []int64{0}, []int64{6}, true, priv1) // earlier answer now relevant again
|
|
mock.CheckBalance(t, mapp, addr1, sdk.Coins{{"badvibesonly", sdk.NewInt(69)}, {"icecold", sdk.NewInt(138)}})
|
|
mock.SignCheckDeliver(t, mapp.BaseApp, []sdk.Msg{setTrendMsg3}, []int64{0}, []int64{7}, false, priv1) // expect to fail to set the trend to something which is not cool
|
|
}
|