testcontainers-python/modules/localstack/example_basic.py at tmp1 · florinutz/testcontainers-python · GitHub

History

72 lines (56 loc) · 2.28 KB

Raw

import json

import boto3

from testcontainers.localstack import LocalStackContainer

def basic_example():

with LocalStackContainer() as localstack:

# Get endpoint URL

endpoint_url = localstack.get_endpoint_url()

# Create S3 client

s3 = boto3.client(

"s3",

endpoint_url=endpoint_url,

aws_access_key_id="test",

aws_secret_access_key="test",

region_name="us-east-1",

)

# Create SQS client

sqs = boto3.client(

"sqs",

endpoint_url=endpoint_url,

aws_access_key_id="test",

aws_secret_access_key="test",

region_name="us-east-1",

)

# Create S3 bucket

bucket_name = "test-bucket"

s3.create_bucket(Bucket=bucket_name)

print(f"Created S3 bucket: {bucket_name}")

# Upload file to S3

test_data = {"message": "Hello from LocalStack!", "timestamp": "2024-01-01"}

s3.put_object(Bucket=bucket_name, Key="test.json", Body=json.dumps(test_data))

print("Uploaded test.json to S3")

# Create SQS queue

queue_name = "test-queue"

queue = sqs.create_queue(QueueName=queue_name)

queue_url = queue["QueueUrl"]

print(f"Created SQS queue: {queue_name}")

# Send message to SQS

message = {"message": "Test message", "number": 42}

sqs.send_message(QueueUrl=queue_url, MessageBody=json.dumps(message))

print("Sent message to SQS")

# Receive message from SQS

response = sqs.receive_message(QueueUrl=queue_url, MaxNumberOfMessages=1)

if "Messages" in response:

received_message = json.loads(response["Messages"][0]["Body"])

print("\nReceived message from SQS:")

print(json.dumps(received_message, indent=2))

# Delete message

sqs.delete_message(QueueUrl=queue_url, ReceiptHandle=response["Messages"][0]["ReceiptHandle"])

print("Deleted message from queue")

# List S3 objects

objects = s3.list_objects(Bucket=bucket_name)

print("\nS3 bucket contents:")

for obj in objects.get("Contents", []):

print(f"Key: {obj['Key']}, Size: {obj['Size']} bytes")

if __name__ == "__main__":

basic_example()

testcontainers-python/modules/localstack/example_basic.py at tmp1 · florinutz/testcontainers-python · GitHub

Expand file tree

Breadcrumbs

example_basic.py

Latest commit

History

Breadcrumbs

example_basic.py

File metadata and controls