trie

type CheckpointHashesHolderStub struct {
	PutCalled             func([]byte, common.ModifiedHashes) bool
	RemoveCommittedCalled func([]byte)
	RemoveCalled          func([]byte)
	ShouldCommitCalled    func([]byte) bool
}

type DataTrieTrackerStub struct {
	RetrieveValueCalled func(key []byte) ([]byte, uint32, error)
	SaveKeyValueCalled  func(key []byte, value []byte) error
	SetDataTrieCalled   func(tr common.Trie)
	DataTrieCalled      func() common.Trie
	SaveDirtyDataCalled func(trie common.Trie) (map[string][]byte, error)
}

type MockStatistics struct {
	WaitForSnapshotsToFinishCalled func()
}

type SnapshotPruningStorerStub struct {
	*testscommon.MemDbMock
	GetFromOldEpochsWithoutAddingToCacheCalled func(key []byte) ([]byte, core.OptionalUint32, error)
	GetFromLastEpochCalled                     func(key []byte) ([]byte, error)
	GetFromCurrentEpochCalled                  func(key []byte) ([]byte, error)
	GetFromEpochCalled                         func(key []byte, epoch uint32) ([]byte, error)
	PutInEpochCalled                           func(key []byte, data []byte, epoch uint32) error
	PutInEpochWithoutCacheCalled               func(key []byte, data []byte, epoch uint32) error
	GetLatestStorageEpochCalled                func() (uint32, error)
	RemoveFromCurrentEpochCalled               func(key []byte) error
}

type TrieStub struct {
	GetCalled                   func(key []byte) ([]byte, uint32, error)
	UpdateCalled                func(key, value []byte) error
	DeleteCalled                func(key []byte) error
	RootCalled                  func() ([]byte, error)
	CommitCalled                func() error
	RecreateCalled              func(root []byte) (common.Trie, error)
	RecreateFromEpochCalled     func(options common.RootHashHolder) (common.Trie, error)
	GetObsoleteHashesCalled     func() [][]byte
	AppendToOldHashesCalled     func([][]byte)
	GetSerializedNodesCalled    func([]byte, uint64) ([][]byte, uint64, error)
	GetAllHashesCalled          func() ([][]byte, error)
	GetAllLeavesOnChannelCalled func(leavesChannels *common.TrieIteratorChannels, ctx context.Context, rootHash []byte, keyBuilder common.KeyBuilder) error
	GetProofCalled              func(key []byte) ([][]byte, []byte, error)
	VerifyProofCalled           func(rootHash []byte, key []byte, proof [][]byte) (bool, error)
	GetStorageManagerCalled     func() common.StorageManager
	GetSerializedNodeCalled     func(bytes []byte) ([]byte, error)
	GetNumNodesCalled           func() common.NumNodesDTO
	GetOldRootCalled            func() []byte
	CloseCalled                 func() error
}

type TriesHolderStub struct {
	PutCalled    func([]byte, common.Trie)
	RemoveCalled func([]byte, common.Trie)
	GetCalled    func([]byte) common.Trie
	GetAllCalled func() []common.Trie
	ResetCalled  func()
}