adds (approximate) lock_size to SQS broker

docs: added lock_size
This commit is contained in:
Ilan Steemers
2015-09-18 15:40:30 +02:00
parent 6af852fcee
commit 16c28a88e2
3 changed files with 9 additions and 0 deletions
+3
View File
@@ -27,6 +27,9 @@ class Sqs(Broker):
def queue_size(self):
return int(self.queue.attributes['ApproximateNumberOfMessages'])
def lock_size(self):
return int(self.queue.attributes['ApproximateNumberOfMessagesNotVisible'])
def delete(self, task_id):
message = self.sqs.Message(self.queue.url, task_id)
message.delete()
+1
View File
@@ -216,6 +216,7 @@ def test_sqs():
broker.acknowledge(task[0])
# duplicate acknowledge
broker.acknowledge(task[0])
assert broker.lock_size() == 0
# delete queue
broker.enqueue('test')
broker.purge_queue()