File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change 11
11
from mangum import Mangum
12
12
13
13
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" )
15
15
16
16
app = FastAPI (
17
17
title = "AWS + FastAPI" ,
@@ -32,7 +32,7 @@ def produce():
32
32
if MESSAGE_QUEUE_NAME is None :
33
33
return {"error" : "Message queue name not set" }
34
34
35
- message = {"date" : datetime .now (), "flavour" : "delicious" }
35
+ message = {"date" : datetime .now (). isoformat () , "flavour" : "delicious" }
36
36
37
37
sqs = boto3 .resource ("sqs" , region_name = REGION )
38
38
queue = sqs .get_queue_by_name (QueueName = MESSAGE_QUEUE_NAME )
Original file line number Diff line number Diff line change @@ -19,7 +19,7 @@ def test_produce(MockDateTime, MockBoto):
19
19
sqs = MagicMock ()
20
20
sqs .get_queue_by_name .return_value = queue
21
21
MockBoto .resource .return_value = sqs
22
- MockDateTime .now .return_value = "1970-01-01"
22
+ MockDateTime .now .return_value . isoformat . return_value = "1970-01-01"
23
23
24
24
assert main .produce () == {"message_id" : "42" }
25
25
@@ -30,5 +30,6 @@ def test_produce(MockDateTime, MockBoto):
30
30
)
31
31
32
32
33
+ @patch ("main.MESSAGE_QUEUE_NAME" , None )
33
34
def test_produce_no_queue ():
34
35
assert main .produce () == {"error" : "Message queue name not set" }
You can’t perform that action at this time.
0 commit comments