Merge pull request #642 from aws-amplify/codegen-config-changes

Add maxDepth parameter fetching via Codegen Config getter API

Author: phani-srikar

packages/amplify-codegen/src/codegen-config/index.js

@@ -43,6 +43,7 @@ function getCodegenConfig(projectPath) {
     : path.dirname(path.dirname(includeFiles));
 
   const { generatedFileName } = cfg.amplifyExtension || {};
+  const { maxDepth } = cfg.amplifyExtension || {};
 
   return {
     getGeneratedQueriesPath: function() {
@@ -62,6 +63,12 @@ function getCodegenConfig(projectPath) {
         return;
       }
       return path.normalize(generatedFileName);
+    },
+    getQueryMaxDepth: function() {
+      if (!maxDepth || !Number(maxDepth)) {
+        return;
+      }
+      return Number(maxDepth);
     }
   }
 }

packages/amplify-codegen/tests/codegen-config/getCodegenConfig.test.js

@@ -57,6 +57,13 @@ describe('get codegen configuration', () => {
     expect(getConfigReturn.getGeneratedFragmentsPath()).toEqual(path.join('src', 'graphql', 'fragments'));
   });
 
+  it('returns correct maxDepth as a Number', () => {
+    fs.existsSync = jest.fn().mockReturnValue(true);
+    graphQLConfig.getGraphQLConfig = jest.fn().mockReturnValue(MOCK_CODEGEN_CONFIG);
+    const getConfigReturn = getCodegenConfig(MOCK_PROJECT_ROOT);
+    expect(getConfigReturn.getQueryMaxDepth()).toEqual(4);
+  });
+
   it('uses the includes property if the generated documents path does not exist', () => {
     const configWithoutDocumentsPath = { ...MOCK_CODEGEN_CONFIG };
     delete configWithoutDocumentsPath.config.extensions.amplify.docsFilePath;
@@ -93,4 +100,13 @@ describe('get codegen configuration', () => {
     const getConfigReturn = getCodegenConfig(MOCK_PROJECT_ROOT);
     expect(getConfigReturn.getGeneratedTypesPath()).toBeUndefined();
   });
+
+  it('returns undefined if maxDepth does not exist', () => {
+    const configWithoutMaxDepth = { ...MOCK_CODEGEN_CONFIG };
+    delete configWithoutMaxDepth.config.extensions.amplify.maxDepth;
+    fs.existsSync = jest.fn().mockReturnValue(true);
+    graphQLConfig.getGraphQLConfig = jest.fn().mockReturnValue(configWithoutMaxDepth);
+    const getConfigReturn = getCodegenConfig(MOCK_PROJECT_ROOT);
+    expect(getConfigReturn.getQueryMaxDepth()).toBeUndefined();
+  });
 });