Attach render test artifacts AWS Device Farm to workflow (#2812)

scripts/aws-device-farm/log-test-spec-output.mjs

@@ -1,47 +0,0 @@
-import { ListArtifactsCommand } from "@aws-sdk/client-device-farm";
-import { getDeviceFarmClient } from "./device-farm-client.mjs";
-
-/**
- * @returns {never}
- */
-function usage() {
-  console.error("Logs test spec output from AWS Device Farm");
-  console.error(`Usage: node ${process.argv.at(1)} RUN_ARN`);
-  process.exit(1);
-}
-
-if (process.argv.length !== 3) usage();
-
-const arn = process.argv.at(2);
-
-if (typeof arn !== 'string') usage();
-
-const deviceFarmClient = getDeviceFarmClient();
-
-/**
- * Looks for the run with the provided ARN and returns the test spec output.
- * 
- * @param {string} arn 
- * @returns string
- */
-async function getTestSpecOutput(arn) {
-  const logs = await deviceFarmClient.send(new ListArtifactsCommand({
-    arn,
-    type: 'FILE'
-  }));
-  for (const artifact of logs.artifacts || []) {
-    if (artifact.type === 'TESTSPEC_OUTPUT' && artifact.url) {
-      return await (await fetch(artifact.url)).text();
-    }
-  }
-  return "";
-}
-
-const testSpecOutput = await getTestSpecOutput(arn);
-
-if (!testSpecOutput) {
-  console.error(`Test spec output for run '${arn}' not found`);
-}
-
-console.log(testSpecOutput);
-

scripts/aws-device-farm/store-test-artifacts.mjs

@@ -0,0 +1,69 @@
+import * as fs from "node:fs";
+import { Readable } from "node:stream";
+import { finished } from "node:stream/promises";
+import * as path from "node:path";
+import * as crypto from "node:crypto";
+
+import { ListArtifactsCommand, ListJobsCommand, ListSuitesCommand } from "@aws-sdk/client-device-farm";
+import { getDeviceFarmClient } from "./device-farm-client.mjs";
+
+/**
+ * @returns {never}
+ */
+function usage() {
+  console.error("Stores artifacts from AWS Device Farm run");
+  console.error(`Usage: node ${process.argv.at(1)} RUN_ARN OUTPUT_DIR`);
+  process.exit(1);
+}
+
+if (process.argv.length !== 4) usage();
+
+const arn = process.argv.at(2);
+const outputDirArg = process.argv.at(3);
+
+if (typeof arn !== 'string') usage();
+if (typeof outputDirArg !== 'string') usage();
+
+const outputDir = outputDirArg;
+
+if (!fs.existsSync(outputDir)) {
+  console.error("Output dir does not exist");
+  process.exit(1);
+}
+
+const deviceFarmClient = getDeviceFarmClient();
+
+/**
+ * Looks for the run with the provided ARN and returns the test spec output.
+ * 
+ * @param {string} arn 
+ * @returns string
+ */
+async function getTestSpecOutput(arn) {
+  const jobs = await deviceFarmClient.send(new ListJobsCommand({
+    arn
+  }));
+  
+  await Promise.all((jobs.jobs || []).map(async (job) => {
+    const suites = await deviceFarmClient.send(new ListSuitesCommand({arn: job.arn}));
+    await Promise.all((suites.suites || []).map(async (suite) => {
+      const artifacts = await deviceFarmClient.send(new ListArtifactsCommand({
+        arn: suite.arn,
+        type: 'FILE'
+      }));
+      await Promise.all((artifacts.artifacts || []).map(async (artifact) => {
+        if (!artifact.name || !artifact.url || !artifact.type) return;
+        if (['TESTSPEC_OUTPUT', 'CUSTOMER_ARTIFACT', 'CUSTOMER_ARTIFACT_LOG', 'DEVICE_LOG'].includes(artifact.type)) {
+          const filename = `${artifact.name.replaceAll(' ', '_')}-${crypto.randomBytes(10).toString('hex')}.${artifact.extension}`;
+          const res = await fetch(artifact.url);
+          if (!res.ok || !res.body) return;
+          const destination = path.resolve(outputDir, filename);
+          const fileStream = fs.createWriteStream(destination, { flags: 'wx' });
+          await finished(Readable.fromWeb(/** @type {any} **/ (res.body)).pipe(fileStream));
+        }
+      }));
+    }));
+  }));
+}
+
+await getTestSpecOutput(arn);