-
Notifications
You must be signed in to change notification settings - Fork 558
/
sqs_ingester_mock.go
43 lines (36 loc) · 1.54 KB
/
sqs_ingester_mock.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package mocks
import (
"github.com/cosmos/cosmos-sdk/types"
"github.com/osmosis-labs/osmosis/v23/ingest/sqs/domain"
)
var _ domain.Ingester = &SQSIngesterMock{}
// SQSIngesterMock is a mock implementation of domain.Ingester.
type SQSIngesterMock struct {
// AllBlockDataError is the error to return when ProcessAllBlockData is called.
AllBlockDataError error
// ChangedBlockDataError is the error to return when ProcessChangedBlockData is called.
ChangedBlockDataError error
// IsProcessAllBlockDataCalled is a flag indicating if ProcessAllBlockData was called.
IsProcessAllBlockDataCalled bool
// IsProcessAllChangedDataCalled is a flag indicating if ProcessChangedBlockData was called.
IsProcessAllChangedDataCalled bool
// LastChangedPoolsObserved is the last changed pools observed by the mock when
// ProcessChangedBlockData is called.
LastChangedPoolsObserved domain.BlockPools
// If this is non-empty, ProcessAllBlockData(...) will panic with this message.
ProcessAllBlockDataPanicMsg string
}
// ProcessAllBlockData implements domain.Ingester.
func (s *SQSIngesterMock) ProcessAllBlockData(ctx types.Context) error {
if s.ProcessAllBlockDataPanicMsg != "" {
panic(s.ProcessAllBlockDataPanicMsg)
}
s.IsProcessAllBlockDataCalled = true
return s.AllBlockDataError
}
// ProcessChangedBlockData implements domain.Ingester.
func (s *SQSIngesterMock) ProcessChangedBlockData(ctx types.Context, changedPools domain.BlockPools) error {
s.IsProcessAllChangedDataCalled = true
s.LastChangedPoolsObserved = changedPools
return s.ChangedBlockDataError
}