Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: keeper state api #24

Merged
merged 11 commits into from
Nov 15, 2023
23 changes: 21 additions & 2 deletions x/feemarket/keeper/keeper.go
Original file line number Diff line number Diff line change
Expand Up @@ -45,12 +45,31 @@ func (k *Keeper) GetAuthority() string {

// GetState returns the feemarket module's state.
func (k *Keeper) GetState(ctx sdk.Context) (types.State, error) {
panic("TODO")
store := ctx.KVStore(k.storeKey)

key := types.KeyState
bz := store.Get(key)

state := types.State{}
if err := state.Unmarshal(bz); err != nil {
return types.State{}, err
}

return state, nil
}

// SetState sets the feemarket module's state.
func (k *Keeper) SetState(ctx sdk.Context, state types.State) error {
panic("TODO")
store := ctx.KVStore(k.storeKey)

bz, err := state.Marshal()
if err != nil {
return err
}

store.Set(types.KeyState, bz)

return nil
}

// GetParams returns the feemarket module's parameters.
Expand Down
33 changes: 33 additions & 0 deletions x/feemarket/keeper/keeper_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,36 @@ func (s *KeeperTestSuite) SetupTest() {
s.msgServer = keeper.NewMsgServer(*s.feemarketKeeper)
s.queryServer = keeper.NewQueryServer(*s.feemarketKeeper)
}

func (s *KeeperTestSuite) TestState() {
s.Run("get state with no state set", func() {
gotState, err := s.feemarketKeeper.GetState(s.ctx)
s.Require().NoError(err)

s.Require().Equal(types.State{}, gotState)
})

s.Run("set and get default eip1559 state", func() {
state := types.DefaultState()

err := s.feemarketKeeper.SetState(s.ctx, state)
s.Require().NoError(err)

gotState, err := s.feemarketKeeper.GetState(s.ctx)
s.Require().NoError(err)

s.Require().EqualValues(state, gotState)
})

s.Run("set and get aimd eip1559 state", func() {
state := types.DefaultAIMDState()

err := s.feemarketKeeper.SetState(s.ctx, state)
s.Require().NoError(err)

gotState, err := s.feemarketKeeper.GetState(s.ctx)
s.Require().NoError(err)

s.Require().Equal(state, gotState)
})
}
Loading