luciano/kinesis-consumer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix tests that don't pass context

Browse source
This commit is contained in:
Harlow Ward 2018-12-28 21:19:39 -08:00
parent 1a14621969
commit 271f32670b
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
consumer.go
View file

@ -12,7 +12,7 @@ import (
"github.com/aws/aws-sdk-go/service/kinesis" "github.com/aws/aws-sdk-go/service/kinesis"
"github.com/aws/aws-sdk-go/service/kinesis/kinesisiface" "github.com/aws/aws-sdk-go/service/kinesis/kinesisiface"
"github.com/opentracing/opentracing-go" opentracing "github.com/opentracing/opentracing-go"
"github.com/opentracing/opentracing-go/ext" "github.com/opentracing/opentracing-go/ext"
) )

10
consumer_test.go
View file

@ -69,7 +69,7 @@ func TestConsumer_Scan(t *testing.T) {
var resultData string var resultData string
var fnCallCounter int var fnCallCounter int
var fn = func(r *Record) ScanStatus { var fn = func(ctx context.Context, r *Record) ScanStatus {
fnCallCounter++ fnCallCounter++
resultData += string(r.Data) resultData += string(r.Data)
return ScanStatus{} return ScanStatus{}
@ -121,7 +121,7 @@ func TestConsumer_Scan_NoShardsAvailable(t *testing.T) {
} }
var fnCallCounter int var fnCallCounter int
var fn = func(r *Record) ScanStatus { var fn = func(ctx context.Context, r *Record) ScanStatus {
fnCallCounter++ fnCallCounter++
return ScanStatus{} return ScanStatus{}
} }
@ -185,7 +185,7 @@ func TestScanShard(t *testing.T) {
// callback fn appends record data // callback fn appends record data
var resultData string var resultData string
var fn = func(r *Record) ScanStatus { var fn = func(ctx context.Context, r *Record) ScanStatus {
resultData += string(r.Data) resultData += string(r.Data)
return ScanStatus{} return ScanStatus{}
} }
@ -246,7 +246,7 @@ func TestScanShard_StopScan(t *testing.T) {
// callback fn appends record data // callback fn appends record data
var resultData string var resultData string
var fn = func(r *Record) ScanStatus { var fn = func(ctx context.Context, r *Record) ScanStatus {
resultData += string(r.Data) resultData += string(r.Data)
return ScanStatus{StopScan: true} return ScanStatus{StopScan: true}
} }
@ -281,7 +281,7 @@ func TestScanShard_ShardIsClosed(t *testing.T) {
t.Fatalf("new consumer error: %v", err) t.Fatalf("new consumer error: %v", err)
} }
var fn = func(r *Record) ScanStatus { var fn = func(ctx context.Context, r *Record) ScanStatus {
return ScanStatus{} return ScanStatus{}
} }