add most of sqs features
This commit is contained in:
parent
67261d3125
commit
41890225e6
5 changed files with 305 additions and 3 deletions
|
|
@ -41,3 +41,74 @@ def test_set_queue_attribute():
|
|||
queue.set_attribute("VisibilityTimeout", 45)
|
||||
queue = conn.get_all_queues()[0]
|
||||
queue.get_timeout().should.equal(45)
|
||||
|
||||
|
||||
@mock_sqs
|
||||
def test_send_message():
|
||||
conn = boto.connect_sqs('the_key', 'the_secret')
|
||||
queue = conn.create_queue("test-queue", visibility_timeout=60)
|
||||
|
||||
conn.send_message(queue, 'this is a test message')
|
||||
conn.send_message(queue, 'this is another test message')
|
||||
|
||||
messages = conn.receive_message(queue, number_messages=1)
|
||||
messages[0].get_body().should.equal('this is a test message')
|
||||
|
||||
|
||||
@mock_sqs
|
||||
def test_queue_length():
|
||||
conn = boto.connect_sqs('the_key', 'the_secret')
|
||||
queue = conn.create_queue("test-queue", visibility_timeout=60)
|
||||
|
||||
conn.send_message(queue, 'this is a test message')
|
||||
conn.send_message(queue, 'this is another test message')
|
||||
queue.count().should.equal(2)
|
||||
|
||||
|
||||
@mock_sqs
|
||||
def test_delete_message():
|
||||
conn = boto.connect_sqs('the_key', 'the_secret')
|
||||
queue = conn.create_queue("test-queue", visibility_timeout=60)
|
||||
|
||||
conn.send_message(queue, 'this is a test message')
|
||||
conn.send_message(queue, 'this is another test message')
|
||||
|
||||
messages = conn.receive_message(queue, number_messages=1)
|
||||
messages[0].delete()
|
||||
|
||||
queue.count().should.equal(1)
|
||||
|
||||
|
||||
@mock_sqs
|
||||
def test_send_batch_operation():
|
||||
conn = boto.connect_sqs('the_key', 'the_secret')
|
||||
queue = conn.create_queue("test-queue", visibility_timeout=60)
|
||||
|
||||
conn.send_message_batch(queue, [
|
||||
("my_first_message", 'test message 1', 0),
|
||||
("my_second_message", 'test message 2', 0),
|
||||
("my_third_message", 'test message 3', 0),
|
||||
])
|
||||
|
||||
messages = queue.get_messages(3)
|
||||
messages[0].get_body().should.equal("test message 1")
|
||||
|
||||
# Test that pulling more messages doesn't break anything
|
||||
messages = queue.get_messages(2)
|
||||
|
||||
|
||||
@mock_sqs
|
||||
def test_delete_batch_operation():
|
||||
conn = boto.connect_sqs('the_key', 'the_secret')
|
||||
queue = conn.create_queue("test-queue", visibility_timeout=60)
|
||||
|
||||
conn.send_message_batch(queue, [
|
||||
("my_first_message", 'test message 1', 0),
|
||||
("my_second_message", 'test message 2', 0),
|
||||
("my_third_message", 'test message 3', 0),
|
||||
])
|
||||
|
||||
messages = queue.get_messages(2)
|
||||
queue.delete_message_batch(messages)
|
||||
|
||||
queue.count().should.equal(1)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue