From 3dbd48441e3235746491e7f6efd67d7248f043f3 Mon Sep 17 00:00:00 2001
From: Samuel Attard <sattard@anthropic.com>
Date: Sat, 11 Apr 2026 23:18:29 -0700
Subject: [PATCH] chore: replace ifit/ifdescribe addOnly shim with vitest runIf

---
 spec/_vitest_runner/setup.ts |  2 +-
 spec/lib/spec-helpers.ts     | 13 ++-----------
 2 files changed, 3 insertions(+), 12 deletions(-)

diff --git a/spec/_vitest_runner/setup.ts b/spec/_vitest_runner/setup.ts
index 20717f2afa..195b853a76 100644
--- a/spec/_vitest_runner/setup.ts
+++ b/spec/_vitest_runner/setup.ts
@@ -10,7 +10,7 @@ import chaiAsPromised = require('chai-as-promised');
 import dirtyChai = require('dirty-chai');
 
 chai.use(chaiAsPromised);
-chai.use(dirtyChai as any);
+chai.use(dirtyChai);
 
 // Show full object diff.
 // https://github.com/chaijs/chai/issues/469
diff --git a/spec/lib/spec-helpers.ts b/spec/lib/spec-helpers.ts
index f662e1ee5f..72e17a74d8 100644
--- a/spec/lib/spec-helpers.ts
+++ b/spec/lib/spec-helpers.ts
@@ -13,17 +13,8 @@ import { setTimeout } from 'node:timers/promises';
 import * as url from 'node:url';
 import * as v8 from 'node:v8';
 
-const addOnly = <T>(fn: Function): T => {
-  const wrapped = (...args: any[]) => {
-    return fn(...args);
-  };
-  (wrapped as any).only = wrapped;
-  (wrapped as any).skip = wrapped;
-  return wrapped as any;
-};
-
-export const ifit = (condition: boolean) => (condition ? it : addOnly<typeof it>(it.skip));
-export const ifdescribe = (condition: boolean) => (condition ? describe : addOnly<typeof describe>(describe.skip));
+export const ifit = (condition: boolean) => it.runIf(condition);
+export const ifdescribe = (condition: boolean) => describe.runIf(condition);
 
 type DoneCallback = (err?: unknown) => void;