From 133f0230c3d347a1d02c5babc3f1a301622c4f67 Mon Sep 17 00:00:00 2001
From: cornmander <shikhman@google.com>
Date: Mon, 18 Aug 2025 01:17:22 -0400
Subject: [PATCH] Add vi globals to fix integration tests. (#6445)

---
 integration-tests/replace.test.ts    | 2 +-
 integration-tests/write_file.test.ts | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/integration-tests/replace.test.ts b/integration-tests/replace.test.ts
index a04025e7..e7ed23a2 100644
--- a/integration-tests/replace.test.ts
+++ b/integration-tests/replace.test.ts
@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { describe, it, expect } from 'vitest';
+import { describe, it, expect, vi } from 'vitest';
 import { TestRig, printDebugInfo, validateModelOutput } from './test-helper.js';
 
 describe('replace', () => {
diff --git a/integration-tests/write_file.test.ts b/integration-tests/write_file.test.ts
index 49cdda38..7dd6445d 100644
--- a/integration-tests/write_file.test.ts
+++ b/integration-tests/write_file.test.ts
@@ -4,7 +4,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { describe, it, expect } from 'vitest';
+import { describe, it, expect, vi } from 'vitest';
 import {
   TestRig,
   createToolCallErrorMessage,