From 4f447ba1d28fd3619f62ab40319de07a48ac5bf2 Mon Sep 17 00:00:00 2001
From: R1kaB3rN <100738684+R1kaB3rN@users.noreply.github.com>
Date: Wed, 11 Dec 2024 19:12:12 -0800
Subject: [PATCH] umu_test: update test

---
 umu/umu_test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/umu/umu_test.py b/umu/umu_test.py
index f1c83cd7..c68c358a 100644
--- a/umu/umu_test.py
+++ b/umu/umu_test.py
@@ -1175,6 +1175,7 @@ def test_ge_proton_none(self):
         Tests the case when the user has no internet connection or GE-Proton
         wasn't found in local system.
         """
+        mock_session_pools = (MagicMock(), MagicMock())
         with (
             self.assertRaises(FileNotFoundError),
             patch.object(umu_proton, "_fetch_releases", return_value=None),
@@ -1182,12 +1183,11 @@ def test_ge_proton_none(self):
             patch.object(
                 umu_proton, "_get_from_steamcompat", return_value=None
             ),
-            ThreadPoolExecutor() as thread_pool,
         ):
             os.environ["WINEPREFIX"] = self.test_file
             os.environ["GAMEID"] = self.test_file
             os.environ["PROTONPATH"] = "GE-Proton"
-            umu_run.check_env(self.env, thread_pool)
+            umu_run.check_env(self.env, mock_session_pools)
             self.assertFalse(
                 os.environ.get("PROTONPATH"), "Expected empty string"
             )