test: add second ref test schema

aws-amplify · Apr 10, 2024 · 22a0149 · 22a0149
1 parent a9d9ccf
commit 22a0149
Show file tree

Hide file tree

Showing 3 changed files with 75 additions and 0 deletions.
diff --git a/packages/amplify-codegen-e2e-tests/src/__tests__/build-app-android.test.ts b/packages/amplify-codegen-e2e-tests/src/__tests__/build-app-android.test.ts
@@ -103,6 +103,31 @@ describe('build app - Android', () => {
     `,
       },
     ],
+    [
+      'references-with-sort-key',
+      {
+        description: '@hasMany using references and @primaryKey using sortKeyFields',
+        transformerVersion: TransformerVersion.v2,
+        supportedPlatforms: TransformerPlatform.all,
+        sdl: `
+      type Primary @model {
+        tenantId: ID! @primaryKey(sortKeyFields: ["instanceId", "recordId"])
+        instanceId: ID!
+        recordId: ID!
+        content: String
+        related: [Related!] @hasMany(references: ["primaryTenantId", "primaryInstanceId", "primaryRecordId"])
+      }
+
+      type Related @model {
+        content: String
+        primaryTenantId: ID!
+        primaryInstanceId: ID!
+        primaryRecordId: ID!
+        primary: Primary @belongsTo(references: ["primaryTenantId", "primaryInstanceId", "primaryRecordId"])
+      }
+    `,
+      },
+    ],
   ]).forEach(([schemaName, schema]) => {
     // @ts-ignore
     const testName = `builds with ${schemaName}: ${schema.description}`;

diff --git a/packages/amplify-codegen-e2e-tests/src/__tests__/build-app-swift.test.ts b/packages/amplify-codegen-e2e-tests/src/__tests__/build-app-swift.test.ts
@@ -114,6 +114,31 @@ describe('build app - Swift', () => {
     `,
       },
     ],
+    [
+      'references-with-sort-key',
+      {
+        description: '@hasMany using references and @primaryKey using sortKeyFields',
+        transformerVersion: TransformerVersion.v2,
+        supportedPlatforms: TransformerPlatform.all,
+        sdl: `
+          type Primary @model {
+            tenantId: ID! @primaryKey(sortKeyFields: ["instanceId", "recordId"])
+            instanceId: ID!
+            recordId: ID!
+            content: String
+            related: [Related!] @hasMany(references: ["primaryTenantId", "primaryInstanceId", "primaryRecordId"])
+          }
+          
+          type Related @model {
+            content: String
+            primaryTenantId: ID!
+            primaryInstanceId: ID!
+            primaryRecordId: ID!
+            primary: Primary @belongsTo(references: ["primaryTenantId", "primaryInstanceId", "primaryRecordId"])
+          }
+        `,
+      },
+    ],
   ]).forEach(([schemaName, schema]) => {
     // @ts-ignore
     const testName = `builds with ${schemaName}: ${schema.description}`;

diff --git a/packages/amplify-codegen-e2e-tests/src/__tests__/build-app-ts.test.ts b/packages/amplify-codegen-e2e-tests/src/__tests__/build-app-ts.test.ts
@@ -93,6 +93,31 @@ describe('build app - JS', () => {
     `,
       },
     ],
+    [
+      'references-with-sort-key',
+      {
+        description: '@hasMany using references and @primaryKey using sortKeyFields',
+        transformerVersion: TransformerVersion.v2,
+        supportedPlatforms: TransformerPlatform.all,
+        sdl: `
+          type Primary @model {
+            tenantId: ID! @primaryKey(sortKeyFields: ["instanceId", "recordId"])
+            instanceId: ID!
+            recordId: ID!
+            content: String
+            related: [Related!] @hasMany(references: ["primaryTenantId", "primaryInstanceId", "primaryRecordId"])
+          }
+
+          type Related @model {
+            content: String
+            primaryTenantId: ID!
+            primaryInstanceId: ID!
+            primaryRecordId: ID!
+            primary: Primary @belongsTo(references: ["primaryTenantId", "primaryInstanceId", "primaryRecordId"])
+          }
+        `,
+      },
+    ],
   ]).forEach(([schemaName, schema]) => {
     const testName = `builds with ${schemaName}: ${(schema as any).description}`;
     const testFunction = async () => {