246 lines
8.1 KiB
Go
246 lines
8.1 KiB
Go
|
package sqs_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"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()
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expect no error, got %v", 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()
|
||
|
|
if err == nil {
|
||
|
|
t.Fatalf("expect error, got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
if e, a := "InvalidChecksum", err.(awserr.Error).Code(); e != a {
|
||
|
|
t.Errorf("expect %v, got %v", e, a)
|
||
|
|
}
|
||
|
|
if e, a := err.(awserr.Error).Message(), "expected MD5 checksum '000', got '098f6bcd4621d373cade4e832627b4f6'"; !strings.Contains(a, e) {
|
||
|
|
t.Errorf("expect %v to be in %v, was not", e, a)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
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()
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expect no error, got %v", 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()
|
||
|
|
if err == nil {
|
||
|
|
t.Fatalf("expect error, got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
if e, a := "InvalidChecksum", err.(awserr.Error).Code(); e != a {
|
||
|
|
t.Errorf("expect %v, got %v", e, a)
|
||
|
|
}
|
||
|
|
if e, a := err.(awserr.Error).Message(), "cannot compute checksum. missing body"; !strings.Contains(a, e) {
|
||
|
|
t.Errorf("expect %v to be in %v, was not", e, a)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
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()
|
||
|
|
if err == nil {
|
||
|
|
t.Fatalf("expect error, got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
if e, a := "InvalidChecksum", err.(awserr.Error).Code(); e != a {
|
||
|
|
t.Errorf("expect %v, got %v", e, a)
|
||
|
|
}
|
||
|
|
if e, a := err.(awserr.Error).Message(), "cannot verify checksum. missing response MD5"; !strings.Contains(a, e) {
|
||
|
|
t.Errorf("expect %v to be in %v, was not", e, a)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
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()
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expect no error, got %v", 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()
|
||
|
|
if err == nil {
|
||
|
|
t.Fatalf("expect error, got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
if e, a := "InvalidChecksum", err.(awserr.Error).Code(); e != a {
|
||
|
|
t.Errorf("expect %v, got %v", e, a)
|
||
|
|
}
|
||
|
|
if e, a := err.(awserr.Error).Message(), "invalid messages: 123, 456"; !strings.Contains(a, e) {
|
||
|
|
t.Errorf("expect %v to be in %v, was not", e, a)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
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()
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("expect no error, got %v", 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()
|
||
|
|
if err == nil {
|
||
|
|
t.Fatalf("expect error, got nil")
|
||
|
|
}
|
||
|
|
|
||
|
|
if e, a := "InvalidChecksum", err.(awserr.Error).Code(); e != a {
|
||
|
|
t.Errorf("expect %v, got %v", e, a)
|
||
|
|
}
|
||
|
|
if e, a := err.(awserr.Error).Message(), "invalid messages: 456, 789"; !strings.Contains(a, e) {
|
||
|
|
t.Errorf("expect %v to be in %v, was not", e, a)
|
||
|
|
}
|
||
|
|
}
|