From 52acd5ee92b8e8afa7f50e4cc2fd23f7b0f2e0c1 Mon Sep 17 00:00:00 2001
From: Uxio Fuentefria <6909403+Uxio0@users.noreply.github.com>
Date: Mon, 18 Mar 2024 14:01:55 +0100
Subject: [PATCH] Fix bundler E2E tests

---
 .github/workflows/python.yml                               | 1 +
 .../tests/account_abstraction/test_e2e_bundler_client.py   | 7 ++++---
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/python.yml b/.github/workflows/python.yml
index a18e22b5b..f1e620c60 100644
--- a/.github/workflows/python.yml
+++ b/.github/workflows/python.yml
@@ -73,6 +73,7 @@ jobs:
         DJANGO_SETTINGS_MODULE: config.settings.test
         ETHEREUM_MAINNET_NODE: ${{ secrets.ETHEREUM_MAINNET_NODE }}
         ETHEREUM_POLYGON_NODE: ${{ secrets.ETHEREUM_POLYGON_NODE }}
+        BUNDLER_CLIENT_URL: ${{ secrets.BUNDLER_CLIENT_URL }}
     - name: Test packaging
       run: pip install -e .
     - name: Send results to coveralls
diff --git a/gnosis/eth/tests/account_abstraction/test_e2e_bundler_client.py b/gnosis/eth/tests/account_abstraction/test_e2e_bundler_client.py
index ad91ccc1d..333758e6d 100644
--- a/gnosis/eth/tests/account_abstraction/test_e2e_bundler_client.py
+++ b/gnosis/eth/tests/account_abstraction/test_e2e_bundler_client.py
@@ -7,7 +7,6 @@
 from ...account_abstraction import BundlerClient, UserOperation, UserOperationReceipt
 from ..mocks.mock_bundler import (
     safe_4337_user_operation_hash_mock,
-    supported_entrypoint_mock,
     user_operation_mock,
     user_operation_receipt_mock,
 )
@@ -68,6 +67,8 @@ def test_get_user_operation_and_receipt(self):
         )
 
     def test_supported_entry_points(self):
-        self.assertEqual(
-            self.bundler.supported_entry_points(), supported_entrypoint_mock["result"]
+        supported_entry_points = self.bundler.supported_entry_points()
+        self.assertIn(len(supported_entry_points), (1, 2))
+        self.assertIn(
+            "0x5FF137D4b0FDCD49DcA30c7CF57E578a026d2789", supported_entry_points
         )