From fc88120fdc4a8b178302ff3eac5127b28a7dc117 Mon Sep 17 00:00:00 2001
From: Reinier Criel <reinier.criel@gmail.com>
Date: Fri, 5 Dec 2025 10:01:55 -0800
Subject: [PATCH] Also for uv and poetry

---
 test/e2e/poetry.e2e.spec.js                | 6 +++---
 test/e2e/safe-chain-cli-python.e2e.spec.js | 2 +-
 test/e2e/uv.e2e.spec.js                    | 3 +++
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/test/e2e/poetry.e2e.spec.js b/test/e2e/poetry.e2e.spec.js
index 9836e18..3d19783 100644
--- a/test/e2e/poetry.e2e.spec.js
+++ b/test/e2e/poetry.e2e.spec.js
@@ -16,6 +16,9 @@ describe("E2E: poetry coverage", () => {
 
     const installationShell = await container.openShell("zsh");
     await installationShell.runCommand("safe-chain setup --include-python");
+    
+    // Clear poetry cache
+    await installationShell.runCommand("command poetry cache clear pypi --all -n");
   });
 
   afterEach(async () => {
@@ -29,9 +32,6 @@ describe("E2E: poetry coverage", () => {
   it(`successfully installs known safe packages with poetry add`, async () => {
     const shell = await container.openShell("zsh");
     
-    // Clear poetry cache using command to bypass safe-chain wrapper
-    await shell.runCommand("command poetry cache clear pypi --all -n");
-    
     // Initialize a new poetry project
     await shell.runCommand("mkdir /tmp/test-poetry-project && cd /tmp/test-poetry-project");
     await shell.runCommand("cd /tmp/test-poetry-project && poetry init --no-interaction");
diff --git a/test/e2e/safe-chain-cli-python.e2e.spec.js b/test/e2e/safe-chain-cli-python.e2e.spec.js
index fa1bfdf..457c624 100644
--- a/test/e2e/safe-chain-cli-python.e2e.spec.js
+++ b/test/e2e/safe-chain-cli-python.e2e.spec.js
@@ -15,7 +15,7 @@ describe("E2E: safe-chain CLI python/pip support", () => {
     // Note: We do NOT run 'safe-chain setup' here.
     // We want to test the 'safe-chain' CLI command directly.
 
-    // Clear pip cache before each test to ensure fresh downloads through proxy
+    // Clear pip cache
     const shell = await container.openShell("zsh");
     await shell.runCommand("pip3 cache purge");
   });
diff --git a/test/e2e/uv.e2e.spec.js b/test/e2e/uv.e2e.spec.js
index eae7c12..7314e65 100644
--- a/test/e2e/uv.e2e.spec.js
+++ b/test/e2e/uv.e2e.spec.js
@@ -16,6 +16,9 @@ describe("E2E: uv coverage", () => {
 
     const installationShell = await container.openShell("zsh");
     await installationShell.runCommand("safe-chain setup --include-python");
+    
+    // Clear uv cache
+    await installationShell.runCommand("uv cache clean");
   });
 
   afterEach(async () => {