Fix breaking legacy table version

viren-nadkarni · viren-nadkarni · commit e67aa3318a2a · 2025-10-09T14:58:18.000+05:30
diff --git a/localstack-core/localstack/services/dynamodb/provider.py b/localstack-core/localstack/services/dynamodb/provider.py
@@ -852,13 +852,10 @@ def describe_table(
 
         # Set defaults for warm throughput
         if "WarmThroughput" not in table_description:
+            billing_mode = table_definitions.get("BillingMode") if table_definitions else None
             table_description["WarmThroughput"] = {
-                "ReadUnitsPerSecond": 12000
-                if table_definitions.get("BillingMode") == "PAY_PER_REQUEST"
-                else 5,
-                "WriteUnitsPerSecond": 4000
-                if table_definitions.get("BillingMode") == "PAY_PER_REQUEST"
-                else 5,
+                "ReadUnitsPerSecond": 12000 if billing_mode == "PAY_PER_REQUEST" else 5,
+                "WriteUnitsPerSecond": 4000 if billing_mode == "PAY_PER_REQUEST" else 5,
             }
         table_description["WarmThroughput"]["Status"] = (
             table_description.get("TableStatus") or "ACTIVE"
diff --git a/localstack-core/localstack/services/dynamodb/v2/provider.py b/localstack-core/localstack/services/dynamodb/v2/provider.py
@@ -663,9 +663,10 @@ def describe_table(
 
         # Set defaults for warm throughput
         if "WarmThroughput" not in table_description:
+            billing_mode = table_definitions.get("BillingMode") if table_definitions else None
             table_description["WarmThroughput"] = {
-                "ReadUnitsPerSecond": 5,
-                "WriteUnitsPerSecond": 5,
+                "ReadUnitsPerSecond": 12000 if billing_mode == "PAY_PER_REQUEST" else 5,
+                "WriteUnitsPerSecond": 4000 if billing_mode == "PAY_PER_REQUEST" else 5,
             }
         table_description["WarmThroughput"]["Status"] = (
             table_description.get("TableStatus") or "ACTIVE"
diff --git a/tests/aws/services/dynamodb/test_dynamodb.py b/tests/aws/services/dynamodb/test_dynamodb.py
@@ -1992,6 +1992,9 @@ def test_dynamodb_pay_per_request(self, dynamodb_create_table_with_parameters, s
         snapshot.match("Error", e.value)
 
     @markers.aws.only_localstack
+    @markers.snapshot.skip_snapshot_verify(
+        paths=["$..KeyMetadata.CurrentKeyMaterialId"]
+    )  # Not supported by LS
     def test_dynamodb_create_table_with_sse_specification(
         self, dynamodb_create_table_with_parameters, account_id, region_name
     ):
@@ -2017,6 +2020,9 @@ def test_dynamodb_create_table_with_sse_specification(
         assert result["TableDescription"]["SSEDescription"]["KMSMasterKeyArn"] == kms_master_key_arn
 
     @markers.aws.validated
+    @markers.snapshot.skip_snapshot_verify(
+        paths=["$..KeyMetadata.CurrentKeyMaterialId"]
+    )  # Not supported by LS
     def test_dynamodb_create_table_with_partial_sse_specification(
         self, dynamodb_create_table_with_parameters, snapshot, aws_client
     ):
