Skip to content

Commit 9394da8

Browse files
committed
add sqs queue init test
1 parent f95395e commit 9394da8

File tree

1 file changed

+35
-0
lines changed

1 file changed

+35
-0
lines changed

tests/plugins/aws/queues/sqs/test_sqs_queue.py

Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,41 @@ async def delete_queue(queue_url: str) -> None:
4242
assert e.response["Error"]["Code"] == "AWS.SimpleQueueService.NonExistentQueue"
4343

4444

45+
@pytest.mark.parametrize(
46+
"config",
47+
[
48+
{
49+
"type": "plugin.aws.sqs",
50+
"name": "app",
51+
"url": "http://motoserver:5000/123456789012/app",
52+
"region": "us-east-1",
53+
"create_queue": True,
54+
"queue_wait_message_time": 2,
55+
"queue_visibility_time": 15,
56+
},
57+
{
58+
"type": "plugin.aws.sqs",
59+
"name": "other",
60+
"url": "some_other_url",
61+
# No region
62+
"create_queue": False,
63+
"queue_wait_message_time": 5,
64+
"queue_visibility_time": 20,
65+
},
66+
],
67+
)
68+
async def test_queue_init_aws_client_params(config):
69+
"""'Queue' should initialize with the correct AWS client parameters"""
70+
queue = sqs_queue.Queue(config=config)
71+
72+
assert queue._aws_client_params["credential_name"] == "application"
73+
assert queue._aws_client_params["service"] == "sqs"
74+
if "region" in config:
75+
assert queue._aws_client_params["region_name"] == config["region"]
76+
else:
77+
assert "region_name" not in queue._aws_client_params
78+
79+
4580
@pytest.mark.parametrize("queue_wait_message_time", [1, 2, 3, 4, 5])
4681
async def test_queue_wait_message_time(queue_wait_message_time):
4782
queue = sqs_queue.Queue(

0 commit comments

Comments
 (0)