Skip to content

Commit d383ada

Browse files
committed
fix: region name and SQS message date format
1 parent 47e555f commit d383ada

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

api/main.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from mangum import Mangum
1212

1313
MESSAGE_QUEUE_NAME = os.environ.get("MESSAGE_QUEUE_NAME", None)
14-
REGION = os.environ.get("AWS_REGION", "ca-canada-1")
14+
REGION = os.environ.get("AWS_REGION", "ca-central-1")
1515

1616
app = FastAPI(
1717
title="AWS + FastAPI",
@@ -32,7 +32,7 @@ def produce():
3232
if MESSAGE_QUEUE_NAME is None:
3333
return {"error": "Message queue name not set"}
3434

35-
message = {"date": datetime.now(), "flavour": "delicious"}
35+
message = {"date": datetime.now().isoformat(), "flavour": "delicious"}
3636

3737
sqs = boto3.resource("sqs", region_name=REGION)
3838
queue = sqs.get_queue_by_name(QueueName=MESSAGE_QUEUE_NAME)

api/tests/test_main.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def test_produce(MockDateTime, MockBoto):
1919
sqs = MagicMock()
2020
sqs.get_queue_by_name.return_value = queue
2121
MockBoto.resource.return_value = sqs
22-
MockDateTime.now.return_value = "1970-01-01"
22+
MockDateTime.now.return_value.isoformat.return_value = "1970-01-01"
2323

2424
assert main.produce() == {"message_id": "42"}
2525

@@ -30,5 +30,6 @@ def test_produce(MockDateTime, MockBoto):
3030
)
3131

3232

33+
@patch("main.MESSAGE_QUEUE_NAME", None)
3334
def test_produce_no_queue():
3435
assert main.produce() == {"error": "Message queue name not set"}

0 commit comments

Comments
 (0)