catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add tags parameter to sqs.create_queue

Browse source
This commit is contained in:
gruebel 2019-10-13 14:51:31 +02:00
commit 49045fe680
3 changed files with 74 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
moto/sqs/models.py
View file

@ -415,7 +415,7 @@ class SQSBackend(BaseBackend):
self.__dict__ = {}
self.__init__(region_name)
def create_queue(self, name, **kwargs):
def create_queue(self, name, tags, **kwargs):
queue = self.queues.get(name)
if queue:
try:
@ -454,6 +454,8 @@ class SQSBackend(BaseBackend):
pass
queue = Queue(name, region=self.region_name, **kwargs)
self.queues[name] = queue
queue.tags = tags
return queue
def list_queues(self, queue_name_prefix):
@ -654,6 +656,10 @@ class SQSBackend(BaseBackend):
def untag_queue(self, queue_name, tag_keys):
queue = self.get_queue(queue_name)
if len(tag_keys) == 0:
raise RESTError('InvalidParameterValue', 'Tag keys must be between 1 and 128 characters in length.')
for key in tag_keys:
try:
del queue.tags[key]

13
moto/sqs/responses.py
View file

@ -33,6 +33,12 @@ class SQSResponse(BaseResponse):
self._attribute = self._get_map_prefix('Attribute', key_end='.Name', value_end='.Value')
return self._attribute
@property
def tags(self):
if not hasattr(self, '_tags'):
self._tags = self._get_map_prefix('Tag', key_end='.Key', value_end='.Value')
return self._tags
def _get_queue_name(self):
try:
queue_name = self.querystring.get('QueueUrl')[0].split("/")[-1]
@ -73,12 +79,12 @@ class SQSResponse(BaseResponse):
queue_name = self._get_param("QueueName")
try:
queue = self.sqs_backend.create_queue(queue_name, **self.attribute)
queue = self.sqs_backend.create_queue(queue_name, self.tags, **self.attribute)
except MessageAttributesInvalid as e:
return self._error('InvalidParameterValue', e.description)
template = self.response_template(CREATE_QUEUE_RESPONSE)
return template.render(queue=queue, request_url=request_url)
return template.render(queue_url=queue.url(request_url))
def get_queue_url(self):
request_url = urlparse(self.uri)
@ -416,8 +422,7 @@ class SQSResponse(BaseResponse):
CREATE_QUEUE_RESPONSE = """<CreateQueueResponse>
<CreateQueueResult>
<QueueUrl>{{ queue.url(request_url) }}</QueueUrl>
<VisibilityTimeout>{{ queue.visibility_timeout }}</VisibilityTimeout>
<QueueUrl>{{ queue_url }}</QueueUrl>
</CreateQueueResult>
<ResponseMetadata>
<RequestId></RequestId>