Skip to content

Commit 2485dd2

Browse files
committed
Rename SqsSendMessage to SqsSendMessageStep
1 parent 2da5569 commit 2485dd2

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

src/stepfunctions/steps/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,4 +19,4 @@
1919
from stepfunctions.steps.sagemaker import TrainingStep, TransformStep, ModelStep, EndpointConfigStep, EndpointStep
2020
from stepfunctions.steps.compute import LambdaStep, BatchSubmitJobStep, GlueStartJobRunStep, EcsRunTaskStep
2121
from stepfunctions.steps.service import DynamoDBGetItemStep, DynamoDBPutItemStep, DynamoDBUpdateItemStep, DynamoDBDeleteItemStep
22-
from stepfunctions.steps.service import SnsPublishStep, SqsSendMessage
22+
from stepfunctions.steps.service import SnsPublishStep, SqsSendMessageStep

src/stepfunctions/steps/service.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def __init__(self, state_id, wait_for_callback=False, **kwargs):
122122
super(SnsPublishStep, self).__init__(state_id, **kwargs)
123123

124124

125-
class SqsSendMessage(Task):
125+
class SqsSendMessageStep(Task):
126126

127127
"""
128128
Creates a Task state to send a message to SQS queue. See `Call Amazon SQS with Step Functions <https://docs.aws.amazon.com/step-functions/latest/dg/connect-sqs.html>`_ for more details.
@@ -146,4 +146,4 @@ def __init__(self, state_id, wait_for_callback=False, **kwargs):
146146
else:
147147
kwargs[Field.Resource.value] = 'arn:aws:states:::sqs:sendMessage'
148148

149-
super(SqsSendMessage, self).__init__(state_id, **kwargs)
149+
super(SqsSendMessageStep, self).__init__(state_id, **kwargs)

tests/unit/test_service_steps.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import pytest
1616

1717
from stepfunctions.steps.service import DynamoDBGetItemStep, DynamoDBPutItemStep, DynamoDBUpdateItemStep, DynamoDBDeleteItemStep
18-
from stepfunctions.steps.service import SnsPublishStep, SqsSendMessage
18+
from stepfunctions.steps.service import SnsPublishStep, SqsSendMessageStep
1919

2020

2121
def test_sns_publish_step_creation():
@@ -57,7 +57,7 @@ def test_sns_publish_step_creation():
5757

5858

5959
def test_sqs_send_message_step_creation():
60-
step = SqsSendMessage('Send to SQS', parameters={
60+
step = SqsSendMessageStep('Send to SQS', parameters={
6161
'QueueUrl': 'https://sqs.us-east-1.amazonaws.com/123456789012/myQueue',
6262
'MessageBody': 'Hello'
6363
})
@@ -72,7 +72,7 @@ def test_sqs_send_message_step_creation():
7272
'End': True
7373
}
7474

75-
step = SqsSendMessage('Send to SQS', wait_for_callback=True, parameters={
75+
step = SqsSendMessageStep('Send to SQS', wait_for_callback=True, parameters={
7676
'QueueUrl': 'https://sqs.us-east-1.amazonaws.com/123456789012/myQueue',
7777
'MessageBody': {
7878
'Input.$': '$',

0 commit comments

Comments
 (0)