Fix types for new WithContext funcs since sqs and sns arent the same …

…thing
nytimes · Mar 31, 2017 · f341875 · f341875
1 parent 3990588
commit f341875
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/pubsub/aws/awspub_test.go b/pubsub/aws/awspub_test.go
@@ -113,7 +113,7 @@ func (t *TestSNSAPI) CheckIfPhoneNumberIsOptedOutRequest(*sns.CheckIfPhoneNumber
 func (t *TestSNSAPI) CheckIfPhoneNumberIsOptedOut(*sns.CheckIfPhoneNumberIsOptedOutInput) (*sns.CheckIfPhoneNumberIsOptedOutOutput, error) {
 	return nil, nil
 }
-func (t *TestSNSAPI) CheckIfPhoneNumberIsOptedOutWithContext(aws.Context, *sns.CheckIfPhoneNumberIsOptedOutInput) (*sns.CheckIfPhoneNumberIsOptedOutOutput, error) {
+func (t *TestSNSAPI) CheckIfPhoneNumberIsOptedOutWithContext(aws.Context, *sns.CheckIfPhoneNumberIsOptedOutInput, ...request.Option) (*sns.CheckIfPhoneNumberIsOptedOutOutput, error) {
 	return nil, nil
 }
 

diff --git a/pubsub/aws/awssub_test.go b/pubsub/aws/awssub_test.go
@@ -11,7 +11,6 @@ import (
 	"github.com/NYTimes/gizmo/pubsub"
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/request"
-	"github.com/aws/aws-sdk-go/service/sns"
 	"github.com/aws/aws-sdk-go/service/sqs"
 	"github.com/aws/aws-sdk-go/service/sqs/sqsiface"
 	"github.com/golang/protobuf/proto"
@@ -370,7 +369,7 @@ func (s *TestSQSAPI) AddPermissionRequest(*sqs.AddPermissionInput) (*request.Req
 func (s *TestSQSAPI) AddPermission(*sqs.AddPermissionInput) (*sqs.AddPermissionOutput, error) {
 	return nil, errNotImpl
 }
-func (t *TestSNSAPI) AddPermissionWithContext(aws.Context, *sns.AddPermissionInput, ...request.Option) (*sns.AddPermissionOutput, error) {
+func (s *TestSNSAPI) AddPermissionWithContext(aws.Context, *sqs.AddPermissionInput, ...request.Option) (*sqs.AddPermissionOutput, error) {
 	return nil, errNotImpl
 }
 func (s *TestSQSAPI) ChangeMessageVisibilityRequest(*sqs.ChangeMessageVisibilityInput) (*request.Request, *sqs.ChangeMessageVisibilityOutput) {