mirror of
https://github.com/grassrootseconomics/cic-chain-events.git
synced 2024-11-22 23:56:46 +01:00
139 lines
3.2 KiB
Go
139 lines
3.2 KiB
Go
package pipeline
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/grassrootseconomics/cic-chain-events/internal/fetch"
|
|
"github.com/grassrootseconomics/cic-chain-events/internal/filter"
|
|
"github.com/stretchr/testify/suite"
|
|
"github.com/zerodha/logf"
|
|
)
|
|
|
|
var (
|
|
graphqlEndpoint = os.Getenv("TEST_GRAPHQL_ENDPOINT")
|
|
)
|
|
|
|
type itPipelineTest struct {
|
|
suite.Suite
|
|
errorPipeline *Pipeline
|
|
normalPipeline *Pipeline
|
|
earlyExitPipeline *Pipeline
|
|
}
|
|
|
|
func TestPipelineSuite(t *testing.T) {
|
|
suite.Run(t, new(itPipelineTest))
|
|
}
|
|
|
|
type errorFilter struct{}
|
|
|
|
func newErrorFilter() filter.Filter {
|
|
return &errorFilter{}
|
|
}
|
|
|
|
func (*errorFilter) Execute(_ context.Context, _ fetch.Transaction) (bool, error) {
|
|
return false, errors.New("crash")
|
|
}
|
|
|
|
type earlyExitFilter struct{}
|
|
|
|
func newEarlyExitFilter() filter.Filter {
|
|
return &earlyExitFilter{}
|
|
}
|
|
|
|
func (*earlyExitFilter) Execute(_ context.Context, _ fetch.Transaction) (bool, error) {
|
|
return false, nil
|
|
}
|
|
|
|
func (s *itPipelineTest) SetupSuite() {
|
|
logger := logf.New(
|
|
logf.Opts{
|
|
Level: logf.DebugLevel,
|
|
},
|
|
)
|
|
|
|
fetcher := fetch.NewGraphqlFetcher(fetch.GraphqlOpts{
|
|
GraphqlEndpoint: graphqlEndpoint,
|
|
})
|
|
|
|
noopFilter := filter.NewNoopFilter(filter.NoopFilterOpts{
|
|
Logg: logger,
|
|
})
|
|
errFilter := newErrorFilter()
|
|
earlyFilter := newEarlyExitFilter()
|
|
|
|
s.errorPipeline = NewPipeline(PipelineOpts{
|
|
Filters: []filter.Filter{
|
|
noopFilter,
|
|
errFilter,
|
|
},
|
|
BlockFetcher: fetcher,
|
|
Logg: logger,
|
|
})
|
|
|
|
s.normalPipeline = NewPipeline(PipelineOpts{
|
|
Filters: []filter.Filter{
|
|
noopFilter,
|
|
},
|
|
BlockFetcher: fetcher,
|
|
Logg: logger,
|
|
})
|
|
|
|
s.earlyExitPipeline = NewPipeline(PipelineOpts{
|
|
Filters: []filter.Filter{
|
|
noopFilter,
|
|
earlyFilter,
|
|
errFilter,
|
|
},
|
|
BlockFetcher: fetcher,
|
|
Logg: logger,
|
|
})
|
|
}
|
|
|
|
func (s *itPipelineTest) Test_E2E_Pipeline_Run_On_Existing_Block_No_Err() {
|
|
err := s.normalPipeline.Run(context.Background(), 14974600)
|
|
s.NoError(err)
|
|
}
|
|
|
|
func (s *itPipelineTest) Test_E2E_Pipeline_Run_On_Non_Existing_Block_No_Err() {
|
|
err := s.normalPipeline.Run(context.Background(), 14974600000)
|
|
s.Error(err)
|
|
}
|
|
|
|
func (s *itPipelineTest) Test_E2E_Pipeline_Run_On_Existing_Block_Early() {
|
|
err := s.earlyExitPipeline.Run(context.Background(), 14974600)
|
|
s.NoError(err)
|
|
}
|
|
|
|
func (s *itPipelineTest) Test_E2E_Pipeline_Run_On_Existing_Block_With_Err() {
|
|
err := s.errorPipeline.Run(context.Background(), 14974600)
|
|
s.Error(err)
|
|
}
|
|
|
|
func (s *itPipelineTest) Test_E2E_Pipeline_Run_On_Non_Existing_Block_With_Err() {
|
|
err := s.errorPipeline.Run(context.Background(), 14974600000)
|
|
s.Error(err)
|
|
}
|
|
|
|
func (s *itPipelineTest) Test_E2E_Pipeline_Run_On_Non_Existing_Block_Early() {
|
|
err := s.earlyExitPipeline.Run(context.Background(), 14974600000)
|
|
s.Error(err)
|
|
}
|
|
|
|
func (s *itPipelineTest) Test_E2E_Pipeline_Run_On_Empty_Block_With_No_Err() {
|
|
err := s.normalPipeline.Run(context.Background(), 15370320)
|
|
s.NoError(err)
|
|
}
|
|
|
|
func (s *itPipelineTest) Test_E2E_Pipeline_Run_On_Empty_Block_With_Err() {
|
|
err := s.errorPipeline.Run(context.Background(), 15370320)
|
|
s.NoError(err)
|
|
}
|
|
|
|
func (s *itPipelineTest) Test_E2E_Pipeline_Run_On_Empty_Block_Early() {
|
|
err := s.earlyExitPipeline.Run(context.Background(), 15370320)
|
|
s.NoError(err)
|
|
}
|