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, 0 insertions, 414 deletions
diff --git a/vendor/github.com/goamz/goamz/sqs/sqs_test.go b/vendor/github.com/goamz/goamz/sqs/sqs_test.go deleted file mode 100644 index a06433535..000000000 --- a/vendor/github.com/goamz/goamz/sqs/sqs_test.go +++ /dev/null @@ -1,414 +0,0 @@ -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) -} |