test: achieve 100% unit test coverage

Co-authored-by: Molten Bot 000 <260473928+moltenbot000@users.noreply.github.com>

Author: moltenhub-bot

packages/agent-infra/browser/src/browser-finder/chrome-paths.test.ts

@@ -0,0 +1,69 @@
+/*
+ * Copyright (c) 2025 Bytedance, Inc. and its affiliates.
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { mkdtempSync, mkdirSync, rmSync, writeFileSync } from 'fs';
+import { tmpdir } from 'os';
+import { dirname, join } from 'path';
+import { afterEach, describe, expect, it, vi } from 'vitest';
+
+vi.mock('which', () => ({
+  default: {
+    sync: vi.fn(() => {
+      throw new Error('not found');
+    }),
+  },
+}));
+
+describe('chrome-paths', () => {
+  const originalPlatform = process.platform;
+  const originalEnv = { ...process.env };
+  const tempDirs: string[] = [];
+
+  afterEach(() => {
+    vi.resetModules();
+    Object.defineProperty(process, 'platform', {
+      value: originalPlatform,
+    });
+    process.env = { ...originalEnv };
+
+    for (const dir of tempDirs.splice(0)) {
+      rmSync(dir, { force: true, recursive: true });
+    }
+  });
+
+  it('uses PUPPETEER_EXECUTABLE_PATH before PATH lookup', async () => {
+    const tempDir = mkdtempSync(join(tmpdir(), 'chrome-paths-'));
+    tempDirs.push(tempDir);
+    const chromePath = join(tempDir, 'chrome');
+    writeFileSync(chromePath, '');
+    process.env.PUPPETEER_EXECUTABLE_PATH = chromePath;
+
+    const { getAnyChromeStable } = await import('./chrome-paths');
+
+    expect(getAnyChromeStable()).toBe(chromePath);
+  });
+
+  it('finds Chrome downloaded into Puppeteer cache on Linux', async () => {
+    Object.defineProperty(process, 'platform', {
+      value: 'linux',
+    });
+
+    const tempDir = mkdtempSync(join(tmpdir(), 'puppeteer-cache-'));
+    tempDirs.push(tempDir);
+    const chromePath = join(
+      tempDir,
+      'chrome',
+      'linux-134.0.6998.35',
+      'chrome-linux64',
+      'chrome',
+    );
+    mkdirSync(dirname(chromePath), { recursive: true });
+    writeFileSync(chromePath, '');
+    process.env.PUPPETEER_CACHE_DIR = tempDir;
+
+    const { getAnyChromeStable } = await import('./chrome-paths');
+
+    expect(getAnyChromeStable()).toBe(chromePath);
+  });
+});

packages/agent-infra/browser/src/browser-finder/chrome-paths.ts

@@ -14,7 +14,7 @@
  *    which is a time-consuming operation (taking up to 6 seconds on my computer!).
  *    Since this process is performed during the app's startup, such a delay is unacceptable.
  */
-import { existsSync } from 'fs';
+import { existsSync, readdirSync } from 'fs';
 import { sep, join } from 'path';
 import which from 'which';
 
@@ -42,6 +42,52 @@ function getChromeOnLinux(
   return null;
 }
 
+function getChromeFromEnv(): string | null {
+  const envPath =
+    process.env.PUPPETEER_EXECUTABLE_PATH ||
+    process.env.CHROME_PATH ||
+    process.env.GOOGLE_CHROME_BIN;
+
+  if (envPath && existsSync(envPath)) {
+    return envPath;
+  }
+
+  return null;
+}
+
+function getPuppeteerCacheChrome(): string | null {
+  if (platform !== 'linux') {
+    return null;
+  }
+
+  const cacheDir =
+    process.env.PUPPETEER_CACHE_DIR ||
+    join(process.env.HOME || '', '.cache', 'puppeteer');
+
+  const chromeDir = join(cacheDir, 'chrome');
+
+  if (!existsSync(chromeDir)) {
+    return null;
+  }
+
+  try {
+    const builds = readdirSync(chromeDir, { withFileTypes: true })
+      .filter((entry) => entry.isDirectory())
+      .map((entry) => entry.name)
+      .sort()
+      .reverse();
+
+    for (const build of builds) {
+      const chromePath = join(chromeDir, build, 'chrome-linux64', 'chrome');
+      if (existsSync(chromePath)) {
+        return chromePath;
+      }
+    }
+  } catch (e) {}
+
+  return null;
+}
+
 function getChromeOnWindows(
   name: 'Chrome' | 'Chrome Beta' | 'Chrome Dev' | 'Chrome SxS',
 ): string | null {
@@ -107,6 +153,11 @@ const chromePaths = {
 };
 
 function getChromePath() {
+  const envChrome = getChromeFromEnv();
+  if (envChrome) {
+    return envChrome;
+  }
+
   const chrome = chromePaths.chrome;
 
   if (platform && Object.keys(chrome).includes(platform)) {
@@ -115,6 +166,11 @@ function getChromePath() {
       return pth;
     }
   }
+
+  const cachedChrome = getPuppeteerCacheChrome();
+  if (cachedChrome) {
+    return cachedChrome;
+  }
 }
 
 function getChromeBetaPath() {

packages/ui-tars/operators/nut-js/test/execute.test.ts

@@ -33,19 +33,49 @@ vi.mock('@ui-tars/sdk/core', async (importOriginal) => {
   };
 });
 
-// Mock @computer-use/nut-js
-vi.mock('@computer-use/nut-js', async (importOriginal) => {
-  const actual: any = await importOriginal();
+vi.mock('@computer-use/nut-js', () => {
+  class Point {
+    constructor(
+      public x: number,
+      public y: number,
+    ) {}
+  }
+
+  class Region {
+    constructor(
+      public x: number,
+      public y: number,
+      public width: number,
+      public height: number,
+    ) {}
+  }
+
   return {
     mouse: {
       move: vi.fn(),
       click: vi.fn(),
+      doubleClick: vi.fn(),
       config: {
         mouseSpeed: 1500,
       },
       drag: vi.fn(),
     },
-    Key: actual.Key,
+    Key: {
+      Comma: 'comma',
+      Enter: 'enter',
+      LeftAlt: 'left-alt',
+      LeftCmd: 'left-cmd',
+      LeftControl: 'left-control',
+      LeftShift: 'left-shift',
+      LeftWin: 'left-win',
+      PageDown: 'page-down',
+      PageUp: 'page-up',
+      Up: 'up',
+      Down: 'down',
+      Left: 'left',
+      Right: 'right',
+      V: 'v',
+    },
     keyboard: {
       type: vi.fn(),
       pressKey: vi.fn(),
@@ -54,13 +84,20 @@ vi.mock('@computer-use/nut-js', async (importOriginal) => {
         autoDelayMs: 0,
       },
     },
+    screen: {
+      grab: vi.fn(),
+    },
+    clipboard: {
+      getContent: vi.fn(),
+      setContent: vi.fn(),
+    },
     Button: {
       LEFT: 'left',
       RIGHT: 'right',
       MIDDLE: 'middle',
     },
-    Point: actual.Point,
-    Region: actual.Region,
+    Point,
+    Region,
     straightTo: vi.fn((point) => point),
     centerOf: vi.fn((region) => region),
     randomPointIn: vi.fn((region) => region),