Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

boto3 SQS set attributes fix.

Browse Source
This commit is contained in:
Scott Leishman 2016-02-19 23:32:01 -08:00
parent a9d161e58e
commit 3fd33c03bf
2 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
moto/sqs/responses.py
View File

@ -90,9 +90,14 @@ class SQSResponse(BaseResponse):
def set_queue_attributes(self):
queue_name = self._get_queue_name()
key = camelcase_to_underscores(self.querystring.get('Attribute.Name')[0])
value = self.querystring.get('Attribute.Value')[0]
self.sqs_backend.set_queue_attribute(queue_name, key, value)
if "Attribute.Name" in self.querystring:
key = camelcase_to_underscores(self.querystring.get("Attribute.Name")[0])
value = self.querystring.get("Attribute.Value")[0]
self.sqs_backend.set_queue_attribute(queue_name, key, value)
for a in self._get_list_prefix("Attribute"):
key = camelcase_to_underscores(a["name"])
value = a["value"]
self.sqs_backend.set_queue_attribute(queue_name, key, value)
return SET_QUEUE_ATTRIBUTE_RESPONSE
def delete_queue(self):

11
tests/test_sqs/test_sqs.py
View File

@ -491,3 +491,14 @@ def test_boto3_message_send():
messages = queue.receive_messages()
messages.should.have.length_of(1)
@mock_sqs
def test_boto3_set_queue_attributes():
sqs = boto3.resource('sqs', region_name='us-east-1')
queue = sqs.create_queue(QueueName="blah")
queue.attributes['VisibilityTimeout'].should.equal("30")
queue.set_attributes(Attributes={"VisibilityTimeout": "45"})
queue.attributes['VisibilityTimeout'].should.equal("45")