Fix breaking legacy table version

viren-nadkarni · viren-nadkarni · commit 921611e287d7 · 2025-10-09T15:22:11.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
@@ -2017,6 +2017,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
     ):
@@ -2051,6 +2054,9 @@ def test_dynamodb_create_table_with_partial_sse_specification(
         assert "SSESpecification" not in result["Table"]
 
     @markers.aws.validated
+    @markers.snapshot.skip_snapshot_verify(
+        paths=["$..KeyMetadata.CurrentKeyMaterialId"]  # Not supported by LS
+    )
     def test_dynamodb_update_table_without_sse_specification_change(
         self, dynamodb_create_table_with_parameters, snapshot, aws_client
     ):
