add tests

Author: rushmash91

Diff for: apis/v1alpha1/ack-generate-metadata.yaml

@@ -1,13 +1,13 @@
 ack_generate_info:
-  build_date: "2025-03-12T23:25:20Z"
-  build_hash: 5645e51ed3c413f616e1b929f1527a5139c44198
+  build_date: "2025-03-14T22:49:47Z"
+  build_hash: 0ab258c51775fd52af32e649da00d57373b20869
   go_version: go1.24.0
-  version: v0.43.2-3-g5645e51
+  version: v0.43.2-6-g0ab258c
 api_directory_checksum: 0d5b95bdbe63c6cfc495149b7a86440c4a5fb33a
 api_version: v1alpha1
 aws_sdk_go_version: v1.32.6
 generator_config_info:
-  file_checksum: 48323d41b9cb7ade08c8a87367846cccdd1f6409
+  file_checksum: 85db04e11e016a28a99058aab04767cf8345242e
   original_file_name: generator.yaml
 last_modification:
   reason: API generation

Diff for: apis/v1alpha1/generator.yaml

@@ -80,6 +80,7 @@ resources:
           code: ResourceNotFoundException
       terminal_codes:
         - InvalidParameter
+        - ValidationException
     update_operation:
       custom_method_name: customUpdateTable
     hooks:

Diff for: generator.yaml

@@ -80,6 +80,7 @@ resources:
           code: ResourceNotFoundException
       terminal_codes:
         - InvalidParameter
+        - ValidationException
     update_operation:
       custom_method_name: customUpdateTable
     hooks:

Diff for: pkg/resource/table/hooks.go

@@ -78,7 +78,7 @@ var (
 	)
 	requeueWaitWhileUpdating = ackrequeue.NeededAfter(
 		ErrTableUpdating,
-		5*time.Second,
+		10*time.Second,
 	)
 	requeueWaitGSIReady = ackrequeue.NeededAfter(
 		ErrTableGSIsUpdating,

Diff for: pkg/resource/table/hooks_replica_updates.go

@@ -7,8 +7,8 @@ import (
 	ackerr "github.com/aws-controllers-k8s/runtime/pkg/errors"
 	ackrtlog "github.com/aws-controllers-k8s/runtime/pkg/runtime/log"
 
-	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws-controllers-k8s/dynamodb-controller/apis/v1alpha1"
+	"github.com/aws/aws-sdk-go-v2/aws"
 	svcsdk "github.com/aws/aws-sdk-go-v2/service/dynamodb"
 	svcsdktypes "github.com/aws/aws-sdk-go-v2/service/dynamodb/types"
 )

Diff for: pkg/resource/table/manager.go

@@ -5,7 +5,6 @@
 	}
 	// Check if replicas were specified during creation
 	if desired.ko.Spec.Replicas != nil && len(desired.ko.Spec.Replicas) > 0 {
-		// Copy the replica configuration to the new resource
 		ko.Spec.Replicas = desired.ko.Spec.Replicas
 		
 		// Return with a requeue to process replica updates

Diff for: pkg/resource/table/sdk.go

@@ -7,19 +7,13 @@
 
 	// If there are replicas, we need to remove them before deleting the table
 	if r.ko.Spec.Replicas != nil && len(r.ko.Spec.Replicas) > 0 {
-		// Create a desired state with no replicas
 		desired := &resource{
 			ko: r.ko.DeepCopy(),
 		}
 		desired.ko.Spec.Replicas = nil
 
-		// Call syncReplicaUpdates to remove all replicas
 		err := rm.syncReplicaUpdates(ctx, r, desired)
 		if err != nil {
-			if err == requeueWaitWhileUpdating {
-				// This is expected - we need to wait for the replica removal to complete
-				return nil, err
-			}
 			return nil, err
 		}
 	}

Diff for: pkg/resource/table/tags.go

@@ -80,12 +80,6 @@
 			} else {
 				f12elem.ReplicaInaccessibleDateTime = nil
 			}
