tests(codecatalyst): assert dev environment vars #3511

Author: JadenSimon

src/testE2E/codecatalyst/client.test.ts

@@ -29,6 +29,7 @@ import { captureEventOnce } from '../../test/testUtil'
 import { toStream } from '../../shared/utilities/collectionUtils'
 import { toCollection } from '../../shared/utilities/asyncCollection'
 import { getLogger } from '../../shared/logger'
+import { isAwsError } from '../../shared/errors'
 
 let spaceName: CodeCatalystOrg['name']
 let projectName: CodeCatalystProject['name']
@@ -238,6 +239,30 @@ describe('Test how this codebase uses the CodeCatalyst API', function () {
             assert(lsOutput.split('\n').includes(expectedFile))
         })
 
+        it('has environment variables that identify the Dev Environment', async function () {
+            const { SessionProcess, hostname, sshPath } = await prepareDevEnvConnection(client, {
+                ...defaultDevEnv,
+            })
+
+            const output = (
+                await new SessionProcess(sshPath, [hostname, 'printenv', '|', 'grep', '__DEV_ENVIRONMENT']).run()
+            ).stdout
+
+            const parsed: Record<string, string> = {}
+            for (const line of output.split('\n')) {
+                const [_, key, value] = line.match(/(.*)=(.*)/) ?? []
+                if (!key || !value) {
+                    throw new Error(`Failed to parse environment variables from dev env`)
+                }
+
+                parsed[key] = value
+            }
+
+            assert.deepStrictEqual(parsed['__DEV_ENVIRONMENT_ID'], defaultDevEnv.id)
+            assert.deepStrictEqual(parsed['__DEV_ENVIRONMENT_PROJECT_NAME'], defaultDevEnv.project.name)
+            assert.deepStrictEqual(parsed['__DEV_ENVIRONMENT_ORGANIZATION_NAME'], defaultDevEnv.org.name)
+        })
+
         /**
          * IMPORTANT: This test takes a while to run. Stopping a Dev Env takes time.
          */
@@ -419,11 +444,22 @@ describe('Test how this codebase uses the CodeCatalyst API', function () {
         spaceName: CodeCatalystOrg['name'],
         projectName: CodeCatalystProject['name'] = 'aws-vscode-toolkit-integ-test-project'
     ): Promise<CodeCatalystProject> {
-        return client.createProject({
-            spaceName,
-            displayName: projectName,
-            description: 'This project is autogenerated by the AWS Toolkit VSCode Integ Test.',
-        })
+        try {
+            return await client.createProject({
+                spaceName,
+                displayName: projectName,
+                description: 'This project is autogenerated by the AWS Toolkit VSCode Integ Test.',
+            })
+        } catch (e) {
+            if (isAwsError(e) && e.code === 'ConflictException') {
+                return client.getProject({
+                    spaceName,
+                    name: projectName,
+                })
+            }
+
+            throw e
+        }
     }
 
     /**