get_delta_table

Author: JamesW1-NHS

lambdas/delta_backend/src/delta.py

@@ -4,10 +4,10 @@
 import time
 from datetime import UTC, datetime, timedelta
 
-import boto3
 from boto3.dynamodb.conditions import Attr
 from botocore.exceptions import ClientError
 
+from common.aws_dynamodb import get_dynamodb_table
 from common.clients import STREAM_NAME, get_sqs_client, logger
 from common.log_firehose import send_log_to_firehose
 from converter import Converter
@@ -30,8 +30,7 @@ def get_delta_table():
     if not delta_table:
         try:
             logger.info("Initializing Delta Table")
-            dynamodb = boto3.resource("dynamodb", region_name)
-            delta_table = dynamodb.Table(delta_table_name)
+            delta_table = get_dynamodb_table(delta_table_name)
         except Exception as e:
             logger.error(f"Error initializing Delta Table: {e}")
             delta_table = None

lambdas/delta_backend/tests/test_delta.py

@@ -641,20 +641,20 @@ def test_returns_table_on_success(self):
         # Should cache the table
         self.assertIs(delta.delta_table, self.mock_delta_table)
 
-    @patch("boto3.resource")
-    def test_returns_cached_table(self, mock_boto3_resource):
+    @patch("delta.get_dynamodb_table")
+    def test_returns_cached_table(self, mock_get_dynamodb_table):
         delta.delta_table = self.mock_delta_table
 
         table = delta.get_delta_table()
         self.assertIs(table, self.mock_delta_table)
-        # Should not call boto3 again
-        mock_boto3_resource.assert_not_called()
+        # Should not call get_dynamodb_table again
+        mock_get_dynamodb_table.assert_not_called()
 
-    # mock boto3.resource to raise an exception
-    @patch("boto3.resource")
-    def test_returns_none_on_exception(self, mock_boto3_resource):
+    # mock get_dynamodb_table to raise an exception
+    @patch("delta.get_dynamodb_table")
+    def test_returns_none_on_exception(self, mock_get_dynamodb_table):
         delta.delta_table = None
-        mock_boto3_resource.side_effect = Exception("fail")
+        mock_get_dynamodb_table.side_effect = Exception("fail")
         table = delta.get_delta_table()
         self.assertIsNone(table)
         self.mock_logger_error.assert_called()