mirror of
https://github.com/tendermint/tendermint.git
synced 2026-09-26 09:54:19 +00:00
Anton KaliaevandGitHub
1905ef7ca8
* lite2: advance to latest header without any exponential steps rename autoUpdate to autoUpdateRoutine * lite2: wait in Cleanup until goroutines finished running
48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package mock
|
|
|
|
import (
|
|
"github.com/tendermint/tendermint/lite2/provider"
|
|
"github.com/tendermint/tendermint/types"
|
|
)
|
|
|
|
// mock provider allows to directly set headers & vals, which can be handy when
|
|
// testing.
|
|
type mock struct {
|
|
chainID string
|
|
headers map[int64]*types.SignedHeader
|
|
vals map[int64]*types.ValidatorSet
|
|
}
|
|
|
|
// New creates a mock provider.
|
|
func New(chainID string, headers map[int64]*types.SignedHeader, vals map[int64]*types.ValidatorSet) provider.Provider {
|
|
return &mock{
|
|
chainID: chainID,
|
|
headers: headers,
|
|
vals: vals,
|
|
}
|
|
}
|
|
|
|
func (p *mock) ChainID() string {
|
|
return p.chainID
|
|
}
|
|
|
|
func (p *mock) SignedHeader(height int64) (*types.SignedHeader, error) {
|
|
if height == 0 && len(p.headers) > 0 {
|
|
return p.headers[int64(len(p.headers))], nil
|
|
}
|
|
if _, ok := p.headers[height]; ok {
|
|
return p.headers[height], nil
|
|
}
|
|
return nil, provider.ErrSignedHeaderNotFound
|
|
}
|
|
|
|
func (p *mock) ValidatorSet(height int64) (*types.ValidatorSet, error) {
|
|
if height == 0 && len(p.vals) > 0 {
|
|
return p.vals[int64(len(p.vals))], nil
|
|
}
|
|
if _, ok := p.vals[height]; ok {
|
|
return p.vals[height], nil
|
|
}
|
|
return nil, provider.ErrValidatorSetNotFound
|
|
}
|