Vendor aws-sdk-go (dep ensure) (#178)
This commit is contained in:
Onur Filiz
245
vendor/github.com/aws/aws-sdk-go/service/sqs/checksums_test.go
generated
vendored
Normal file
245
vendor/github.com/aws/aws-sdk-go/service/sqs/checksums_test.go
generated
vendored
Normal file
@@ -0,0 +1,245 @@
|
||||
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)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user