Switch to using the dep tool and update all the dependencies

2017-05-11 15:39:54 +01:00
parent 5135ff73cb
commit 98c2d2c41b
5321 changed files with 4483197 additions and 5918 deletions
@@ -0,0 +1,208 @@
+package sqs_test
+
+import (
+	"bytes"
+	"io/ioutil"
+	"net/http"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/awserr"
+	"github.com/aws/aws-sdk-go/aws/request"
+	"github.com/aws/aws-sdk-go/awstesting/unit"
+	"github.com/aws/aws-sdk-go/service/sqs"
+)
+
+var svc = func() *sqs.SQS {
+	s := sqs.New(unit.Session, &aws.Config{
+		DisableParamValidation: aws.Bool(true),
+	})
+	s.Handlers.Send.Clear()
+	return s
+}()
+
+func TestSendMessageChecksum(t *testing.T) {
+	req, _ := svc.SendMessageRequest(&sqs.SendMessageInput{
+		MessageBody: aws.String("test"),
+	})
+	req.Handlers.Send.PushBack(func(r *request.Request) {
+		body := ioutil.NopCloser(bytes.NewReader([]byte("")))
+		r.HTTPResponse = &http.Response{StatusCode: 200, Body: body}
+		r.Data = &sqs.SendMessageOutput{
+			MD5OfMessageBody: aws.String("098f6bcd4621d373cade4e832627b4f6"),
+			MessageId:        aws.String("12345"),
+		}
+	})
+	err := req.Send()
+	assert.NoError(t, err)
+}
+
+func TestSendMessageChecksumInvalid(t *testing.T) {
+	req, _ := svc.SendMessageRequest(&sqs.SendMessageInput{
+		MessageBody: aws.String("test"),
+	})
+	req.Handlers.Send.PushBack(func(r *request.Request) {
+		body := ioutil.NopCloser(bytes.NewReader([]byte("")))
+		r.HTTPResponse = &http.Response{StatusCode: 200, Body: body}
+		r.Data = &sqs.SendMessageOutput{
+			MD5OfMessageBody: aws.String("000"),
+			MessageId:        aws.String("12345"),
+		}
+	})
+	err := req.Send()
+	assert.Error(t, err)
+
+	assert.Equal(t, "InvalidChecksum", err.(awserr.Error).Code())
+	assert.Contains(t, err.(awserr.Error).Message(), "expected MD5 checksum '000', got '098f6bcd4621d373cade4e832627b4f6'")
+}
+
+func TestSendMessageChecksumInvalidNoValidation(t *testing.T) {
+	s := sqs.New(unit.Session, &aws.Config{
+		DisableParamValidation:  aws.Bool(true),
+		DisableComputeChecksums: aws.Bool(true),
+	})
+	s.Handlers.Send.Clear()
+
+	req, _ := s.SendMessageRequest(&sqs.SendMessageInput{
+		MessageBody: aws.String("test"),
+	})
+	req.Handlers.Send.PushBack(func(r *request.Request) {
+		body := ioutil.NopCloser(bytes.NewReader([]byte("")))
+		r.HTTPResponse = &http.Response{StatusCode: 200, Body: body}
+		r.Data = &sqs.SendMessageOutput{
+			MD5OfMessageBody: aws.String("000"),
+			MessageId:        aws.String("12345"),
+		}
+	})
+	err := req.Send()
+	assert.NoError(t, err)
+}
+
+func TestSendMessageChecksumNoInput(t *testing.T) {
+	req, _ := svc.SendMessageRequest(&sqs.SendMessageInput{})
+	req.Handlers.Send.PushBack(func(r *request.Request) {
+		body := ioutil.NopCloser(bytes.NewReader([]byte("")))
+		r.HTTPResponse = &http.Response{StatusCode: 200, Body: body}
+		r.Data = &sqs.SendMessageOutput{}
+	})
+	err := req.Send()
+	assert.Error(t, err)
+
+	assert.Equal(t, "InvalidChecksum", err.(awserr.Error).Code())
+	assert.Contains(t, err.(awserr.Error).Message(), "cannot compute checksum. missing body")
+}
+
+func TestSendMessageChecksumNoOutput(t *testing.T) {
+	req, _ := svc.SendMessageRequest(&sqs.SendMessageInput{
+		MessageBody: aws.String("test"),
+	})
+	req.Handlers.Send.PushBack(func(r *request.Request) {
+		body := ioutil.NopCloser(bytes.NewReader([]byte("")))
+		r.HTTPResponse = &http.Response{StatusCode: 200, Body: body}
+		r.Data = &sqs.SendMessageOutput{}
+	})
+	err := req.Send()
+	assert.Error(t, err)
+
+	assert.Equal(t, "InvalidChecksum", err.(awserr.Error).Code())
+	assert.Contains(t, err.(awserr.Error).Message(), "cannot verify checksum. missing response MD5")
+}
+
+func TestRecieveMessageChecksum(t *testing.T) {
+	req, _ := svc.ReceiveMessageRequest(&sqs.ReceiveMessageInput{})
+	req.Handlers.Send.PushBack(func(r *request.Request) {
+		md5 := "098f6bcd4621d373cade4e832627b4f6"
+		body := ioutil.NopCloser(bytes.NewReader([]byte("")))
+		r.HTTPResponse = &http.Response{StatusCode: 200, Body: body}
+		r.Data = &sqs.ReceiveMessageOutput{
+			Messages: []*sqs.Message{
+				{Body: aws.String("test"), MD5OfBody: &md5},
+				{Body: aws.String("test"), MD5OfBody: &md5},
+				{Body: aws.String("test"), MD5OfBody: &md5},
+				{Body: aws.String("test"), MD5OfBody: &md5},
+			},
+		}
+	})
+	err := req.Send()
+	assert.NoError(t, err)
+}
+
+func TestRecieveMessageChecksumInvalid(t *testing.T) {
+	req, _ := svc.ReceiveMessageRequest(&sqs.ReceiveMessageInput{})
+	req.Handlers.Send.PushBack(func(r *request.Request) {
+		md5 := "098f6bcd4621d373cade4e832627b4f6"
+		body := ioutil.NopCloser(bytes.NewReader([]byte("")))
+		r.HTTPResponse = &http.Response{StatusCode: 200, Body: body}
+		r.Data = &sqs.ReceiveMessageOutput{
+			Messages: []*sqs.Message{
+				{Body: aws.String("test"), MD5OfBody: &md5},
+				{Body: aws.String("test"), MD5OfBody: aws.String("000"), MessageId: aws.String("123")},
+				{Body: aws.String("test"), MD5OfBody: aws.String("000"), MessageId: aws.String("456")},
+				{Body: aws.String("test"), MD5OfBody: aws.String("000")},
+				{Body: aws.String("test"), MD5OfBody: &md5},
+			},
+		}
+	})
+	err := req.Send()
+	assert.Error(t, err)
+
+	assert.Equal(t, "InvalidChecksum", err.(awserr.Error).Code())
+	assert.Contains(t, err.(awserr.Error).Message(), "invalid messages: 123, 456")
+}
+
+func TestSendMessageBatchChecksum(t *testing.T) {
+	req, _ := svc.SendMessageBatchRequest(&sqs.SendMessageBatchInput{
+		Entries: []*sqs.SendMessageBatchRequestEntry{
+			{Id: aws.String("1"), MessageBody: aws.String("test")},
+			{Id: aws.String("2"), MessageBody: aws.String("test")},
+			{Id: aws.String("3"), MessageBody: aws.String("test")},
+			{Id: aws.String("4"), MessageBody: aws.String("test")},
+		},
+	})
+	req.Handlers.Send.PushBack(func(r *request.Request) {
+		md5 := "098f6bcd4621d373cade4e832627b4f6"
+		body := ioutil.NopCloser(bytes.NewReader([]byte("")))
+		r.HTTPResponse = &http.Response{StatusCode: 200, Body: body}
+		r.Data = &sqs.SendMessageBatchOutput{
+			Successful: []*sqs.SendMessageBatchResultEntry{
+				{MD5OfMessageBody: &md5, MessageId: aws.String("123"), Id: aws.String("1")},
+				{MD5OfMessageBody: &md5, MessageId: aws.String("456"), Id: aws.String("2")},
+				{MD5OfMessageBody: &md5, MessageId: aws.String("789"), Id: aws.String("3")},
+				{MD5OfMessageBody: &md5, MessageId: aws.String("012"), Id: aws.String("4")},
+			},
+		}
+	})
+	err := req.Send()
+	assert.NoError(t, err)
+}
+
+func TestSendMessageBatchChecksumInvalid(t *testing.T) {
+	req, _ := svc.SendMessageBatchRequest(&sqs.SendMessageBatchInput{
+		Entries: []*sqs.SendMessageBatchRequestEntry{
+			{Id: aws.String("1"), MessageBody: aws.String("test")},
+			{Id: aws.String("2"), MessageBody: aws.String("test")},
+			{Id: aws.String("3"), MessageBody: aws.String("test")},
+			{Id: aws.String("4"), MessageBody: aws.String("test")},
+		},
+	})
+	req.Handlers.Send.PushBack(func(r *request.Request) {
+		md5 := "098f6bcd4621d373cade4e832627b4f6"
+		body := ioutil.NopCloser(bytes.NewReader([]byte("")))
+		r.HTTPResponse = &http.Response{StatusCode: 200, Body: body}
+		r.Data = &sqs.SendMessageBatchOutput{
+			Successful: []*sqs.SendMessageBatchResultEntry{
+				{MD5OfMessageBody: &md5, MessageId: aws.String("123"), Id: aws.String("1")},
+				{MD5OfMessageBody: aws.String("000"), MessageId: aws.String("456"), Id: aws.String("2")},
+				{MD5OfMessageBody: aws.String("000"), MessageId: aws.String("789"), Id: aws.String("3")},
+				{MD5OfMessageBody: &md5, MessageId: aws.String("012"), Id: aws.String("4")},
+			},
+		}
+	})
+	err := req.Send()
+	assert.Error(t, err)
+
+	assert.Equal(t, "InvalidChecksum", err.(awserr.Error).Code())
+	assert.Contains(t, err.(awserr.Error).Message(), "invalid messages: 456, 789")
+}