open-design/e2e/specs/dialog/main.spec.ts

// @vitest-environment node

import { randomUUID } from 'node:crypto';
import { join } from 'node:path';

import { describe, expect, test } from 'vitest';

import { createFakeAgentRuntimes } from '@/fake-agents';
import {
  extractArtifactFromRunEvents,
  persistExtractedArtifact,
  type ProjectFile,
} from '@/vitest/artifacts';
import { requestJson, requestText } from '@/vitest/http';
import { listMessages, saveMessage } from '@/vitest/messages';
import { startRun, readRunEvents, waitForRunStatus } from '@/vitest/runs';
import { createSmokeSuite } from '@/vitest/suite';

const GENERATED_FILE = 'real-daemon-smoke.html';
const GENERATED_HEADING = 'Real Daemon Smoke';
const PROMPT = 'Create a deterministic smoke artifact';

type ProjectResponse = {
  conversationId: string;
  project: {
    id: string;
    metadata?: {
      kind?: string;
    };
    name: string;
  };
};

type ProjectFilesResponse = {
  files: ProjectFile[];
};

describe('dialog main spec', () => {
  test('creates a project, runs the configured agent, streams events, and persists the generated artifact', async () => {
    const suite = await createSmokeSuite('dialog-main');

    await suite.with.toolsDev(async ({ runtime, status, webUrl }) => {
      const fakeAgents = await createFakeAgentRuntimes({
        root: join(suite.scratchDir, 'fake-agents'),
        runtimeIds: ['codex'],
      });

      await requestJson<{ config: Record<string, unknown> }>(webUrl, '/api/app-config', {
        body: {
          agentCliEnv: { codex: fakeAgents.codex.env },
          agentId: 'codex',
          agentModels: { codex: { model: 'default', reasoning: 'default' } },
          designSystemId: null,
          onboardingCompleted: true,
          skillId: null,
          telemetry: { artifactManifest: true, content: false, metrics: false },
        },
        method: 'PUT',
      });

      const project = await requestJson<ProjectResponse>(webUrl, '/api/projects', {
        body: {
          designSystemId: null,
          id: randomUUID(),
          metadata: { kind: 'prototype' },
          name: 'Dialog main smoke project',
          pendingPrompt: null,
          skillId: null,
        },
      });
      expect(project.conversationId).toEqual(expect.any(String));
      expect(project.project.metadata?.kind).toBe('prototype');

      const requestId = `dialog-main-${Date.now()}`;
      const now = Date.now();
      const userMessageId = `user-${requestId}`;
      const assistantMessageId = `assistant-${requestId}`;
      await saveMessage(webUrl, project.project.id, project.conversationId, {
        content: PROMPT,
        createdAt: now,
        id: userMessageId,
        role: 'user',
      });
      await saveMessage(webUrl, project.project.id, project.conversationId, {
        agentId: 'codex',
        agentName: 'Codex',
        content: '',
        createdAt: now,
        events: [],
        id: assistantMessageId,
        role: 'assistant',
        runStatus: 'running',
        startedAt: now,
      });

      const run = await startRun(webUrl, {
        agentId: 'codex',
        assistantMessageId,
        clientRequestId: requestId,
        conversationId: project.conversationId,
        designSystemId: null,
        message: PROMPT,
        model: 'default',
        projectId: project.project.id,
        reasoning: 'default',
        skillId: null,
      });

      const finalRun = await waitForRunStatus(webUrl, run.runId, 'succeeded', { timeoutMs: 30_000 });
      expect(finalRun.projectId).toBe(project.project.id);
      expect(finalRun.conversationId).toBe(project.conversationId);
      expect(finalRun.agentId).toBe('codex');

      const events = await readRunEvents(webUrl, run.runId);
      expect(events).toContain('real-daemon-smoke');
      expect(events).toContain('"type":"usage"');
      expect(events).toContain('"status":"succeeded"');

      const artifact = extractArtifactFromRunEvents(events);
      expect(artifact.identifier).toBe('real-daemon-smoke');
      expect(artifact.title).toBe(GENERATED_HEADING);
      expect(artifact.html).toContain(GENERATED_HEADING);

      const persistedArtifact = await persistExtractedArtifact(
        webUrl,
        project.project.id,
        artifact,
        { designSystemId: null, sourceSkillId: null },
      );
      expect(persistedArtifact.name).toBe(GENERATED_FILE);
      expect(persistedArtifact.kind).toBe('html');

      const files = await requestJson<ProjectFilesResponse>(
        webUrl,
        `/api/projects/${encodeURIComponent(project.project.id)}/files`,
      );
      const generated = files.files.find((file) => file.name === GENERATED_FILE);
      expect(generated?.kind).toBe('html');
      expect(generated?.artifactManifest?.title).toBe(GENERATED_HEADING);
      expect(generated?.artifactManifest?.renderer).toBe('html');

      const rawHtml = await requestText(
        webUrl,
        `/api/projects/${encodeURIComponent(project.project.id)}/raw/${GENERATED_FILE}`,
      );
      expect(rawHtml).toContain(GENERATED_HEADING);
      expect(rawHtml).toContain('Generated through the daemon run path.');

      await saveMessage(webUrl, project.project.id, project.conversationId, {
        agentId: 'codex',
        agentName: 'Codex',
        content: artifact.rawText,
        createdAt: now,
        endedAt: Date.now(),
        events: [],
        id: assistantMessageId,
        producedFiles: [persistedArtifact],
        role: 'assistant',
        runId: finalRun.id,
        runStatus: 'succeeded',
        startedAt: now,
        telemetryFinalized: true,
      });

      const messages = await listMessages(
        webUrl,
        project.project.id,
        project.conversationId,
      );
      const assistantMessage = messages.find((message) => message.id === finalRun.assistantMessageId);
      expect(assistantMessage?.runStatus).toBe('succeeded');
      expect(assistantMessage?.producedFiles).toEqual(
        expect.arrayContaining([
          expect.objectContaining({
            artifactManifest: expect.objectContaining({ title: GENERATED_HEADING }),
            name: GENERATED_FILE,
          }),
        ]),
      );

      await suite.report.json('summary.json', {
        artifact: {
          manifest: generated?.artifactManifest,
          name: generated?.name,
          size: generated?.size,
        },
        conversationId: project.conversationId,
        files: files.files.map((file) => ({
          artifactManifest: file.artifactManifest,
          kind: file.kind,
          name: file.name,
          size: file.size,
        })),
        namespace: suite.namespace,
        project: project.project,
        run: finalRun,
        runtime: {
          daemonPort: runtime.daemonPort,
          webPort: runtime.webPort,
          webUrl,
        },
        status,
      });
    }, {
      // This fixture emits generic artifact output, not OD Next protocol blocks.
      env: { OD_NEXT_STRATEGY_ROLLOUT: 'off' },
    });
  }, 180_000);
});