feat: create produce endpoint to write SQS message (#7)

Author: patheard

api/main.py

@@ -2,9 +2,17 @@
 Main API handler that defines all routes.
 """
 
+import json
+import os
+from datetime import datetime
+
+import boto3
 from fastapi import FastAPI
 from mangum import Mangum
 
+MESSAGE_QUEUE_NAME = os.environ.get("MESSAGE_QUEUE_NAME", None)
+REGION = os.environ.get("AWS_REGION", "ca-canada-1")
+
 app = FastAPI(
     title="AWS + FastAPI",
     description="AWS API Gateway, Lambdas and FastAPI (oh my)",
@@ -17,5 +25,21 @@ def hello():
     return {"Hello": "World"}
 
 
+@app.get("/produce")
+def produce():
+    "Produce an SQS message"
+
+    if MESSAGE_QUEUE_NAME is None:
+        return {"error": "Message queue name not set"}
+
+    message = {"date": datetime.now(), "flavour": "delicious"}
+
+    sqs = boto3.resource("sqs", region_name=REGION)
+    queue = sqs.get_queue_by_name(QueueName=MESSAGE_QUEUE_NAME)
+    response = queue.send_message(MessageBody=json.dumps(message))
+
+    return {"message_id": response.get("MessageId")}
+
+
 # Mangum allows us to use Lambdas to process requests
 handler = Mangum(app=app)

api/requirements.txt

@@ -1,2 +1,3 @@
+boto3==1.17.*
 fastapi==0.67.0
 mangum==0.12.1

api/tests/test_main.py

@@ -1,5 +1,34 @@
+import json
 import main
+from unittest.mock import call, MagicMock, patch
 
 
 def test_hello():
     assert main.hello() == {"Hello": "World"}
+
+
+@patch("main.boto3")
+@patch("main.datetime")
+@patch("main.MESSAGE_QUEUE_NAME", "Muffins")
+@patch("main.REGION", "us-east-1")
+def test_produce(MockDateTime, MockBoto):
+    response = MagicMock()
+    response.get.return_value = "42"
+    queue = MagicMock()
+    queue.send_message.return_value = response
+    sqs = MagicMock()
+    sqs.get_queue_by_name.return_value = queue
+    MockBoto.resource.return_value = sqs
+    MockDateTime.now.return_value = "1970-01-01"
+
+    assert main.produce() == {"message_id": "42"}
+
+    MockBoto.resource.assert_called_with("sqs", region_name="us-east-1")
+    sqs.get_queue_by_name.assert_called_with(QueueName="Muffins")
+    queue.send_message.assert_called_with(
+        MessageBody=json.dumps({"date": "1970-01-01", "flavour": "delicious"})
+    )
+
+
+def test_produce_no_queue():
+    assert main.produce() == {"error": "Message queue name not set"}

infra/env/dev/api-gateway/terragrunt.hcl

@@ -21,16 +21,16 @@ dependency "sqs" {
 
   mock_outputs_allowed_terraform_commands = ["init", "fmt", "validate", "plan", "show"]
   mock_outputs = {
-    message_queue_arn = ""
-    message_queue_url = ""
+    message_queue_arn  = ""
+    message_queue_name = ""
   }
 }
 
 inputs = {
   lambda_subnet_ids         = dependency.network.outputs.lambda_subnet_ids
   lambda_security_group_id  = dependency.network.outputs.lambda_security_group_id
   message_queue_arn         = dependency.sqs.outputs.message_queue_arn
-  message_queue_url         = dependency.sqs.outputs.message_queue_url
+  message_queue_name        = dependency.sqs.outputs.message_queue_name
 }
 
 terraform {

infra/modules/api-gateway/lambda.tf

@@ -20,7 +20,8 @@ resource "aws_lambda_function" "api_lambda" {
 
   environment {
     variables = {
-      message_queue_url = var.message_queue_url
+      MESSAGE_QUEUE_NAME = var.message_queue_name
+      REGION             = var.region
     }
   }
 

infra/modules/api-gateway/variables.tf

@@ -13,7 +13,7 @@ variable "message_queue_arn" {
   type        = string
 }
 
-variable "message_queue_url" {
-  description = "URL of the SQS to send messages to"
+variable "message_queue_name" {
+  description = "Name of the SQS to send messages to"
   type        = string
 }

infra/modules/sqs/output.tf

@@ -2,6 +2,10 @@ output "message_queue_arn" {
   value = aws_sqs_queue.message_queue.arn
 }
 
+output "message_queue_name" {
+  value = aws_sqs_queue.message_queue.name
+}
+
 output "message_queue_url" {
   value = aws_sqs_queue.message_queue.url
 }