From d4d193968ff603ab49de72c603f4eb0f44397ec6 Mon Sep 17 00:00:00 2001
From: jmjaffe37 <jmjaffe@microsoft.com>
Date: Wed, 1 Jul 2026 13:23:54 -0700
Subject: [PATCH] added suggestion from PR feedback

---
 __tests__/distributors/jetbrains-installer.test.ts | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/__tests__/distributors/jetbrains-installer.test.ts b/__tests__/distributors/jetbrains-installer.test.ts
index d292e574..2d32e7e6 100644
--- a/__tests__/distributors/jetbrains-installer.test.ts
+++ b/__tests__/distributors/jetbrains-installer.test.ts
@@ -89,14 +89,20 @@ describe('findPackageForDownload', () => {
 
       await new Promise<void>((resolve, reject) => {
         const request = https.request(url, options, res => {
+          let assertionError: unknown;
+
           try {
             // JetBrains uses 403 for non-existent packages
             expect(res.statusCode).not.toBe(403);
-            res.resume();
-            res.on('end', resolve);
           } catch (error) {
-            reject(error);
+            assertionError = error;
           }
+
+          res.resume();
+          res.once('error', reject);
+          res.once('end', () =>
+            assertionError ? reject(assertionError as Error) : resolve()
+          );
         });
 
         request.on('error', reject);