From ce466e3f7fb1218efaee0e2a7e5e2427beacab6a Mon Sep 17 00:00:00 2001 From: Deeman Date: Sun, 1 Mar 2026 22:36:30 +0100 Subject: [PATCH] test(proxy): update supervisor tests for 2-tier proxy (no Webshare) Co-Authored-By: Claude Sonnet 4.6 --- web/tests/test_supervisor.py | 26 +++++++++++--------------- 1 file changed, 11 insertions(+), 15 deletions(-) diff --git a/web/tests/test_supervisor.py b/web/tests/test_supervisor.py index be721e7..a04a502 100644 --- a/web/tests/test_supervisor.py +++ b/web/tests/test_supervisor.py @@ -286,24 +286,20 @@ class TestLoadProxyTiers: assert len(tiers) == 1 assert tiers[0] == ["http://res1:8080"] - def test_three_tiers_correct_order(self, monkeypatch): + def test_two_tiers_correct_order(self, monkeypatch): self._clear_proxy_env(monkeypatch) - with patch("padelnomics_extract.proxy.fetch_webshare_proxies", return_value=["http://user:pass@1.2.3.4:1080"]): - monkeypatch.setenv("WEBSHARE_DOWNLOAD_URL", "http://example.com/list") - monkeypatch.setenv("PROXY_URLS_DATACENTER", "http://dc1:8080") - monkeypatch.setenv("PROXY_URLS_RESIDENTIAL", "http://res1:8080") - tiers = load_proxy_tiers() - assert len(tiers) == 3 - assert tiers[0] == ["http://user:pass@1.2.3.4:1080"] # free - assert tiers[1] == ["http://dc1:8080"] # datacenter - assert tiers[2] == ["http://res1:8080"] # residential + monkeypatch.setenv("PROXY_URLS_DATACENTER", "http://dc1:8080") + monkeypatch.setenv("PROXY_URLS_RESIDENTIAL", "http://res1:8080") + tiers = load_proxy_tiers() + assert len(tiers) == 2 + assert tiers[0] == ["http://dc1:8080"] # datacenter (tier 1) + assert tiers[1] == ["http://res1:8080"] # residential (tier 2) - def test_webshare_fetch_failure_skips_tier(self, monkeypatch): + def test_webshare_env_var_is_ignored(self, monkeypatch): self._clear_proxy_env(monkeypatch) - with patch("padelnomics_extract.proxy.fetch_webshare_proxies", return_value=[]): - monkeypatch.setenv("WEBSHARE_DOWNLOAD_URL", "http://example.com/list") - monkeypatch.setenv("PROXY_URLS_DATACENTER", "http://dc1:8080") - tiers = load_proxy_tiers() + monkeypatch.setenv("WEBSHARE_DOWNLOAD_URL", "http://example.com/list") + monkeypatch.setenv("PROXY_URLS_DATACENTER", "http://dc1:8080") + tiers = load_proxy_tiers() assert len(tiers) == 1 assert tiers[0] == ["http://dc1:8080"]