@autobest-ui/agent 1.0.11 → 1.0.13

package/mcp/trace-mcp-recorder/index.test.js

@@ -0,0 +1,189 @@
+import assert from 'node:assert/strict';
+import { EventEmitter } from 'node:events';
+import path from 'node:path';
+import test from 'node:test';
+import { fileURLToPath } from 'node:url';
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
+import { PNG } from 'pngjs';
+import { pixelDifference, TraceRecorder } from './index.js';
+
+const directory = path.dirname(fileURLToPath(import.meta.url));
+const sessionUuid = '123e4567-e89b-42d3-a456-426614174000';
+
+test('calculates pixel differences for visual probes', () => {
+  const first = new PNG({ width: 2, height: 1 });
+  const second = new PNG({ width: 2, height: 1 });
+  first.data.fill(255);
+  second.data.fill(255);
+  second.data[0] = 0;
+  assert.equal(pixelDifference(PNG.sync.write(first), PNG.sync.write(second)), 0.5);
+});
+
+function createFixture(options = {}) {
+  const calls = [];
+  const browser = new EventEmitter();
+  const tracing = {
+    start: async options => calls.push(['trace-start', options]),
+    stop: async options => calls.push(['trace-stop', options])
+  };
+  const page = {
+    goto: async url => calls.push(['goto', url]),
+    evaluate: async () => 'Fixture User Agent',
+    viewportSize: () => ({ width: 1280, height: 720 }),
+    close: async () => calls.push(['page-close'])
+  };
+  const context = {
+    tracing,
+    newPage: async () => page,
+    close: async () => calls.push(['context-close'])
+  };
+  browser.newContext = async () => context;
+  browser.version = () => 'Fixture Chromium 1';
+  browser.close = async () => calls.push(['browser-close']);
+
+  const http = {
+    post: async (endpoint, body) => {
+      calls.push(['post', endpoint, body]);
+      if (endpoint === '/api/trace-sessions' && body !== form) return { data: { uuid: sessionUuid } };
+      if (options.uploadError) throw options.uploadError;
+      return { data: { fileUrl: '/private/traces/trace.zip' } };
+    },
+    patch: async (endpoint, body) => {
+      calls.push(['patch', endpoint, body]);
+      return { data: {} };
+    }
+  };
+  const form = {
+    append: (...args) => calls.push(['form-append', ...args]),
+    getHeaders: () => ({ 'content-type': 'multipart/form-data; boundary=fixture' })
+  };
+  const timer = { unref() {} };
+  const recorder = new TraceRecorder({
+    environment: {
+      BACKEND_BASE_URL: 'https://trace.example.test/root/',
+      MAX_RECORD_DURATION: '1000'
+    },
+    launchChromium: async options => {
+      calls.push(['launch', options]);
+      return browser;
+    },
+    createHttpClient: baseUrl => {
+      calls.push(['http-client', baseUrl]);
+      return http;
+    },
+    createFormData: () => form,
+    createReadStream: tracePath => ({ tracePath }),
+    makeTemporaryDirectory: async () => '/tmp/trace-fixture',
+    removeTemporaryDirectory: async temporaryDirectory => calls.push(['remove', temporaryDirectory]),
+    setTimer: (callback, duration) => {
+      calls.push(['timer', duration]);
+      timer.callback = callback;
+      return timer;
+    },
+    clearTimer: value => calls.push(['clear-timer', value]),
+    platformInfo: () => ({ platform: 'fixture-os' })
+  });
+  return { browser, calls, recorder, timer };
+}
+
+test('starts, records metadata, uploads, completes and cleans up one session', async () => {
+  const { calls, recorder } = createFixture();
+  const started = await recorder.start(
+    {
+      url: 'https://app.example.test/login',
+      version: 'v2.1.0',
+      title: 'bug3452',
+      operator: '传入操作人'
+    },
+    { authInfo: { extra: { operator: '上下文操作人' } } }
+  );
+
+  assert.equal(started.uuid, undefined);
+  assert.equal(started.metadata.operator, '上下文操作人');
+  assert.equal(started.metadata.browserVersion, 'Fixture Chromium 1');
+  await assert.rejects(
+    recorder.start({ url: 'https://app.example.test/another' }),
+    /已有录制任务/
+  );
+
+  const confirmation = await recorder.stop();
+  assert.match(confirmation.message, /是否上传/);
+  const stopped = await recorder.stop({ confirm: 'upload' });
+  assert.equal(
+    stopped.replayLink,
+    'https://trace.example.test/root/trace-viewer/?trace=/private/traces/trace.zip'
+  );
+  assert.equal(stopped.metadata.title, 'bug3452');
+  assert.ok(calls.some(call => call[0] === 'trace-start' && call[1].sources === true));
+  assert.ok(calls.some(call => call[0] === 'post' && call[1] === '/api/trace-sessions'));
+  assert.ok(calls.some(call => call[0] === 'form-append' && call[1] === 'target_url'));
+  assert.ok(calls.some(call => call[0] === 'remove' && call[1] === '/tmp/trace-fixture'));
+});
+
+test('rejects missing configuration before contacting the backend', async () => {
+  const recorder = new TraceRecorder({ environment: {} });
+  await assert.rejects(recorder.start({ url: 'https://app.example.test' }), /BACKEND_BASE_URL/);
+});
+
+test('cleans up when Chromium launch fails without a backend session', async () => {
+  const { calls, recorder } = createFixture();
+  recorder.launchChromium = async () => {
+    throw new Error('launch failed');
+  };
+
+  await assert.rejects(recorder.start({ url: 'https://app.example.test' }), /launch failed/);
+  assert.equal(calls.some(call => call[0] === 'post'), false);
+});
+
+test('marks the session failed and removes the local trace when upload fails', async () => {
+  const { calls, recorder } = createFixture({ uploadError: new Error('upload failed') });
+  await recorder.start({ url: 'https://app.example.test' });
+
+  await recorder.stop();
+  await assert.rejects(recorder.stop({ confirm: 'upload' }), /upload failed/);
+  assert.ok(calls.some(call => call[0] === 'remove' && call[1] === '/tmp/trace-fixture'));
+});
+
+test('automatically stops on timeout and returns that result on the next stop call', async () => {
+  const { recorder, timer } = createFixture();
+  await recorder.start({ url: 'https://app.example.test' });
+  timer.callback();
+
+  while (recorder.activeSession) await new Promise(resolve => setImmediate(resolve));
+  const result = await recorder.stop();
+  assert.equal(result.stoppedBy, 'timeout');
+  await assert.rejects(recorder.stop(), /当前没有正在录制/);
+});
+
+test('exposes both tools and the bundled skill over stdio', async () => {
+  const client = new Client({ name: 'trace-mcp-recorder-test', version: '1.0.0' });
+  const transport = new StdioClientTransport({
+    command: process.execPath,
+    args: [path.join(directory, 'index.js')],
+    cwd: directory,
+    env: { ...process.env },
+    stderr: 'pipe'
+  });
+
+  try {
+    await client.connect(transport);
+    assert.equal(client.getServerVersion()?.name, 'trace-mcp-recorder');
+    const { tools } = await client.listTools();
+    assert.deepEqual(
+      tools.map(tool => tool.name),
+      ['start_trace_recording', 'stop_trace_recording']
+    );
+    const { resources } = await client.listResources();
+    assert.ok(resources.some(resource => resource.uri === 'skill://trace-recorder/SKILL.md'));
+
+    const result = await client.callTool({
+      name: 'start_trace_recording',
+      arguments: { url: 'https://app.example.test' }
+    });
+    assert.equal(result.isError, true);
+    assert.match(result.content[0].text, /BACKEND_BASE_URL/);
+  } finally {
+    await client.close();
+  }
+});