84 lines
2.8 KiB
TypeScript
84 lines
2.8 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
// Unset NO_COLOR environment variable to ensure consistent theme behavior between local and CI test runs
|
|
if (process.env['NO_COLOR'] !== undefined) {
|
|
delete process.env['NO_COLOR'];
|
|
}
|
|
|
|
import { mkdir, readdir, rm } from 'node:fs/promises';
|
|
import { join, dirname } from 'node:path';
|
|
import { fileURLToPath } from 'node:url';
|
|
import { canUseRipgrep } from '../packages/core/src/tools/ripGrep.js';
|
|
|
|
const __dirname = dirname(fileURLToPath(import.meta.url));
|
|
const rootDir = join(__dirname, '..');
|
|
const integrationTestsDir = join(rootDir, '.integration-tests');
|
|
let runDir = ''; // Make runDir accessible in teardown
|
|
|
|
export async function setup() {
|
|
runDir = join(integrationTestsDir, `${Date.now()}`);
|
|
await mkdir(runDir, { recursive: true });
|
|
|
|
// Set the home directory to the test run directory to avoid conflicts
|
|
// with the user's local config.
|
|
process.env['HOME'] = runDir;
|
|
if (process.platform === 'win32') {
|
|
process.env['USERPROFILE'] = runDir;
|
|
}
|
|
// We also need to set the config dir explicitly, since the code might
|
|
// construct the path before the HOME env var is set.
|
|
process.env['GEMINI_CONFIG_DIR'] = join(runDir, '.gemini');
|
|
|
|
// Download ripgrep to avoid race conditions in parallel tests
|
|
const available = await canUseRipgrep();
|
|
if (!available) {
|
|
throw new Error('Failed to download ripgrep binary');
|
|
}
|
|
|
|
// Clean up old test runs, but keep the latest few for debugging
|
|
try {
|
|
const testRuns = await readdir(integrationTestsDir);
|
|
if (testRuns.length > 5) {
|
|
const oldRuns = testRuns.sort().slice(0, testRuns.length - 5);
|
|
await Promise.all(
|
|
oldRuns.map((oldRun) =>
|
|
rm(join(integrationTestsDir, oldRun), {
|
|
recursive: true,
|
|
force: true,
|
|
}),
|
|
),
|
|
);
|
|
}
|
|
} catch (e) {
|
|
console.error('Error cleaning up old test runs:', e);
|
|
}
|
|
|
|
process.env['INTEGRATION_TEST_FILE_DIR'] = runDir;
|
|
process.env['GEMINI_CLI_INTEGRATION_TEST'] = 'true';
|
|
// Force file storage to avoid keychain prompts/hangs in CI, especially on macOS
|
|
process.env['GEMINI_FORCE_FILE_STORAGE'] = 'true';
|
|
process.env['TELEMETRY_LOG_FILE'] = join(runDir, 'telemetry.log');
|
|
|
|
if (process.env['KEEP_OUTPUT']) {
|
|
console.log(`Keeping output for test run in: ${runDir}`);
|
|
}
|
|
process.env['VERBOSE'] = process.env['VERBOSE'] ?? 'false';
|
|
|
|
console.log(`\nIntegration test output directory: ${runDir}`);
|
|
}
|
|
|
|
export async function teardown() {
|
|
// Cleanup the test run directory unless KEEP_OUTPUT is set
|
|
if (process.env['KEEP_OUTPUT'] !== 'true' && runDir) {
|
|
try {
|
|
await rm(runDir, { recursive: true, force: true });
|
|
} catch (e) {
|
|
console.warn('Failed to clean up test run directory:', e);
|
|
}
|
|
}
|
|
}
|