package mock import ( "context" "errors" "fmt" "strings" "github.com/tendermint/tendermint/light/provider" "github.com/tendermint/tendermint/types" ) type Mock struct { id string headers map[int64]*types.SignedHeader vals map[int64]*types.ValidatorSet evidenceToReport map[string]types.Evidence // hash => evidence } var _ provider.Provider = (*Mock)(nil) // New creates a mock provider with the given set of headers and validator // sets. func New(id string, headers map[int64]*types.SignedHeader, vals map[int64]*types.ValidatorSet) *Mock { return &Mock{ id: id, headers: headers, vals: vals, evidenceToReport: make(map[string]types.Evidence), } } func (p *Mock) String() string { var headers strings.Builder for _, h := range p.headers { fmt.Fprintf(&headers, " %d:%X", h.Height, h.Hash()) } var vals strings.Builder for _, v := range p.vals { fmt.Fprintf(&vals, " %X", v.Hash()) } return fmt.Sprintf("Mock{id: %s, headers: %s, vals: %v}", p.id, headers.String(), vals.String()) } func (p *Mock) LightBlock(_ context.Context, height int64) (*types.LightBlock, error) { var lb *types.LightBlock if height == 0 && len(p.headers) > 0 { sh := p.headers[int64(len(p.headers))] vals := p.vals[int64(len(p.vals))] lb = &types.LightBlock{ SignedHeader: sh, ValidatorSet: vals, } } if _, ok := p.headers[height]; ok { sh := p.headers[height] vals := p.vals[height] lb = &types.LightBlock{ SignedHeader: sh, ValidatorSet: vals, } } if lb == nil { return nil, provider.ErrLightBlockNotFound } if lb.SignedHeader == nil || lb.ValidatorSet == nil { return nil, provider.ErrBadLightBlock{Reason: errors.New("nil header or vals")} } if err := lb.ValidateBasic(lb.ChainID); err != nil { return nil, provider.ErrBadLightBlock{Reason: err} } return lb, nil } func (p *Mock) ReportEvidence(_ context.Context, ev types.Evidence) error { p.evidenceToReport[string(ev.Hash())] = ev return nil } func (p *Mock) HasEvidence(ev types.Evidence) bool { _, ok := p.evidenceToReport[string(ev.Hash())] return ok }