Add a test for partition key rename

Author: lauzadis

hll/dynamodb-mapper/dynamodb-mapper-schema-generator-plugin/src/test/kotlin/aws/sdk/kotlin/hll/dynamodbmapper/plugins/SchemaGeneratorPluginTest.kt

@@ -552,4 +552,28 @@ class SchemaGeneratorPluginTest {
         val testResult = runner.withArguments("test").build()
         assertContains(setOf(TaskOutcome.SUCCESS, TaskOutcome.UP_TO_DATE), testResult.task(":test")?.outcome)
     }
+
+    @Test
+    fun testRenamedPartitionKey() {
+        createClassFile("RenamedPartitionKey")
+
+        val result = runner.build()
+        assertContains(setOf(TaskOutcome.SUCCESS, TaskOutcome.UP_TO_DATE), result.task(":build")?.outcome)
+
+        val schemaFile = File(testProjectDir, "build/generated/ksp/main/kotlin/org/example/dynamodbmapper/generatedschemas/RenamedPartitionKeySchema.kt")
+        assertTrue(schemaFile.exists())
+
+        val schemaContents = schemaFile.readText()
+
+        // Schema should use the renamed partition key
+        assertContains(
+            schemaContents,
+            """
+        object RenamedPartitionKeySchema : ItemSchema.PartitionKey<RenamedPartitionKey, Int> {
+            override val converter : RenamedPartitionKeyConverter = RenamedPartitionKeyConverter
+            override val partitionKey: KeySpec<Number> = aws.sdk.kotlin.hll.dynamodbmapper.items.KeySpec.Number("user_id")
+        }
+            """.trimIndent(),
+        )
+    }
 }

hll/dynamodb-mapper/dynamodb-mapper-schema-generator-plugin/src/test/resources/RenamedPartitionKey.kt

@@ -0,0 +1,16 @@
+package org.example
+
+import aws.sdk.kotlin.hll.dynamodbmapper.DynamoDbAttribute
+import aws.sdk.kotlin.hll.dynamodbmapper.DynamoDbItem
+import aws.sdk.kotlin.hll.dynamodbmapper.DynamoDbPartitionKey
+
+@DynamoDbItem
+public data class RenamedPartitionKey(
+    @DynamoDbPartitionKey
+    @DynamoDbAttribute("user_id")
+    var id: Int,
+
+    @DynamoDbAttribute("fName") var givenName: String,
+    @DynamoDbAttribute("lName") var surname: String,
+    var age: Int,
+)