Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use sqs_collect_sns_messages method in localstack

Best Python code snippet using localstack_python

test_s3_notifications_sns.py

Source:test_s3_notifications_sns.py

...44                )45            ]46        ),47    )48def sqs_collect_sns_messages(49    sqs_client: "SQSClient", queue_url: str, min_messages: int, timeout: int = 1050) -> List[Dict]:51    """52    Polls the given queue for the given amount of time and extracts the received SQS messages all SNS messages (messages that have a "TopicArn" field).53    :param sqs_client: the boto3 client to use54    :param queue_url: the queue URL connected to the topic55    :param min_messages: the minimum number of messages to wait for56    :param timeout: the number of seconds to wait before raising an assert error57    :return: a list with the deserialized SNS messages58    """59    collected_messages = []60    def collect_events() -> int:61        _response = sqs_client.receive_message(62            QueueUrl=queue_url, WaitTimeSeconds=timeout, MaxNumberOfMessages=163        )64        messages = _response.get("Messages", [])65        if not messages:66            LOG.info("no messages received from %s after %d seconds", queue_url, timeout)67        for m in messages:68            body = m["Body"]69            # see https://www.mikulskibartosz.name/what-is-s3-test-event/70            if "s3:TestEvent" in body:71                continue72            doc = json.loads(body)73            assert "TopicArn" in doc, f"unexpected event in message {m}"74            collected_messages.append(doc)75        return len(collected_messages)76    assert poll_condition(lambda: collect_events() >= min_messages, timeout=timeout)77    return collected_messages78class TestS3NotificationsToSns:79    @pytest.mark.aws_validated80    def test_object_created_put(81        self,82        s3_client,83        sqs_client,84        sns_client,85        s3_create_bucket,86        sqs_create_queue,87        sns_create_topic,88        sns_create_sqs_subscription,89    ):90        bucket_name = s3_create_bucket()91        topic_arn = sns_create_topic()["TopicArn"]92        queue_url = sqs_create_queue()93        key_name = "bucket-key"94        # connect topic to queue95        sns_create_sqs_subscription(topic_arn, queue_url)96        create_sns_bucket_notification(97            s3_client, sns_client, bucket_name, topic_arn, ["s3:ObjectCreated:*"]98        )99        # trigger the events100        s3_client.put_object(Bucket=bucket_name, Key=key_name, Body="first event")101        s3_client.put_object(Bucket=bucket_name, Key=key_name, Body="second event")102        # collect messages103        messages = sqs_collect_sns_messages(sqs_client, queue_url, 2)104        # asserts105        # first event106        message = messages[0]107        assert message["Type"] == "Notification"108        assert message["TopicArn"] == topic_arn109        assert message["Subject"] == "Amazon S3 Notification"110        event = json.loads(message["Message"])["Records"][0]111        assert event["eventSource"] == "aws:s3"112        assert event["eventName"] == "ObjectCreated:Put"113        assert event["s3"]["bucket"]["name"] == bucket_name114        assert event["s3"]["object"]["key"] == key_name115        assert event["s3"]["object"]["size"] == len("first event")116        # second event117        message = messages[1]...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.