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