chore: update toxiproxy remove proxy (#382)

aws · Jan 23, 2025 · 1ec01a3 · 1ec01a3
1 parent a4d9d68
commit 1ec01a3
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/integration/container/tests/utils/proxy_helper.ts b/tests/integration/container/tests/utils/proxy_helper.ts
@@ -70,10 +70,14 @@ export class ProxyHelper {
     const proxy = proxyInfo.proxy;
 
     if (proxy !== undefined) {
-      const toxics = proxy.toxics;
-      toxics.forEach((toxic) => {
-        toxic.remove();
-      });
+      try {
+        const upstreamToxic = await proxy.getToxic("UP-STREAM");
+        await upstreamToxic.remove();
+        const downstreamToxic = await proxy.getToxic("DOWN-STREAM");
+        await downstreamToxic.remove();
+      } catch (e) {
+        // Ignore
+      }
     }
   }
 }