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

Fix sqs tests region.

Browse Source
This commit is contained in:
Steve Pulec 2017-02-11 19:50:26 -05:00
parent 2d03182ae2
commit 9076e48fee
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_sqs/test_sqs.py
View File

@ -15,12 +15,11 @@ from tests.helpers import requires_boto_gte
import tests.backport_assert_raises # noqa import tests.backport_assert_raises # noqa
from nose.tools import assert_raises from nose.tools import assert_raises
sqs = boto3.resource('sqs') sqs = boto3.resource('sqs', region_name='us-east-1')
@mock_sqs @mock_sqs
def test_create_queue(): def test_create_queue():
sqs = boto3.resource('sqs', region_name='us-east-1')
new_queue = sqs.create_queue(QueueName='test-queue') new_queue = sqs.create_queue(QueueName='test-queue')
new_queue.should_not.be.none new_queue.should_not.be.none
new_queue.should.have.property('url').should.contain('test-queue') new_queue.should.have.property('url').should.contain('test-queue')
@ -35,13 +34,11 @@ def test_create_queue():
@mock_sqs @mock_sqs
def test_get_inexistent_queue(): def test_get_inexistent_queue():
sqs = boto3.resource('sqs', region_name='us-east-1')
sqs.get_queue_by_name.when.called_with(QueueName='nonexisting-queue').should.throw(botocore.exceptions.ClientError) sqs.get_queue_by_name.when.called_with(QueueName='nonexisting-queue').should.throw(botocore.exceptions.ClientError)
@mock_sqs @mock_sqs
def test_message_send(): def test_message_send():
sqs = boto3.resource('sqs', region_name='us-east-1')
queue = sqs.create_queue(QueueName="blah") queue = sqs.create_queue(QueueName="blah")
msg = queue.send_message(MessageBody="derp") msg = queue.send_message(MessageBody="derp")
@ -55,7 +52,6 @@ def test_message_send():
@mock_sqs @mock_sqs
def test_set_queue_attributes(): def test_set_queue_attributes():
sqs = boto3.resource('sqs', region_name='us-east-1')
queue = sqs.create_queue(QueueName="blah") queue = sqs.create_queue(QueueName="blah")
queue.attributes['VisibilityTimeout'].should.equal("30") queue.attributes['VisibilityTimeout'].should.equal("30")
@ -94,7 +90,7 @@ def test_get_queue_with_prefix():
@mock_sqs @mock_sqs
def test_delete_queue(): def test_delete_queue():
conn = boto3.client("sqs") conn = boto3.client("sqs", region_name='us-east-1')
conn.create_queue(QueueName="test-queue", Attributes={"VisibilityTimeout": "60"}) conn.create_queue(QueueName="test-queue", Attributes={"VisibilityTimeout": "60"})
queue = sqs.Queue('test-queue') queue = sqs.Queue('test-queue')
@ -109,7 +105,7 @@ def test_delete_queue():
@mock_sqs @mock_sqs
def test_set_queue_attribute(): def test_set_queue_attribute():
conn = boto3.client("sqs") conn = boto3.client("sqs", region_name='us-east-1')
conn.create_queue(QueueName="test-queue", Attributes={"VisibilityTimeout": '60'}) conn.create_queue(QueueName="test-queue", Attributes={"VisibilityTimeout": '60'})
queue = sqs.Queue("test-queue") queue = sqs.Queue("test-queue")
@ -122,7 +118,7 @@ def test_set_queue_attribute():
@mock_sqs @mock_sqs
def test_send_message(): def test_send_message():
conn = boto3.client("sqs") conn = boto3.client("sqs", region_name='us-east-1')
conn.create_queue(QueueName="test-queue") conn.create_queue(QueueName="test-queue")
queue = sqs.Queue("test-queue") queue = sqs.Queue("test-queue")