Add support for fetching test suite configuration from graphql

packages/cypress/src/mode.ts

@@ -135,12 +135,17 @@ function getRepeatCount(stressCount = 10) {
   }
 }
 
-export function getDiagnosticConfig(config: Cypress.PluginConfigOptions): {
+export function getDiagnosticConfig(
+  config: Cypress.PluginConfigOptions,
+  extraEnv: NodeJS.ProcessEnv = {}
+): {
   noRecord: boolean;
   env: NodeJS.ProcessEnv;
 } {
   let noRecord = false;
-  let env: NodeJS.ProcessEnv = {};
+  let env: NodeJS.ProcessEnv = {
+    ...extraEnv,
+  };
 
   const { cypress_repeat_k } = config.env;
   const repeatIndex = cypress_repeat_k ? Number.parseInt(cypress_repeat_k) : undefined;

packages/cypress/src/reporter.ts

@@ -4,6 +4,7 @@ import {
   ReplayReporter,
   TestMetadataV2,
   ReporterError,
+  fetchWorkspaceConfig,
 } from "@replayio/test-utils";
 import debug from "debug";
 
@@ -59,12 +60,12 @@ class CypressReporter {
 
   async authenticate(apiKey: string) {
     this.reporter.setApiKey(apiKey);
-    // TODO: we can fetch diagnostics from the cloud later here
-    this.configureDiagnostics();
+    const { env } = await fetchWorkspaceConfig(apiKey);
+    this.configureDiagnostics(env);
   }
 
-  configureDiagnostics() {
-    this.diagnosticConfig = getDiagnosticConfig(this.config);
+  configureDiagnostics(extraEnv?: NodeJS.ProcessEnv) {
+    this.diagnosticConfig = getDiagnosticConfig(this.config, extraEnv);
 
     // Mix diagnostic env into process env so it can be picked up by test
     // metrics and reported to telemetry

packages/test-utils/src/config.ts

@@ -0,0 +1,68 @@
+import fetch from "node-fetch";
+import dbg from "debug";
+
+const debug = dbg("replay:test-utils:config");
+
+async function query(apiKey: string, name: string, query: string, variables = {}) {
+  const options = {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      Authorization: `Bearer ${apiKey}`,
+    },
+    body: JSON.stringify({
+      query,
+      name,
+      variables,
+    }),
+  };
+
+  const server = process.env.REPLAY_API_SERVER || "https://api.replay.io";
+  debug("Querying %s graphql endpoint", server);
+  const result = await fetch(`${server}/v1/graphql`, options);
+
+  return await result.json();
+}
+
+async function fetchWorkspaceConfig(apiKey: string) {
+  const json = await query(
+    apiKey,
+    "GetWorkspaceConfig",
+    `
+        query GetWorkspaceConfig {
+        auth {
+            workspaces {
+            edges {
+                node {
+                id
+                settings {
+                    features
+                }
+                }
+            }
+            }
+        }
+        }
+  `
+  );
+
+  if (json.errors) {
+    debug("GraphQL failed: %s", json.errors);
+    throw new Error(json.errors[0].message || "Unexpected error");
+  }
+
+  const edges = json.data?.auth.workspaces.edges;
+  if (!edges || edges.length !== 1) {
+    debug("Failed to find workspace: %o", json.data);
+    throw new Error("Failed to find a workspace for the given apiKey");
+  }
+
+  debug("Workspace settings: %o", edges[0].node.settings);
+  const features = edges[0].node.settings.features;
+
+  return {
+    env: features?.testSuites?.env || {},
+  };
+}
+
+export { fetchWorkspaceConfig };

packages/test-utils/src/index.ts

@@ -4,5 +4,6 @@ export type { TestMetadataV1, TestMetadataV2, ReplayReporterConfig } from "./rep
 export { ReporterError } from "./reporter";
 export { pingTestMetrics } from "./metrics";
 export { removeAnsiCodes } from "./terminal";
+export { fetchWorkspaceConfig } from "./config";
 export { ReplayReporter };
 export { getMetadataFilePath, initMetadataFile } from "./metadata";