@jackwener/opencli 0.9.6 → 1.0.0

package/src/doctor.test.ts

@@ -1,223 +1,62 @@
 import { describe, expect, it } from 'vitest';
-import {
-  readTokenFromShellContent,
-  renderBrowserDoctorReport,
-  upsertShellToken,
-  readTomlConfigToken,
-  upsertTomlConfigToken,
-  upsertJsonConfigToken,
-} from './doctor.js';
-
-describe('shell token helpers', () => {
-  it('reads token from shell export', () => {
-    expect(readTokenFromShellContent('export PLAYWRIGHT_MCP_EXTENSION_TOKEN="abc123"\n')).toBe('abc123');
-  });
-
-  it('appends token export when missing', () => {
-    const next = upsertShellToken('export PATH="/usr/bin"\n', 'abc123');
-    expect(next).toContain('export PLAYWRIGHT_MCP_EXTENSION_TOKEN="abc123"');
-  });
-
-  it('replaces token export when present', () => {
-    const next = upsertShellToken('export PLAYWRIGHT_MCP_EXTENSION_TOKEN="old"\n', 'new');
-    expect(next).toContain('export PLAYWRIGHT_MCP_EXTENSION_TOKEN="new"');
-    expect(next).not.toContain('"old"');
-  });
-});
-
-describe('toml token helpers', () => {
-  it('reads token from playwright env section', () => {
-    const content = `
-[mcp_servers.playwright.env]
-PLAYWRIGHT_MCP_EXTENSION_TOKEN = "abc123"
-`;
-    expect(readTomlConfigToken(content)).toBe('abc123');
-  });
-
-  it('updates token inside existing env section', () => {
-    const content = `
-[mcp_servers.playwright.env]
-PLAYWRIGHT_MCP_EXTENSION_TOKEN = "old"
-`;
-    const next = upsertTomlConfigToken(content, 'new');
-    expect(next).toContain('PLAYWRIGHT_MCP_EXTENSION_TOKEN = "new"');
-    expect(next).not.toContain('"old"');
-  });
-
-  it('creates env section when missing', () => {
-    const content = `
-[mcp_servers.playwright]
-type = "stdio"
-`;
-    const next = upsertTomlConfigToken(content, 'abc123');
-    expect(next).toContain('[mcp_servers.playwright.env]');
-    expect(next).toContain('PLAYWRIGHT_MCP_EXTENSION_TOKEN = "abc123"');
-  });
-});
-
-describe('json token helpers', () => {
-  it('writes token into standard mcpServers config', () => {
-    const next = upsertJsonConfigToken(JSON.stringify({
-      mcpServers: {
-        playwright: {
-          command: 'npx',
-          args: ['-y', '@playwright/mcp@latest', '--extension'],
-        },
-      },
-    }), 'abc123');
-    const parsed = JSON.parse(next);
-    expect(parsed.mcpServers.playwright.env.PLAYWRIGHT_MCP_EXTENSION_TOKEN).toBe('abc123');
-  });
-
-  it('writes token into opencode mcp config', () => {
-    const next = upsertJsonConfigToken(JSON.stringify({
-      $schema: 'https://opencode.ai/config.json',
-      mcp: {
-        playwright: {
-          command: ['npx', '-y', '@playwright/mcp@latest', '--extension'],
-          enabled: true,
-          type: 'local',
-        },
-      },
-    }), 'abc123');
-    const parsed = JSON.parse(next);
-    expect(parsed.mcp.playwright.environment.PLAYWRIGHT_MCP_EXTENSION_TOKEN).toBe('abc123');
-  });
-
-  it('creates standard mcpServers format for empty file (not OpenCode)', () => {
-    const next = upsertJsonConfigToken('', 'abc123');
-    const parsed = JSON.parse(next);
-    expect(parsed.mcpServers.playwright.env.PLAYWRIGHT_MCP_EXTENSION_TOKEN).toBe('abc123');
-    expect(parsed.mcp).toBeUndefined();
-  });
-
-  it('creates OpenCode format when filePath contains opencode', () => {
-    const next = upsertJsonConfigToken('', 'abc123', '/home/user/.config/opencode/opencode.json');
-    const parsed = JSON.parse(next);
-    expect(parsed.mcp.playwright.environment.PLAYWRIGHT_MCP_EXTENSION_TOKEN).toBe('abc123');
-    expect(parsed.mcpServers).toBeUndefined();
-  });
-
-  it('creates standard format when filePath is claude.json', () => {
-    const next = upsertJsonConfigToken('', 'abc123', '/home/user/.claude.json');
-    const parsed = JSON.parse(next);
-    expect(parsed.mcpServers.playwright.env.PLAYWRIGHT_MCP_EXTENSION_TOKEN).toBe('abc123');
-  });
-});
-
-describe('fish shell support', () => {
-  it('generates fish set -gx syntax for fish config path', () => {
-    const next = upsertShellToken('', 'abc123', '/home/user/.config/fish/config.fish');
-    expect(next).toContain('set -gx PLAYWRIGHT_MCP_EXTENSION_TOKEN "abc123"');
-    expect(next).not.toContain('export');
-  });
-
-  it('replaces existing fish set line', () => {
-    const content = 'set -gx PLAYWRIGHT_MCP_EXTENSION_TOKEN "old"\n';
-    const next = upsertShellToken(content, 'new', '/home/user/.config/fish/config.fish');
-    expect(next).toContain('set -gx PLAYWRIGHT_MCP_EXTENSION_TOKEN "new"');
-    expect(next).not.toContain('"old"');
-  });
-
-  it('appends fish syntax to existing fish config', () => {
-    const content = 'set -gx PATH /usr/bin\n';
-    const next = upsertShellToken(content, 'abc123', '/home/user/.config/fish/config.fish');
-    expect(next).toContain('set -gx PLAYWRIGHT_MCP_EXTENSION_TOKEN "abc123"');
-    expect(next).toContain('set -gx PATH /usr/bin');
-  });
-
-  it('uses export syntax for zshrc even with filePath', () => {
-    const next = upsertShellToken('', 'abc123', '/home/user/.zshrc');
-    expect(next).toContain('export PLAYWRIGHT_MCP_EXTENSION_TOKEN="abc123"');
-    expect(next).not.toContain('set -gx');
-  });
-});
+import { renderBrowserDoctorReport } from './doctor.js';
 
 describe('doctor report rendering', () => {
   const strip = (s: string) => s.replace(/\x1b\[[0-9;]*m/g, '');
 
-  it('renders OK-style report when tokens match', () => {
+  it('renders OK-style report when daemon and extension connected', () => {
     const text = strip(renderBrowserDoctorReport({
-      envToken: 'abc123',
-      envFingerprint: 'fp1',
-      extensionToken: 'abc123',
-      extensionFingerprint: 'fp1',
-      extensionInstalled: true,
-      extensionBrowsers: ['Chrome'],
-      shellFiles: [{ path: '/tmp/.zshrc', exists: true, token: 'abc123', fingerprint: 'fp1' }],
-      configs: [{ path: '/tmp/mcp.json', exists: true, format: 'json', token: 'abc123', fingerprint: 'fp1', writable: true }],
-      recommendedToken: 'abc123',
-      recommendedFingerprint: 'fp1',
-      warnings: [],
+      daemonRunning: true,
+      extensionConnected: true,
       issues: [],
     }));
 
-    expect(text).toContain('[OK] Extension installed (Chrome)');
-    expect(text).toContain('[OK] Environment token: configured (fp1)');
-    expect(text).toContain('[OK] /tmp/mcp.json');
-    expect(text).toContain('configured (fp1)');
+    expect(text).toContain('[OK] Daemon: running on port 19825');
+    expect(text).toContain('[OK] Extension: connected');
+    expect(text).toContain('Everything looks good!');
+  });
+
+  it('renders MISSING when daemon not running', () => {
+    const text = strip(renderBrowserDoctorReport({
+      daemonRunning: false,
+      extensionConnected: false,
+      issues: ['Daemon is not running.'],
+    }));
+
+    expect(text).toContain('[MISSING] Daemon: not running');
+    expect(text).toContain('[MISSING] Extension: not connected');
+    expect(text).toContain('Daemon is not running.');
   });
 
-  it('renders MISMATCH-style report when fingerprints differ', () => {
+  it('renders extension not connected when daemon is running', () => {
     const text = strip(renderBrowserDoctorReport({
-      envToken: 'abc123',
-      envFingerprint: 'fp1',
-      extensionToken: null,
-      extensionFingerprint: null,
-      extensionInstalled: false,
-      extensionBrowsers: [],
-      shellFiles: [{ path: '/tmp/.zshrc', exists: true, token: 'def456', fingerprint: 'fp2' }],
-      configs: [{ path: '/tmp/mcp.json', exists: true, format: 'json', token: 'abc123', fingerprint: 'fp1', writable: true }],
-      recommendedToken: 'abc123',
-      recommendedFingerprint: 'fp1',
-      warnings: [],
-      issues: ['Detected inconsistent Playwright MCP tokens across env/config files.'],
+      daemonRunning: true,
+      extensionConnected: false,
+      issues: ['Daemon is running but the Chrome extension is not connected.'],
     }));
 
-    expect(text).toContain('[MISSING] Extension not installed in any browser');
-    expect(text).toContain('[MISMATCH] Environment token: configured (fp1)');
-    expect(text).toContain('[MISMATCH] /tmp/.zshrc');
-    expect(text).toContain('configured (fp2)');
-    expect(text).toContain('[MISMATCH] Recommended token fingerprint: fp1');
+    expect(text).toContain('[OK] Daemon: running on port 19825');
+    expect(text).toContain('[MISSING] Extension: not connected');
   });
 
   it('renders connectivity OK when live test succeeds', () => {
     const text = strip(renderBrowserDoctorReport({
-      envToken: 'abc123',
-      envFingerprint: 'fp1',
-      extensionToken: 'abc123',
-      extensionFingerprint: 'fp1',
-      extensionInstalled: true,
-      extensionBrowsers: ['Chrome'],
-      shellFiles: [],
-      configs: [],
-      recommendedToken: 'abc123',
-      recommendedFingerprint: 'fp1',
+      daemonRunning: true,
+      extensionConnected: true,
       connectivity: { ok: true, durationMs: 1234 },
-      warnings: [],
       issues: [],
     }));
 
-    expect(text).toContain('[OK] Browser connectivity: connected in 1.2s');
+    expect(text).toContain('[OK] Connectivity: connected in 1.2s');
   });
 
-  it('renders connectivity WARN when not tested', () => {
+  it('renders connectivity SKIP when not tested', () => {
     const text = strip(renderBrowserDoctorReport({
-      envToken: 'abc123',
-      envFingerprint: 'fp1',
-      extensionToken: 'abc123',
-      extensionFingerprint: 'fp1',
-      extensionInstalled: true,
-      extensionBrowsers: ['Chrome'],
-      shellFiles: [],
-      configs: [],
-      recommendedToken: 'abc123',
-      recommendedFingerprint: 'fp1',
-      warnings: [],
+      daemonRunning: true,
+      extensionConnected: true,
       issues: [],
     }));
 
-    expect(text).toContain('[WARN] Browser connectivity: not tested (use --live)');
+    expect(text).toContain('[SKIP] Connectivity: not tested (use --live)');
   });
 });
-