-
Notifications
You must be signed in to change notification settings - Fork 9
/
sqs_test.go
128 lines (97 loc) · 4.43 KB
/
sqs_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
package sqs_test
import (
"crypto/md5"
"hash"
"fmt"
"launchpad.net/gocheck"
"launchpad.net/goamz/aws"
"launchpad.net/goamz/sqs"
)
var _ = gocheck.Suite(&S{})
type S struct {
HTTPSuite
sqs *sqs.SQS
}
func (s *S) SetUpSuite(c *gocheck.C) {
s.HTTPSuite.SetUpSuite(c)
auth := aws.Auth{"abc", "123"}
s.sqs = sqs.New(auth, aws.Region{SQSEndpoint: testServer.URL})
}
func (s *S) TestCreateQueue(c *gocheck.C) {
testServer.PrepareResponse(200, nil, TestCreateQueueXmlOK)
resp, err := s.sqs.CreateQueue("testQueue")
req := testServer.WaitRequest()
c.Assert(req.Method, gocheck.Equals, "GET")
c.Assert(req.URL.Path, gocheck.Equals, "/")
c.Assert(req.Header["Date"], gocheck.Not(gocheck.Equals), "")
c.Assert(resp.Url, gocheck.Equals, "http://sqs.us-east-1.amazonaws.com/123456789012/testQueue")
c.Assert(err, gocheck.IsNil)
}
func (s *S) TestListQueues(c *gocheck.C) {
testServer.PrepareResponse(200, nil, TestListQueuesXmlOK)
resp, err := s.sqs.ListQueues("")
req := testServer.WaitRequest()
c.Assert(req.Method, gocheck.Equals, "GET")
c.Assert(req.URL.Path, gocheck.Equals, "/")
c.Assert(req.Header["Date"], gocheck.Not(gocheck.Equals), "")
c.Assert(len(resp.QueueUrl), gocheck.Not(gocheck.Equals), 0)
c.Assert(resp.QueueUrl[0], gocheck.Equals, "http://sqs.us-east-1.amazonaws.com/123456789012/testQueue")
c.Assert(resp.ResponseMetadata.RequestId, gocheck.Equals, "725275ae-0b9b-4762-b238-436d7c65a1ac")
c.Assert(err, gocheck.IsNil)
}
func (s *S) TestDeleteQueue(c *gocheck.C) {
testServer.PrepareResponse(200, nil, TestDeleteQueueXmlOK)
q := &sqs.Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"}
resp, err := q.Delete()
req := testServer.WaitRequest()
c.Assert(req.Method, gocheck.Equals, "GET")
c.Assert(req.URL.Path, gocheck.Equals, "/123456789012/testQueue/")
c.Assert(req.Header["Date"], gocheck.Not(gocheck.Equals), "")
c.Assert(resp.ResponseMetadata.RequestId, gocheck.Equals, "6fde8d1e-52cd-4581-8cd9-c512f4c64223")
c.Assert(err, gocheck.IsNil)
}
func (s *S) TestSendMessage(c *gocheck.C) {
testServer.PrepareResponse(200, nil, TestSendMessageXmlOK)
q := &sqs.Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"}
resp, err := q.SendMessage("This is a test message")
req := testServer.WaitRequest()
c.Assert(req.Method, gocheck.Equals, "GET")
c.Assert(req.URL.Path, gocheck.Equals, "/123456789012/testQueue/")
c.Assert(req.Header["Date"], gocheck.Not(gocheck.Equals), "")
msg := "This is a test message"
var h hash.Hash = md5.New()
h.Write([]byte(msg))
c.Assert(resp.MD5, gocheck.Equals, fmt.Sprintf("%x", h.Sum()))
c.Assert(resp.Id, gocheck.Equals, "5fea7756-0ea4-451a-a703-a558b933e274")
c.Assert(err, gocheck.IsNil)
}
func (s *S) TestReceiveMessage(c *gocheck.C) {
testServer.PrepareResponse(200, nil, TestReceiveMessageXmlOK)
q := &sqs.Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"}
resp, err := q.ReceiveMessage(5, 30)
req := testServer.WaitRequest()
c.Assert(req.Method, gocheck.Equals, "GET")
c.Assert(req.URL.Path, gocheck.Equals, "/123456789012/testQueue/")
c.Assert(req.Header["Date"], gocheck.Not(gocheck.Equals), "")
c.Assert(len(resp.Messages), gocheck.Not(gocheck.Equals), 0)
c.Assert(resp.Messages[0].MessageId, gocheck.Equals, "5fea7756-0ea4-451a-a703-a558b933e274")
c.Assert(resp.Messages[0].MD5OfBody, gocheck.Equals, "fafb00f5732ab283681e124bf8747ed1")
c.Assert(resp.Messages[0].ReceiptHandle, gocheck.Equals, "MbZj6wDWli+JvwwJaBV+3dcjk2YW2vA3+STFFljTM8tJJg6HRG6PYSasuWXPJB+CwLj1FjgXUv1uSj1gUPAWV66FU/WeR4mq2OKpEGYWbnLmpRCJVAyeMjeU5ZBdtcQ+QEauMZc8ZRv37sIW2iJKq3M9MFx1YvV11A2x/KSbkJ0=")
c.Assert(resp.Messages[0].Body, gocheck.Equals, "This is a test message")
c.Assert(len(resp.Messages[0].Attribute), gocheck.Not(gocheck.Equals), 0)
c.Assert(err, gocheck.IsNil)
}
func (s *S) TestChangeMessageVisibility(c *gocheck.C) {
testServer.PrepareResponse(200, nil, TestReceiveMessageXmlOK)
q := &sqs.Queue{s.sqs, testServer.URL + "/123456789012/testQueue/"}
resp1, err := q.ReceiveMessage(1, 30)
req := testServer.WaitRequest()
testServer.PrepareResponse(200, nil, TestChangeMessageVisibilityXmlOK)
resp, err := q.ChangeMessageVisibility(&resp1.Messages[0], 50)
req = testServer.WaitRequest()
c.Assert(req.Method, gocheck.Equals, "GET")
c.Assert(req.URL.Path, gocheck.Equals, "/123456789012/testQueue/")
c.Assert(req.Header["Date"], gocheck.Not(gocheck.Equals), "")
c.Assert(resp.ResponseMetadata.RequestId, gocheck.Equals, "6a7a282a-d013-4a59-aba9-335b0fa48bed")
c.Assert(err, gocheck.IsNil)
}