From 271f32670b40e21272537e35a6ca778c3f8a50d0 Mon Sep 17 00:00:00 2001
From: Harlow Ward <harlow@hward.com>
Date: Fri, 28 Dec 2018 21:19:39 -0800
Subject: [PATCH] Fix tests that don't pass context

---
 consumer.go      |  2 +-
 consumer_test.go | 10 +++++-----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/consumer.go b/consumer.go
index 44e690d..e65a08d 100644
--- a/consumer.go
+++ b/consumer.go
@@ -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"
 )
 
diff --git a/consumer_test.go b/consumer_test.go
index 5eb8a9e..0765dc1 100644
--- a/consumer_test.go
+++ b/consumer_test.go
@@ -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{}
 	}