feat: allow configuring atmosphere service for cli integ tests (#837)

* mid work

* mid work

Author: iliapolo

API.md

@@ -2,6 +2,25 @@ import { Component, github, javascript } from 'projen';
 
 const NOT_FLAGGED_EXPR = "!contains(github.event.pull_request.labels.*.name, 'pr/exempt-integ-test')";
 
+/**
+ * Options for atmosphere service usage.
+ */
+export interface AtmosphereOptions {
+  /**
+   * Atmosphere service endpoint.
+   */
+  readonly endpoint: string;
+  /**
+   * Which pool to retrieve environments from.
+   */
+  readonly pool: string;
+  /**
+   * OIDC role to assume prior to using atmosphere. Must be allow listed
+   * on the service endpoint.
+   */
+  readonly oidcRoleArn: string;
+}
+
 export interface CdkCliIntegTestsWorkflowProps {
   /**
    * Runners for the workflow
@@ -49,6 +68,13 @@ export interface CdkCliIntegTestsWorkflowProps {
    * @default - No upstream versions
    */
   readonly allowUpstreamVersions?: string[];
+
+  /**
+   * Enable atmosphere service to retrieve AWS test environments.
+   *
+   * @default - atmosphere is not used
+   */
+  readonly enableAtmosphere?: AtmosphereOptions;
 }
 
 /**
@@ -286,7 +312,7 @@ export class CdkCliIntegTestsWorkflow extends Component {
             'aws-region': 'us-east-1',
             'role-duration-seconds': 4 * 60 * 60,
             // Expect this in Environment Variables
-            'role-to-assume': '${{ vars.AWS_ROLE_TO_ASSUME_FOR_TESTING }}',
+            'role-to-assume': props.enableAtmosphere ? props.enableAtmosphere.oidcRoleArn : '${{ vars.AWS_ROLE_TO_ASSUME_FOR_TESTING }}',
             'role-session-name': 'run-tests@aws-cdk-cli-integ',
             'output-credentials': true,
           },
@@ -361,13 +387,19 @@ export class CdkCliIntegTestsWorkflow extends Component {
             'bin/run-suite --use-cli-release=${{ steps.versions.outputs.cli_version }} --framework-version=${{ steps.versions.outputs.lib_version }} ${{ matrix.suite }}',
           ].join('\n'),
           env: {
-            // Concurrency only for long-running cli-integ-tests
-            JEST_TEST_CONCURRENT: "${{ matrix.suite == 'cli-integ-tests' && 'true' || 'false' }}",
             JSII_SILENCE_WARNING_DEPRECATED_NODE_VERSION: 'true',
             JSII_SILENCE_WARNING_UNTESTED_NODE_VERSION: 'true',
             JSII_SILENCE_WARNING_KNOWN_BROKEN_NODE_VERSION: 'true',
             DOCKERHUB_DISABLED: 'true',
-            AWS_REGIONS: ['us-east-2', 'eu-west-1', 'eu-north-1', 'ap-northeast-1', 'ap-south-1'].join(','),
+            ...(props.enableAtmosphere ?
+              {
+                CDK_INTEG_ATMOSPHERE_ENABLED: 'true',
+                CDK_INTEG_ATMOSPHERE_ENDPOINT: props.enableAtmosphere.endpoint,
+                CDK_INTEG_ATMOSPHERE_POOL: props.enableAtmosphere.pool,
+              } :
+              {
+                AWS_REGIONS: ['us-east-2', 'eu-west-1', 'eu-north-1', 'ap-northeast-1', 'ap-south-1'].join(','),
+              }),
             CDK_MAJOR_VERSION: '2',
             RELEASE_TAG: 'latest',
             GITHUB_TOKEN: '${{ secrets.GITHUB_TOKEN }}',