test: add additional assertions

aws-amplify · Oct 4, 2024 · 3849ed1 · 3849ed1
1 parent 6ecafeb
commit 3849ed1
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/packages/appsync-modelgen-plugin/src/__tests__/visitors/appsync-swift-visitor.test.ts b/packages/appsync-modelgen-plugin/src/__tests__/visitors/appsync-swift-visitor.test.ts
@@ -3195,7 +3195,10 @@ describe('AppSyncSwiftVisitor', () => {
           context: String
         }
       `;
-      expect(getVisitorPipelinedTransformer(schema, 'MyModel', CodeGenGenerateEnum.code, {}).generate()).toMatchSnapshot();
+      const result = getVisitorPipelinedTransformer(schema, 'MyModel', CodeGenGenerateEnum.code, {}).generate();
+      expect(result).toContain('myCustomTypes = try values.decode([MyCustomType?].self, forKey: .myCustomTypes)');
+      expect(result).toContain('scalarArray = try values.decode([String?].self, forKey: .scalarArray)');
+      expect(result).toMatchSnapshot();
     });
     test('decode optional list with required elements', () => {
       const schema = /* GraphQL */ `
@@ -3219,7 +3222,10 @@ describe('AppSyncSwiftVisitor', () => {
           context: String
         }
       `;
-      expect(getVisitorPipelinedTransformer(schema, 'MyModel', CodeGenGenerateEnum.code, {}).generate()).toMatchSnapshot();
+      const result = getVisitorPipelinedTransformer(schema, 'MyModel', CodeGenGenerateEnum.code, {}).generate();
+      expect(result).toContain('myCustomTypes = try? values.decode([MyCustomType].self, forKey: .myCustomTypes)');
+      expect(result).toContain('scalarArray = try? values.decode([String].self, forKey: .scalarArray)');
+      expect(result).toMatchSnapshot();
     });
     test('decode required list with required elements', () => {
       const schema = /* GraphQL */ `
@@ -3243,7 +3249,10 @@ describe('AppSyncSwiftVisitor', () => {
           context: String
         }
       `;
-      expect(getVisitorPipelinedTransformer(schema, 'MyModel', CodeGenGenerateEnum.code, {}).generate()).toMatchSnapshot();
+      const result = getVisitorPipelinedTransformer(schema, 'MyModel', CodeGenGenerateEnum.code, {}).generate();
+      expect(result).toContain('myCustomTypes = try values.decode([MyCustomType].self, forKey: .myCustomTypes)');
+      expect(result).toContain('scalarArray = try values.decode([String].self, forKey: .scalarArray)');
+      expect(result).toMatchSnapshot();
     });
 
     test('decode optional list with nullable elements', () => {
@@ -3268,7 +3277,10 @@ describe('AppSyncSwiftVisitor', () => {
           context: String
         }
       `;
-      expect(getVisitorPipelinedTransformer(schema, 'MyModel', CodeGenGenerateEnum.code, {}).generate()).toMatchSnapshot();
+      const result = getVisitorPipelinedTransformer(schema, 'MyModel', CodeGenGenerateEnum.code, {}).generate();
+      expect(result).toContain('myCustomTypes = try? values.decode([MyCustomType?].self, forKey: .myCustomTypes)');
+      expect(result).toContain('scalarArray = try? values.decode([String?].self, forKey: .scalarArray)');
+      expect(result).toMatchSnapshot();
     });
   });
 });