refactor to test

evdspy/EVDSlocal/console/proxy_for_menu.py

@@ -1,19 +1,11 @@
 import os
 
-
-def get_proxies_env() -> dict:
-    from ..config.dotenv import load_env
+def get_proxies_env_helper(a  , b  , a1 , b1 ) -> dict:
     def get_one(x, y):
         if x is None:
             return y
         return x
 
-    load_env()  # http_proxy
-    a = os.getenv("http_proxy")
-    a1 = os.getenv("PROXY_http")
-
-    b = os.getenv("https_proxy")
-    b1 = os.getenv("PROXY_https")
     a = get_one(a, a1)
     b = get_one(b, b1)
 
@@ -27,5 +19,13 @@ def get_one(x, y):
         "http": a,
         "https": b
     }
+def get_proxies_env() -> dict:
+    from ..config.dotenv import load_env
+    load_env()  # http_proxy
+    a = os.getenv("http_proxy")
+    a1 = os.getenv("PROXY_http")
+    b = os.getenv("https_proxy")
+    b1 = os.getenv("PROXY_https")
+    return get_proxies_env_helper( a , b, a1 , b1 )
+
 
-# check_proxy()

evdspy/EVDSlocal/tests/test_get_proxies_env.py

@@ -0,0 +1,14 @@
+from evdspy.EVDSlocal.console.proxy_for_menu import get_proxies_env_helper
+
+
+def test_get_proxies_env_helper(capsys):
+    x1 = (["example:80", None, "example:80", None], {'http': 'example:80'})
+    x2 = (["example:80", "example:80", "example:80", None], {'http': 'example:80', 'https': 'example:80'})
+    x3 = (["example:80", "example:89", "example:90", "example:91"], {'http': 'example:80', 'https': 'example:89'})
+    x4 = ([None, None, "example:90", "example:91"], {'http': 'example:90', 'https': 'example:91'})
+    liste = [x1, x2, x3, x4]
+    with capsys.disabled():
+        for x, expected in liste:
+            res = get_proxies_env_helper(*x)
+            print(res)
+            assert res == expected