From f8d0d2d599318e74a7c62dd4e07f81f74fa2d9a6 Mon Sep 17 00:00:00 2001
From: Samuel Attard <sattard@anthropic.com>
Date: Sat, 11 Apr 2026 22:59:00 -0700
Subject: [PATCH] chore: migrate api-net-session-spec.ts to vitest

---
 ...-net-session-spec.ts => api-net-session.spec.ts} | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)
 rename spec/{api-net-session-spec.ts => api-net-session.spec.ts} (98%)

diff --git a/spec/api-net-session-spec.ts b/spec/api-net-session.spec.ts
similarity index 98%
rename from spec/api-net-session-spec.ts
rename to spec/api-net-session.spec.ts
index 657aec3665..6d98c1986b 100644
--- a/spec/api-net-session-spec.ts
+++ b/spec/api-net-session.spec.ts
@@ -1,6 +1,7 @@
 import { net, session, BrowserWindow, ClientRequestConstructorOptions } from 'electron/main';
 
 import { expect } from 'chai';
+import { afterEach, beforeEach, describe, it } from 'vitest';
 
 import * as dns from 'node:dns';
 
@@ -14,14 +15,12 @@ describe('net module (session)', () => {
   beforeEach(() => {
     respondNTimes.routeFailure = false;
   });
-  afterEach(async function () {
+  afterEach(async (ctx) => {
     await session.defaultSession.clearCache();
-    if (respondNTimes.routeFailure && this.test) {
-      if (!this.test.isFailed()) {
-        throw new Error(
-          'Failing this test due an unhandled error in the respondOnce route handler, check the logs above for the actual error'
-        );
-      }
+    if (respondNTimes.routeFailure && ctx.task.result?.state !== 'fail') {
+      throw new Error(
+        'Failing this test due an unhandled error in the respondOnce route handler, check the logs above for the actual error'
+      );
     }
   });