diff options
Diffstat (limited to 'vendor/github.com/goamz/goamz/sqs/sqs_test.go')
-rw-r--r-- | vendor/github.com/goamz/goamz/sqs/sqs_test.go | 414 |
1 files changed, 414 insertions, 0 deletions
diff --git a/vendor/github.com/goamz/goamz/sqs/sqs_test.go b/vendor/github.com/goamz/goamz/sqs/sqs_test.go new file mode 100644 index 000000000..a06433535 --- /dev/null +++ b/vendor/github.com/goamz/goamz/sqs/sqs_test.go @@ -0,0 +1,414 @@ +package sqs + +import ( + "crypto/md5" + "encoding/binary" + "fmt" + "hash" + + "github.com/goamz/goamz/aws" + . "gopkg.in/check.v1" +) + +var _ = Suite(&S{}) + +type S struct { + HTTPSuite + sqs *SQS +} + +func (s *S) SetUpSuite(c *C) { + s.HTTPSuite.SetUpSuite(c) + auth := aws.Auth{AccessKey: "abc", SecretKey: "123"} + s.sqs = New(auth, aws.Region{SQSEndpoint: testServer.URL}) +} + +func (s *S) TestCreateQueue(c *C) { + testServer.PrepareResponse(200, nil, TestCreateQueueXmlOK) + + resp, err := s.sqs.CreateQueue("testQueue") + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/") + c.Assert(req.Header["Date"], Not(Equals), "") + fmt.Printf("%+v\n", req) + c.Assert(req.Form["Action"], DeepEquals, []string{"CreateQueue"}) + c.Assert(req.Form["Attribute.1.Name"], DeepEquals, []string{"VisibilityTimeout"}) + c.Assert(req.Form["Attribute.1.Value"], DeepEquals, []string{"30"}) + + c.Assert(resp.Url, Equals, "http://sqs.us-east-1.amazonaws.com/123456789012/testQueue") + c.Assert(err, IsNil) +} + +func (s *S) TestCreateQueueWithTimeout(c *C) { + testServer.PrepareResponse(200, nil, TestCreateQueueXmlOK) + + s.sqs.CreateQueueWithTimeout("testQueue", 180) + req := testServer.WaitRequest() + + // TestCreateQueue() tests the core functionality, just check the timeout in this test + c.Assert(req.Form["Attribute.1.Name"], DeepEquals, []string{"VisibilityTimeout"}) + c.Assert(req.Form["Attribute.1.Value"], DeepEquals, []string{"180"}) +} + +func (s *S) TestCreateQueueWithAttributes(c *C) { + testServer.PrepareResponse(200, nil, TestCreateQueueXmlOK) + + s.sqs.CreateQueueWithAttributes("testQueue", map[string]string{ + "ReceiveMessageWaitTimeSeconds": "20", + "VisibilityTimeout": "240", + }) + req := testServer.WaitRequest() + + // TestCreateQueue() tests the core functionality, just check the timeout in this test + var receiveMessageWaitSet bool + var visibilityTimeoutSet bool + + for i := 1; i <= 2; i++ { + prefix := fmt.Sprintf("Attribute.%d.", i) + attr := req.FormValue(prefix + "Name") + value := req.FormValue(prefix + "Value") + switch attr { + case "ReceiveMessageWaitTimeSeconds": + c.Assert(value, DeepEquals, "20") + receiveMessageWaitSet = true + case "VisibilityTimeout": + c.Assert(value, DeepEquals, "240") + visibilityTimeoutSet = true + } + } + c.Assert(receiveMessageWaitSet, Equals, true) + c.Assert(visibilityTimeoutSet, Equals, true) +} + +func (s *S) TestListQueues(c *C) { + testServer.PrepareResponse(200, nil, TestListQueuesXmlOK) + + resp, err := s.sqs.ListQueues("") + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/") + c.Assert(req.Header["Date"], Not(Equals), "") + + c.Assert(len(resp.QueueUrl), Not(Equals), 0) + c.Assert(resp.QueueUrl[0], Equals, "http://sqs.us-east-1.amazonaws.com/123456789012/testQueue") + c.Assert(resp.ResponseMetadata.RequestId, Equals, "725275ae-0b9b-4762-b238-436d7c65a1ac") + c.Assert(err, IsNil) +} + +func (s *S) TestDeleteQueue(c *C) { + testServer.PrepareResponse(200, nil, TestDeleteQueueXmlOK) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + resp, err := q.Delete() + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + c.Assert(resp.ResponseMetadata.RequestId, Equals, "6fde8d1e-52cd-4581-8cd9-c512f4c64223") + c.Assert(err, IsNil) +} + +func (s *S) TestPurgeQueue(c *C) { + testServer.PrepareResponse(200, nil, TestPurgeQueueXmlOK) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + resp, err := q.Purge() + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + c.Assert(resp.ResponseMetadata.RequestId, Equals, "6fde8d1e-52cd-4581-8cd9-c512f4c64223") + c.Assert(err, IsNil) +} + +func (s *S) TestSendMessage(c *C) { + testServer.PrepareResponse(200, nil, TestSendMessageXmlOK) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + resp, err := q.SendMessage("This is a test message") + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + msg := "This is a test message" + var h hash.Hash = md5.New() + h.Write([]byte(msg)) + c.Assert(resp.MD5, Equals, fmt.Sprintf("%x", h.Sum(nil))) + c.Assert(resp.Id, Equals, "5fea7756-0ea4-451a-a703-a558b933e274") + c.Assert(err, IsNil) +} + +func (s *S) TestSendMessageRelativePath(c *C) { + testServer.PrepareResponse(200, nil, TestSendMessageXmlOK) + + q := &Queue{s.sqs, "/123456789012/testQueue/"} + resp, err := q.SendMessage("This is a test message") + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + msg := "This is a test message" + var h hash.Hash = md5.New() + h.Write([]byte(msg)) + c.Assert(resp.MD5, Equals, fmt.Sprintf("%x", h.Sum(nil))) + c.Assert(resp.Id, Equals, "5fea7756-0ea4-451a-a703-a558b933e274") + c.Assert(err, IsNil) +} + +func encodeMessageAttribute(str string) []byte { + bstr := []byte(str) + bs := make([]byte, 4+len(bstr)) + binary.BigEndian.PutUint32(bs, uint32(len(bstr))) + copy(bs[4:len(bs)], bstr) + return bs +} + +func (s *S) TestSendMessageWithAttributes(c *C) { + testServer.PrepareResponse(200, nil, TestSendMessageXmlOK) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + attrs := map[string]string{ + "test_attribute_name_1": "test_attribute_value_1", + } + resp, err := q.SendMessageWithAttributes("This is a test message", attrs) + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + var attrsHash = md5.New() + attrsHash.Write(encodeMessageAttribute("test_attribute_name_1")) + attrsHash.Write(encodeMessageAttribute("String")) + attrsHash.Write([]byte{1}) + attrsHash.Write(encodeMessageAttribute("test_attribute_value_1")) + c.Assert(resp.MD5OfMessageAttributes, Equals, fmt.Sprintf("%x", attrsHash.Sum(nil))) + + msg := "This is a test message" + var h hash.Hash = md5.New() + h.Write([]byte(msg)) + c.Assert(resp.MD5, Equals, fmt.Sprintf("%x", h.Sum(nil))) + c.Assert(resp.Id, Equals, "5fea7756-0ea4-451a-a703-a558b933e274") + c.Assert(err, IsNil) +} + +func (s *S) TestSendMessageBatch(c *C) { + testServer.PrepareResponse(200, nil, TestSendMessageBatchXmlOk) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + + msgList := []string{"test message body 1", "test message body 2"} + resp, err := q.SendMessageBatchString(msgList) + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + for idx, msg := range msgList { + var h hash.Hash = md5.New() + h.Write([]byte(msg)) + c.Assert(resp.SendMessageBatchResult[idx].MD5OfMessageBody, Equals, fmt.Sprintf("%x", h.Sum(nil))) + c.Assert(err, IsNil) + } +} + +func (s *S) TestDeleteMessageBatch(c *C) { + testServer.PrepareResponse(200, nil, TestDeleteMessageBatchXmlOK) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + + msgList := []Message{*(&Message{ReceiptHandle: "gfk0T0R0waama4fVFffkjPQrrvzMrOg0fTFk2LxT33EuB8wR0ZCFgKWyXGWFoqqpCIiprQUEhir%2F5LeGPpYTLzjqLQxyQYaQALeSNHb0us3uE84uujxpBhsDkZUQkjFFkNqBXn48xlMcVhTcI3YLH%2Bd%2BIqetIOHgBCZAPx6r%2B09dWaBXei6nbK5Ygih21DCDdAwFV68Jo8DXhb3ErEfoDqx7vyvC5nCpdwqv%2BJhU%2FTNGjNN8t51v5c%2FAXvQsAzyZVNapxUrHIt4NxRhKJ72uICcxruyE8eRXlxIVNgeNP8ZEDcw7zZU1Zw%3D%3D"}), + *(&Message{ReceiptHandle: "gfk0T0R0waama4fVFffkjKzmhMCymjQvfTFk2LxT33G4ms5subrE0deLKWSscPU1oD3J9zgeS4PQQ3U30qOumIE6AdAv3w%2F%2Fa1IXW6AqaWhGsEPaLm3Vf6IiWqdM8u5imB%2BNTwj3tQRzOWdTOePjOjPcTpRxBtXix%2BEvwJOZUma9wabv%2BSw6ZHjwmNcVDx8dZXJhVp16Bksiox%2FGrUvrVTCJRTWTLc59oHLLF8sEkKzRmGNzTDGTiV%2BYjHfQj60FD3rVaXmzTsoNxRhKJ72uIHVMGVQiAGgB%2BqAbSqfKHDQtVOmJJgkHug%3D%3D"}), + } + + resp, err := q.DeleteMessageBatch(msgList) + c.Assert(err, IsNil) + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + for idx, _ := range msgList { + c.Assert(resp.DeleteMessageBatchResult[idx].Id, Equals, fmt.Sprintf("msg%d", idx+1)) + } +} + +func (s *S) TestDeleteMessageUsingReceiptHandle(c *C) { + testServer.PrepareResponse(200, nil, TestDeleteMessageUsingReceiptXmlOK) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + + msg := &Message{ReceiptHandle: "gfk0T0R0waama4fVFffkjRQrrvzMrOg0fTFk2LxT33EuB8wR0ZCFgKWyXGWFoqqpCIiprQUEhir%2F5LeGPpYTLzjqLQxyQYaQALeSNHb0us3uE84uujxpBhsDkZUQkjFFkNqBXn48xlMcVhTcI3YLH%2Bd%2BIqetIOHgBCZAPx6r%2B09dWaBXei6nbK5Ygih21DCDdAwFV68Jo8DXhb3ErEfoDqx7vyvC5nCpdwqv%2BJhU%2FTNGjNN8t51v5c%2FAXvQsAzyZVNapxUrHIt4NxRhKJ72uICcxruyE8eRXlxIVNgeNP8ZEDcw7zZU1Zw%3D%3D"} + + resp, err := q.DeleteMessageUsingReceiptHandle(msg.ReceiptHandle) + c.Assert(err, IsNil) + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + c.Assert(resp.ResponseMetadata.RequestId, Equals, "d6d86b7a-74d1-4439-b43f-196a1e29cd85") +} + +func (s *S) TestReceiveMessage(c *C) { + testServer.PrepareResponse(200, nil, TestReceiveMessageXmlOK) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + resp, err := q.ReceiveMessage(5) + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + c.Assert(len(resp.Messages), Not(Equals), 0) + c.Assert(resp.Messages[0].MessageId, Equals, "5fea7756-0ea4-451a-a703-a558b933e274") + c.Assert(resp.Messages[0].MD5OfBody, Equals, "fafb00f5732ab283681e124bf8747ed1") + c.Assert(resp.Messages[0].ReceiptHandle, Equals, "MbZj6wDWli+JvwwJaBV+3dcjk2YW2vA3+STFFljTM8tJJg6HRG6PYSasuWXPJB+CwLj1FjgXUv1uSj1gUPAWV66FU/WeR4mq2OKpEGYWbnLmpRCJVAyeMjeU5ZBdtcQ+QEauMZc8ZRv37sIW2iJKq3M9MFx1YvV11A2x/KSbkJ0=") + c.Assert(resp.Messages[0].Body, Equals, "This is a test message") + + c.Assert(len(resp.Messages[0].Attribute), Not(Equals), 0) + + expectedAttributeResults := []struct { + Name string + Value string + }{ + {Name: "SenderId", Value: "195004372649"}, + {Name: "SentTimestamp", Value: "1238099229000"}, + {Name: "ApproximateReceiveCount", Value: "5"}, + {Name: "ApproximateFirstReceiveTimestamp", Value: "1250700979248"}, + } + + for i, expected := range expectedAttributeResults { + c.Assert(resp.Messages[0].Attribute[i].Name, Equals, expected.Name) + c.Assert(resp.Messages[0].Attribute[i].Value, Equals, expected.Value) + } + + c.Assert(len(resp.Messages[0].MessageAttribute), Not(Equals), 0) + + expectedMessageAttributeResults := []struct { + Name string + Value struct { + DataType string + BinaryValue []byte + StringValue string + + // Not yet implemented (Reserved for future use) + BinaryListValues [][]byte + StringListValues []string + } + }{ + { + Name: "CustomAttribute", + Value: struct { + DataType string + BinaryValue []byte + StringValue string + + // Not yet implemented (Reserved for future use) + BinaryListValues [][]byte + StringListValues []string + }{ + DataType: "String", + StringValue: "Testing, testing, 1, 2, 3", + }, + }, + { + Name: "BinaryCustomAttribute", + Value: struct { + DataType string + BinaryValue []byte + StringValue string + + // Not yet implemented (Reserved for future use) + BinaryListValues [][]byte + StringListValues []string + }{ + DataType: "Binary", + BinaryValue: []byte("iVBORw0KGgoAAAANSUhEUgAAABIAAAASCAYAAABWzo5XAAABA0lEQVQ4T72UrQ4CMRCEewhyiiBPopBgcfAUSIICB88CDhRB8hTgsCBRyJMEdUFwZJpMs/3LHQlhVdPufJ1ut03UjyKJcR5zVc4umbW87eeqvVFBjTdJwP54D+4xGXVUCGiBxoOsJOCd9IKgRnnV8wAezrnRmwGcpKtCJ8UgJBNWLFNzVAOimyqIhElXGkQ3LmQ6fKrdqaW1cixhdKVBcEOBLEwViBugVv8B1elVuLYcoTea624drcl5LW4KTRsFhQpLtVzzQKGCh2DuHI8FvdVH7vGQKEPerHRjgegKMESsXgAgWBtu5D1a9BQWCXSrzx9BvjPPkRQR6IJcQNTRV/cvkj93DqUTWzVDIQAAAABJRU5ErkJggg=="), + }, + }, + } + + for i, expected := range expectedMessageAttributeResults { + c.Assert(resp.Messages[0].MessageAttribute[i].Name, Equals, expected.Name) + c.Assert(resp.Messages[0].MessageAttribute[i].Value.DataType, Equals, expected.Value.DataType) + c.Assert(string(resp.Messages[0].MessageAttribute[i].Value.BinaryValue), Equals, string(expected.Value.BinaryValue)) + c.Assert(resp.Messages[0].MessageAttribute[i].Value.StringValue, Equals, expected.Value.StringValue) + } + + c.Assert(err, IsNil) +} + +func (s *S) TestChangeMessageVisibility(c *C) { + testServer.PrepareResponse(200, nil, TestReceiveMessageXmlOK) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + + resp1, err := q.ReceiveMessage(1) + req := testServer.WaitRequest() + + testServer.PrepareResponse(200, nil, TestChangeMessageVisibilityXmlOK) + + resp, err := q.ChangeMessageVisibility(&resp1.Messages[0], 50) + req = testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + c.Assert(req.Header["Date"], Not(Equals), "") + + c.Assert(resp.ResponseMetadata.RequestId, Equals, "6a7a282a-d013-4a59-aba9-335b0fa48bed") + c.Assert(err, IsNil) +} + +func (s *S) TestGetQueueAttributes(c *C) { + testServer.PrepareResponse(200, nil, TestGetQueueAttributesXmlOK) + + q := &Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"} + + resp, err := q.GetQueueAttributes("All") + req := testServer.WaitRequest() + + c.Assert(req.Method, Equals, "GET") + c.Assert(req.URL.Path, Equals, "/123456789012/testQueue/") + + c.Assert(resp.ResponseMetadata.RequestId, Equals, "1ea71be5-b5a2-4f9d-b85a-945d8d08cd0b") + + c.Assert(len(resp.Attributes), Equals, 9) + + expectedResults := []struct { + Name string + Value string + }{ + {Name: "ReceiveMessageWaitTimeSeconds", Value: "2"}, + {Name: "VisibilityTimeout", Value: "30"}, + {Name: "ApproximateNumberOfMessages", Value: "0"}, + {Name: "ApproximateNumberOfMessagesNotVisible", Value: "0"}, + {Name: "CreatedTimestamp", Value: "1286771522"}, + {Name: "LastModifiedTimestamp", Value: "1286771522"}, + {Name: "QueueArn", Value: "arn:aws:sqs:us-east-1:123456789012:qfoo"}, + {Name: "MaximumMessageSize", Value: "8192"}, + {Name: "MessageRetentionPeriod", Value: "345600"}, + } + + for i, expected := range expectedResults { + c.Assert(resp.Attributes[i].Name, Equals, expected.Name) + c.Assert(resp.Attributes[i].Value, Equals, expected.Value) + } + + c.Assert(err, IsNil) +} |