Update new-clients tests to run subset of clients (#281)

Author: trivikr

scripts/generateNewClientTests/config.ts

@@ -0,0 +1 @@
+export const CLIENTS_TO_TEST = ["ACM", "AccessAnalyzer", "Discovery"];

scripts/generateNewClientTests/getClientNamesSortedByPackageName.ts

@@ -1,6 +1,9 @@
 import { CLIENT_PACKAGE_NAMES_MAP } from "../../src/transforms/v2-to-v3/config";
 
-export const getClientNamesSortedByPackageName = () =>
-  Object.keys(CLIENT_PACKAGE_NAMES_MAP).sort((a, b) =>
-    CLIENT_PACKAGE_NAMES_MAP[a].localeCompare(CLIENT_PACKAGE_NAMES_MAP[b])
-  );
+export const getClientNamesSortedByPackageName = (clientsToTest: string[]) =>
+  Object.keys(
+    clientsToTest.reduce((acc, clientName) => {
+      acc[clientName] = CLIENT_PACKAGE_NAMES_MAP[clientName];
+      return acc;
+    }, {})
+  ).sort((a, b) => CLIENT_PACKAGE_NAMES_MAP[a].localeCompare(CLIENT_PACKAGE_NAMES_MAP[b]));

scripts/generateNewClientTests/getGlobalImportInput.ts

@@ -1,10 +1,11 @@
+import { CLIENT_NAMES } from "../../src/transforms/v2-to-v3/config";
 import { getV2ClientsNewExpressionCode } from "./getV2ClientsNewExpressionCode";
 
 export const getGlobalImportInput = (codegenComment: string) => {
   let globalImportInputContent = `${codegenComment}\n`;
 
   globalImportInputContent += `import AWS from "aws-sdk";\n\n`;
-  globalImportInputContent += getV2ClientsNewExpressionCode(`AWS.`);
+  globalImportInputContent += getV2ClientsNewExpressionCode(CLIENT_NAMES, `AWS.`);
 
   return globalImportInputContent;
 };

scripts/generateNewClientTests/getGlobalImportOutput.ts

@@ -1,13 +1,16 @@
+import { CLIENT_NAMES } from "../../src/transforms/v2-to-v3/config";
 import { getClientNamesSortedByPackageName } from "./getClientNamesSortedByPackageName";
 import { getV3ClientsNewExpressionCode } from "./getV3ClientsNewExpressionCode";
 import { getV3PackageImportsCode } from "./getV3PackageImportsCode";
 
 export const getGlobalImportOutput = (codegenComment: string) => {
   let globalImportOutputContent = `${codegenComment}\n`;
 
-  globalImportOutputContent += getV3PackageImportsCode(getClientNamesSortedByPackageName());
+  globalImportOutputContent += getV3PackageImportsCode(
+    getClientNamesSortedByPackageName(CLIENT_NAMES)
+  );
   globalImportOutputContent += `\n`;
-  globalImportOutputContent += getV3ClientsNewExpressionCode();
+  globalImportOutputContent += getV3ClientsNewExpressionCode(CLIENT_NAMES);
 
   return globalImportOutputContent;
 };

scripts/generateNewClientTests/getGlobalRequireInput.ts

@@ -1,10 +1,11 @@
+import { CLIENTS_TO_TEST } from "./config";
 import { getV2ClientsNewExpressionCode } from "./getV2ClientsNewExpressionCode";
 
 export const getGlobalRequireInput = (codegenComment: string) => {
   let globalRequireInputContent = `${codegenComment}\n`;
 
   globalRequireInputContent += `const AWS = require("aws-sdk");\n\n`;
-  globalRequireInputContent += getV2ClientsNewExpressionCode(`AWS.`);
+  globalRequireInputContent += getV2ClientsNewExpressionCode(CLIENTS_TO_TEST, `AWS.`);
 
   return globalRequireInputContent;
 };

scripts/generateNewClientTests/getGlobalRequireOutput.ts

@@ -1,11 +1,12 @@
 import { CLIENT_NAMES_MAP, CLIENT_PACKAGE_NAMES_MAP } from "../../src/transforms/v2-to-v3/config";
+import { CLIENTS_TO_TEST } from "./config";
 import { getClientNamesSortedByPackageName } from "./getClientNamesSortedByPackageName";
 import { getV3ClientsNewExpressionCode } from "./getV3ClientsNewExpressionCode";
 
 export const getGlobalRequireOutput = (codegenComment: string) => {
   let globalRequireOutputContent = `${codegenComment}\n`;
 
-  const sortedClientNames = getClientNamesSortedByPackageName();
+  const sortedClientNames = getClientNamesSortedByPackageName(CLIENTS_TO_TEST);
   globalRequireOutputContent += `const `;
   for (const v2ClientName of sortedClientNames) {
     const v3ClientName = CLIENT_NAMES_MAP[v2ClientName];
@@ -19,7 +20,7 @@ export const getGlobalRequireOutput = (codegenComment: string) => {
       `      `;
   }
   globalRequireOutputContent = globalRequireOutputContent.replace(/,\n {6}$/, ";\n\n");
-  globalRequireOutputContent += getV3ClientsNewExpressionCode();
+  globalRequireOutputContent += getV3ClientsNewExpressionCode(CLIENTS_TO_TEST);
 
   return globalRequireOutputContent;
 };

scripts/generateNewClientTests/getServiceImportDeepInput.ts

@@ -1,14 +1,14 @@
-import { CLIENT_NAMES } from "../../src/transforms/v2-to-v3/config";
+import { CLIENTS_TO_TEST } from "./config";
 import { getV2ClientsNewExpressionCode } from "./getV2ClientsNewExpressionCode";
 
 export const getServiceImportDeepInput = (codegenComment: string) => {
   let serviceImportInputContent = `${codegenComment}\n`;
 
-  for (const clientName of CLIENT_NAMES) {
+  for (const clientName of CLIENTS_TO_TEST) {
     serviceImportInputContent += `import ${clientName} from "aws-sdk/clients/${clientName.toLowerCase()}";\n`;
   }
   serviceImportInputContent += `\n`;
-  serviceImportInputContent += getV2ClientsNewExpressionCode();
+  serviceImportInputContent += getV2ClientsNewExpressionCode(CLIENTS_TO_TEST);
 
   return serviceImportInputContent;
 };

scripts/generateNewClientTests/getServiceImportInput.ts

@@ -1,14 +1,14 @@
-import { CLIENT_NAMES } from "../../src/transforms/v2-to-v3/config";
+import { CLIENTS_TO_TEST } from "./config";
 import { getV2ClientsNewExpressionCode } from "./getV2ClientsNewExpressionCode";
 
 export const getServiceImportInput = (codegenComment: string) => {
   let serviceImportInputContent = `${codegenComment}\n`;
 
-  for (const clientName of CLIENT_NAMES) {
+  for (const clientName of CLIENTS_TO_TEST) {
     serviceImportInputContent += `import { ${clientName} } from "aws-sdk";\n`;
   }
   serviceImportInputContent += `\n`;
-  serviceImportInputContent += getV2ClientsNewExpressionCode();
+  serviceImportInputContent += getV2ClientsNewExpressionCode(CLIENTS_TO_TEST);
 
   return serviceImportInputContent;
 };

scripts/generateNewClientTests/getServiceImportOutput.ts

@@ -1,13 +1,13 @@
-import { CLIENT_NAMES } from "../../src/transforms/v2-to-v3/config";
+import { CLIENTS_TO_TEST } from "./config";
 import { getV3ClientsNewExpressionCode } from "./getV3ClientsNewExpressionCode";
 import { getV3PackageImportsCode } from "./getV3PackageImportsCode";
 
 export const getServiceImportOutput = (codegenComment: string) => {
   let serviceImportOutputContent = `${codegenComment}\n`;
 
-  serviceImportOutputContent += getV3PackageImportsCode(CLIENT_NAMES);
+  serviceImportOutputContent += getV3PackageImportsCode(CLIENTS_TO_TEST);
   serviceImportOutputContent += `\n`;
-  serviceImportOutputContent += getV3ClientsNewExpressionCode();
+  serviceImportOutputContent += getV3ClientsNewExpressionCode(CLIENTS_TO_TEST);
 
   return serviceImportOutputContent;
 };

scripts/generateNewClientTests/getServiceRequireDeepInput.ts

@@ -1,14 +1,14 @@
-import { CLIENT_NAMES } from "../../src/transforms/v2-to-v3/config";
+import { CLIENTS_TO_TEST } from "./config";
 import { getV2ClientsNewExpressionCode } from "./getV2ClientsNewExpressionCode";
 
 export const getServiceRequireDeepInput = (codegenComment: string) => {
   let serviceRequireInputContent = `${codegenComment}\n`;
 
-  for (const clientName of CLIENT_NAMES) {
+  for (const clientName of CLIENTS_TO_TEST) {
     serviceRequireInputContent += `const ${clientName} = require("aws-sdk/clients/${clientName.toLowerCase()}");\n`;
   }
   serviceRequireInputContent += `\n`;
-  serviceRequireInputContent += getV2ClientsNewExpressionCode();
+  serviceRequireInputContent += getV2ClientsNewExpressionCode(CLIENTS_TO_TEST);
 
   return serviceRequireInputContent;
 };