Improve test (#237890)

microsoft · Jan 14, 2025 · 384b832 · 384b832
1 parent ef520e4
commit 384b832
Showing 1 changed file with 20 additions and 11 deletions.
diff --git a/extensions/vscode-api-tests/src/singlefolder-tests/proxy.test.ts b/extensions/vscode-api-tests/src/singlefolder-tests/proxy.test.ts
@@ -105,18 +105,27 @@ import { middleware, Straightforward } from 'straightforward';
 			await proxyListen;
 			const proxyPort = (sf.server.address() as AddressInfo).port;
 
-			await vscode.workspace.getConfiguration().update('integration-test.http.proxy', `PROXY 127.0.0.1:${proxyPort}`, vscode.ConfigurationTarget.Global);
-			await delay(1000); // Wait for the configuration change to propagate.
-			await new Promise<void>((resolve, reject) => {
-				https.get(url, res => {
-					if (res.statusCode === 418) {
-						resolve();
-					} else {
-						reject(new Error(`Unexpected status code (expected 418): ${res.statusCode}`));
+			for (let i = 0; i < 3; i++) {
+				await vscode.workspace.getConfiguration().update('integration-test.http.proxy', `PROXY 127.0.0.1:${proxyPort}`, vscode.ConfigurationTarget.Global);
+				await delay(1000); // Wait for the configuration change to propagate.
+				try {
+					await new Promise<void>((resolve, reject) => {
+						https.get(url, res => {
+							if (res.statusCode === 418) {
+								resolve();
+							} else {
+								reject(new Error(`Unexpected status code (expected 418): ${res.statusCode}`));
+							}
+						})
+							.on('error', reject);
+					});
+					break; // Exit the loop if the request is successful
+				} catch (err) {
+					if (i === 2) {
+						throw err; // Rethrow the error if it's the last attempt
 					}
-				})
-					.on('error', reject);
-			});
+				}
+			}
 
 			authEnabled = true;
 			await new Promise<void>((resolve, reject) => {