mirror of https://github.com/certusone/wasmd.git
27 lines
571 B
Go
27 lines
571 B
Go
package wasmtesting
|
|
|
|
import (
|
|
storetypes "github.com/cosmos/cosmos-sdk/store/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
// MockCommitMultiStore mock with a CacheMultiStore to capture commits
|
|
type MockCommitMultiStore struct {
|
|
sdk.CommitMultiStore
|
|
Committed []bool
|
|
}
|
|
|
|
func (m *MockCommitMultiStore) CacheMultiStore() storetypes.CacheMultiStore {
|
|
m.Committed = append(m.Committed, false)
|
|
return &mockCMS{m, &m.Committed[len(m.Committed)-1]}
|
|
}
|
|
|
|
type mockCMS struct {
|
|
sdk.CommitMultiStore
|
|
committed *bool
|
|
}
|
|
|
|
func (m *mockCMS) Write() {
|
|
*m.committed = true
|
|
}
|