@aws-amplify/graphql-model-transformer 1.3.3-alpha.6 → 1.3.3-cb-test-beta.0

package/src/__tests__/model-transformer-override.test.ts

@@ -0,0 +1,73 @@
+import { ModelTransformer } from '@aws-amplify/graphql-model-transformer';
+import { GraphQLTransform } from '@aws-amplify/graphql-transformer-core';
+import path from 'path';
+import { stateManager } from '@aws-amplify/amplify-cli-core';
+
+jest.spyOn(stateManager, 'getLocalEnvInfo').mockReturnValue({ envName: 'testEnvName' });
+jest.spyOn(stateManager, 'getProjectConfig').mockReturnValue({ projectName: 'testProjectName' });
+
+const featureFlags = {
+  getBoolean: jest.fn(),
+  getNumber: jest.fn(),
+  getObject: jest.fn(),
+};
+
+describe('ModelTransformer: ', () => {
+  it('should override  model objects when given override config', () => {
+    const validSchema = `
+      type Post @model {
+        id: ID!
+        comments: [Comment]
+      }
+      type Comment @model{
+        id: String!
+        text: String!
+      }
+    `;
+    const transformer = new GraphQLTransform({
+      transformers: [new ModelTransformer()],
+      overrideConfig: {
+        overrideDir: path.join(__dirname, 'overrides'),
+        overrideFlag: true,
+        resourceName: 'myResource',
+      },
+      featureFlags,
+    });
+    const out = transformer.transform(validSchema);
+    expect(out).toBeDefined();
+    const postStack = out.stacks.Post;
+    const commentStack = out.stacks.Comment;
+
+    expect(postStack).toMatchSnapshot();
+    expect(commentStack).toMatchSnapshot();
+  });
+
+  it('should not override model objects when override file does not exist', () => {
+    const validSchema = `
+      type Post @model {
+        id: ID!
+        comments: [Comment]
+      }
+      type Comment @model{
+        id: String!
+        text: String!
+      }
+    `;
+    const transformer = new GraphQLTransform({
+      transformers: [new ModelTransformer()],
+      overrideConfig: {
+        overrideDir: path.join(__dirname, 'non-existing-override-directory'),
+        overrideFlag: true,
+        resourceName: 'myResource',
+      },
+      featureFlags,
+    });
+    const out = transformer.transform(validSchema);
+    expect(out).toBeDefined();
+    const postStack = out.stacks.Post;
+    const commentStack = out.stacks.Comment;
+
+    expect(postStack).toMatchSnapshot();
+    expect(commentStack).toMatchSnapshot();
+  });
+});

package/src/__tests__/overrides/build/override.js

@@ -0,0 +1,16 @@
+function override(resource, amplifyProjectInfo) {
+  resource.api.GraphQLAPI.xrayEnabled = true;
+  resource.models['Post'].modelDDBTable.billingMode = 'PROVISIONED';
+  resource.models['Comment'].modelDDBTable.billingMode = 'PROVISIONED';
+  // override resolver
+  resource.models['Post'].resolvers['subscriptionOnUpdatePostResolver'].requestMappingTemplate = 'mockTemplate';
+
+  if (amplifyProjectInfo.envName != 'testEnvName') {
+    throw new Error(`Unexpected envName: ${amplifyProjectInfo.envName}`);
+  }
+
+  if (amplifyProjectInfo.projectName != 'testProjectName') {
+    throw new Error(`Unexpected envName: ${amplifyProjectInfo.envName}`);
+  }
+}
+exports.override = override;