Added support for DynamoDbAutoGeneratedKey annotation

Author: anasatirbasa

services-custom/dynamodb-enhanced/src/test/java/software/amazon/awssdk/enhanced/dynamodb/functionaltests/AutoGeneratedKeyRecordTest.java

@@ -292,43 +292,6 @@ public void autogenerateKey_onVersionedRecord_setOnPrimaryKey_performsUuidGenera
         }
     }
 
-    @Test
-    public void autogenerateKey_withUpdateBehaviorOnPrimaryKey_preservesKeyOnUpdate() {
-        String tableName = getConcreteTableName("autogen-pk-update-behavior");
-        DynamoDbEnhancedClient client = DynamoDbEnhancedClient.builder()
-                                                              .dynamoDbClient(getDynamoDbClient())
-                                                              .extensions(AutoGeneratedKeyExtension.builder().build())
-                                                              .build();
-
-        TableSchema<PrimaryKeyWithUpdateBehaviorBean> schema = TableSchema.fromBean(PrimaryKeyWithUpdateBehaviorBean.class);
-        DynamoDbTable<PrimaryKeyWithUpdateBehaviorBean> table = client.table(tableName, schema);
-
-        try {
-            table.createTable(r -> r.provisionedThroughput(getDefaultProvisionedThroughput()));
-
-            // First put - should generate UUID for primary key
-            PrimaryKeyWithUpdateBehaviorBean bean = new PrimaryKeyWithUpdateBehaviorBean();
-            bean.setPayload("initial");
-            table.putItem(bean);
-
-            PrimaryKeyWithUpdateBehaviorBean afterPut = table.scan().items().stream().findFirst()
-                    .orElseThrow(() -> new AssertionError("No record found"));
-            String generatedId = afterPut.getId();
-            assertValidUuid(generatedId);
-            assertEquals("initial", afterPut.getPayload());
-
-            // Update - should preserve the generated primary key due to WRITE_IF_NOT_EXISTS
-            afterPut.setPayload("updated");
-            table.updateItem(afterPut);
-
-            PrimaryKeyWithUpdateBehaviorBean afterUpdate = table.getItem(r -> r.key(k -> k.partitionValue(generatedId)));
-            assertThat(afterUpdate.getId()).isEqualTo(generatedId); // Key preserved
-            assertEquals("updated", afterUpdate.getPayload());
-        } finally {
-            deleteTableByName(tableName);
-        }
-    }
-
     /**
      * - createdKey:          GSI SK + @DynamoDbAutoGeneratedKey annotation (UpdateBehaviour is WRITE_IF_NOT_EXISTS)
      * - lastUpdatedKey:      GSI PK + @DynamoDbAutoGeneratedKey annotation (UpdateBehaviour is WRITE_ALWAYS - default value)
@@ -570,31 +533,6 @@ public void setPayload(String payload) {
         }
     }
 
-    @DynamoDbBean
-    public static class PrimaryKeyWithUpdateBehaviorBean {
-        private String id;
-        private String payload;
-
-        @DynamoDbPartitionKey
-        @DynamoDbAutoGeneratedKey
-        @DynamoDbUpdateBehavior(UpdateBehavior.WRITE_IF_NOT_EXISTS)
-        public String getId() {
-            return id;
-        }
-
-        public void setId(String id) {
-            this.id = id;
-        }
-
-        public String getPayload() {
-            return payload;
-        }
-
-        public void setPayload(String payload) {
-            this.payload = payload;
-        }
-    }
-
     private void deleteTableByName(String tableName) {
         getDynamoDbClient().deleteTable(b -> b.tableName(tableName));
     }