🐛 Preserve cloud build context in CLI runs

Author: Robdel12

clients/storybook/src/index.js

@@ -128,7 +128,7 @@ export async function run(storybookPath, options = {}, context = {}) {
           branch = gitInfo.branch;
           commit = gitInfo.commit;
           message = gitInfo.message;
-          buildName = gitInfo.buildName;
+          buildName = vizzlyConfig?.build?.name || gitInfo.buildName;
           pullRequestNumber = gitInfo.prNumber;
         } else {
           // Fallback for older CLI versions - use environment variables
@@ -138,7 +138,8 @@ export async function run(storybookPath, options = {}, context = {}) {
           branch = process.env.VIZZLY_BRANCH || 'main';
           commit = process.env.VIZZLY_COMMIT_SHA || undefined;
           message = process.env.VIZZLY_COMMIT_MESSAGE || undefined;
-          buildName = `Storybook ${new Date().toISOString()}`;
+          buildName =
+            vizzlyConfig?.build?.name || `Storybook ${new Date().toISOString()}`;
           pullRequestNumber = process.env.VIZZLY_PR_NUMBER
             ? parseInt(process.env.VIZZLY_PR_NUMBER, 10)
             : undefined;

src/services/test-runner.js

@@ -18,6 +18,7 @@ import { VizzlyError } from '../errors/vizzly-error.js';
 import {
   cancelTests,
   createBuild,
+  fetchBuildUrl,
   finalizeBuild,
   initializeDaemon,
   runTests,
@@ -95,26 +96,44 @@ export class TestRunner extends EventEmitter {
   }
 
   async createBuild(options, tdd) {
+    let createdApiBuild = null;
     let buildId = await createBuild({
       runOptions: options,
       tdd,
       config: this.config,
       deps: {
         buildManager: this.buildManager,
         createApiClient: this.deps.createApiClient,
-        createApiBuild: this.deps.createApiBuild,
+        createApiBuild: async (client, payload) => {
+          createdApiBuild = await this.deps.createApiBuild(client, payload);
+          return createdApiBuild;
+        },
         output: this.deps.output,
       },
     });
 
     if (!tdd && buildId) {
+      let buildUrl =
+        createdApiBuild?.url ||
+        (await fetchBuildUrl({
+          buildId,
+          config: this.config,
+          deps: {
+            createApiClient: this.deps.createApiClient,
+            getBuild: this.deps.getBuild,
+            output: this.deps.output,
+          },
+        }));
+
       let writeSession = this.deps.writeSession || defaultWriteSession;
       writeSession({
         buildId,
         branch: options?.branch,
         commit: options?.commit,
         parallelId: options?.parallelId,
       });
+
+      this.emit('build-created', { buildId, url: buildUrl });
     }
 
     return buildId;

tests/services/test-runner.test.js

@@ -8,8 +8,14 @@ function createDeps(overrides = {}) {
       throw new Error('spawn should not be called in createBuild tests');
     },
     createApiClient: () => ({}),
-    createApiBuild: async () => ({ id: 'api-build-123' }),
-    getBuild: async () => ({ id: 'api-build-123' }),
+    createApiBuild: async () => ({
+      id: 'api-build-123',
+      url: 'https://app.vizzly.dev/org/project/builds/api-build-123',
+    }),
+    getBuild: async () => ({
+      id: 'api-build-123',
+      url: 'https://app.vizzly.dev/org/project/builds/api-build-123',
+    }),
     finalizeApiBuild: async () => {},
     output: {
       debug: () => {},
@@ -42,6 +48,10 @@ describe('services/test-runner', () => {
         {},
         { deps }
       );
+      let buildEvents = [];
+      testRunner.on('build-created', event => {
+        buildEvents.push(event);
+      });
 
       let buildId = await testRunner.createBuild(
         {
@@ -54,6 +64,12 @@ describe('services/test-runner', () => {
       );
 
       assert.strictEqual(buildId, 'api-build-123');
+      assert.deepStrictEqual(buildEvents, [
+        {
+          buildId: 'api-build-123',
+          url: 'https://app.vizzly.dev/org/project/builds/api-build-123',
+        },
+      ]);
       assert.deepStrictEqual(writtenSessions, [
         {
           buildId: 'api-build-123',