-			if f12iter.ReplicaTableClassSummary != nil && f12iter.ReplicaTableClassSummary.TableClass != "" {
-				f12elem.ReplicaTableClassSummary.TableClass = aws.String(string(f12iter.ReplicaTableClassSummary.TableClass))
-				f12elem.ReplicaTableClassSummary.LastUpdateDateTime = &metav1.Time{Time: *f12iter.ReplicaTableClassSummary.LastUpdateDateTime}
-			} else {
-				f12elem.ReplicaTableClassSummary.TableClass = aws.String("STANDARD")
-			}
 			f12 = append(f12, f12elem)
 		}
 		ko.Status.ReplicasDescriptions = f12

Diff for: templates/hooks/table/sdk_create_post_set_output.go.tpl

@@ -0,0 +1,23 @@
+apiVersion: dynamodb.services.k8s.aws/v1alpha1
+kind: Table
+metadata:
+  name: $TABLE_NAME
+spec:
+  tableName: $TABLE_NAME
+  tableClass: STANDARD
+  attributeDefinitions:
+    - attributeName: PK
+      attributeType: S
+    - attributeName: SK
+      attributeType: S
+  keySchema:
+    - attributeName: PK
+      keyType: HASH
+    - attributeName: SK
+      keyType: RANGE
+  billingMode: PAY_PER_REQUEST
+  streamSpecification:
+    streamEnabled: true
+    streamViewType: "NEW_AND_OLD_IMAGES"
+  replicas:
+    - regionName: $REPLICA_REGION_1

Diff for: templates/hooks/table/sdk_delete_pre_build_request.go.tpl

@@ -0,0 +1,20 @@
+apiVersion: dynamodb.services.k8s.aws/v1alpha1
+kind: Table
+metadata:
+  name: $TABLE_NAME
+spec:
+  tableName: $TABLE_NAME
+  tableClass: STANDARD
+  attributeDefinitions:
+    - attributeName: PK
+      attributeType: S
+    - attributeName: SK
+      attributeType: S
+  keySchema:
+    - attributeName: PK
+      keyType: HASH
+    - attributeName: SK
+      keyType: RANGE
+  billingMode: PAY_PER_REQUEST
+  replicas:
+    - regionName: $REPLICA_REGION_1

Diff for: templates/hooks/table/sdk_read_one_post_set_output.go.tpl

@@ -260,3 +260,61 @@ def get_point_in_time_recovery_enabled(table_name):
         return resp['ContinuousBackupsDescription']['PointInTimeRecoveryDescription']['PointInTimeRecoveryStatus'] == 'ENABLED'
     except c.exceptions.ResourceNotFoundException:
         return None
+
+
+class ReplicaMatcher:
+    def __init__(self, expected_regions):
+        self.expected_regions = expected_regions
+
+    def __call__(self, record: dict) -> bool:
+        if 'Replicas' not in record:
+            return False
+
+        actual_regions = set()
+        for replica in record['Replicas']:
+            if 'RegionName' in replica:
+                actual_regions.add(replica['RegionName'])
+
+        return set(self.expected_regions) == actual_regions
+
+
+def replicas_match(expected_regions) -> TableMatchFunc:
+    return ReplicaMatcher(expected_regions)
+
+
+class ReplicaStatusMatcher:
+    def __init__(self, region, status):
+        self.region = region
+        self.status = status
+
+    def __call__(self, record: dict) -> bool:
+        if 'Replicas' not in record:
+            return False
+
+        for replica in record['Replicas']:
+            if 'RegionName' in replica and replica['RegionName'] == self.region:
+                return 'ReplicaStatus' in replica and replica['ReplicaStatus'] == self.status
+
+        return False
+
+
+def replica_status_matches(region, status) -> TableMatchFunc:
+    return ReplicaStatusMatcher(region, status)
+
+
+def get_replicas(table_name):
+    """Returns the replicas for a DynamoDB table
+
+    Args:
+        table_name: the name of the DynamoDB table to get replicas for
+    Returns:
+        A list of replicas or None if the table doesn't exist
+    """
+    dynamodb = boto3.client('dynamodb')
+    try:
+        response = dynamodb.describe_table(TableName=table_name)
+        if 'Table' in response and 'Replicas' in response['Table']:
+            return response['Table']['Replicas']
+        return []
+    except dynamodb.exceptions.ResourceNotFoundException:
+        return None