diracx-testing 0.0.1a25__py3-none-any.whl → 0.0.1a26__py3-none-any.whl

diracx/testing/utils.py

@@ -18,8 +18,8 @@ from typing import TYPE_CHECKING, Generator
 from urllib.parse import parse_qs, urljoin, urlparse
 from uuid import uuid4
 
+import httpx
 import pytest
-import requests
 
 if TYPE_CHECKING:
     from diracx.core.settings import DevelopmentSettings
@@ -616,7 +616,7 @@ async def test_login(monkeypatch, capfd, cli_env):
 
     poll_attempts = 0
 
-    def fake_sleep(*args, **kwargs):
+    async def fake_sleep(*args, **kwargs):
         nonlocal poll_attempts
 
         # Keep track of the number of times this is called
@@ -629,13 +629,13 @@ async def test_login(monkeypatch, capfd, cli_env):
             match = re.search(rf"{cli_env['DIRACX_URL']}[^\n]+", captured.out)
             assert match, captured
 
-            do_device_flow_with_dex(match.group(), cli_env["DIRACX_CA_PATH"])
+            await do_device_flow_with_dex(match.group(), cli_env["DIRACX_CA_PATH"])
 
         # Ensure we don't poll forever
         assert poll_attempts <= 100
 
         # Reduce the sleep duration to zero to speed up the test
-        return unpatched_sleep(0)
+        await unpatched_sleep(0.0)
 
     # We monkeypatch asyncio.sleep to provide a hook to run the actions that
     # would normally be done by a user. This includes capturing the login URL
@@ -650,7 +650,7 @@ async def test_login(monkeypatch, capfd, cli_env):
 
     # Run the login command
     with monkeypatch.context() as m:
-        m.setattr("asyncio.sleep", fake_sleep)
+        m.setattr("diracx.cli.auth.sleep", fake_sleep)
         await cli.auth.login(vo="diracAdmin", group=None, property=None)
     captured = capfd.readouterr()
     assert "Login successful!" in captured.out
@@ -664,7 +664,7 @@ async def test_login(monkeypatch, capfd, cli_env):
     return expected_credentials_path.read_text()
 
 
-def do_device_flow_with_dex(url: str, ca_path: str) -> None:
+async def do_device_flow_with_dex(url: str, ca_path: str) -> None:
     """Do the device flow with dex."""
 
     class DexLoginFormParser(HTMLParser):
@@ -672,10 +672,14 @@ def do_device_flow_with_dex(url: str, ca_path: str) -> None:
             nonlocal action_url
             if "form" in str(tag):
                 assert action_url is None
-                action_url = urljoin(login_page_url, dict(attrs)["action"])
+                action_url = urljoin(str(login_page_url), dict(attrs)["action"])
 
+    ssl_context = ssl.create_default_context(cafile=ca_path)
+    client_kwargs = dict(verify=ssl_context, follow_redirects=True)
     # Get the login page
-    r = requests.get(url, verify=ca_path)
+    async with httpx.AsyncClient(**client_kwargs) as client:
+        r = await client.get(url)
+
     r.raise_for_status()
     login_page_url = r.url  # This is not the same as URL as we redirect to dex
     login_page_body = r.text
@@ -686,19 +690,24 @@ def do_device_flow_with_dex(url: str, ca_path: str) -> None:
     assert action_url is not None, login_page_body
 
     # Do the actual login
-    r = requests.post(
-        action_url,
-        data={"login": "admin@example.com", "password": "password"},
-        verify=ca_path,
-    )
+    async with httpx.AsyncClient(**client_kwargs) as client:
+        r = await client.post(
+            action_url,
+            data={"login": "admin@example.com", "password": "password"},
+        )
+
     r.raise_for_status()
     approval_url = r.url  # This is not the same as URL as we redirect to dex
     # Do the actual approval
-    r = requests.post(
-        approval_url,
-        {"approval": "approve", "req": parse_qs(urlparse(r.url).query)["req"][0]},
-        verify=ca_path,
-    )
+
+    async with httpx.AsyncClient(**client_kwargs) as client:
+        r = await client.post(
+            approval_url,
+            data={
+                "approval": "approve",
+                "req": parse_qs(urlparse(str(r.url)).query)["req"][0],
+            },
+        )
 
     # This should have redirected to the DiracX page that shows the login is complete
     assert "Please close the window" in r.text

{diracx_testing-0.0.1a25.dist-info → diracx_testing-0.0.1a26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: diracx-testing
-Version: 0.0.1a25
+Version: 0.0.1a26
 Summary: TODO
 License: GPL-3.0-only
 Classifier: Intended Audience :: Science/Research
@@ -14,5 +14,6 @@ Requires-Dist: pytest
 Requires-Dist: pytest-asyncio
 Requires-Dist: pytest-cov
 Requires-Dist: pytest-xdist
+Requires-Dist: httpx
 Provides-Extra: testing
 Requires-Dist: diracx-testing; extra == "testing"

{diracx_testing-0.0.1a25.dist-info → diracx_testing-0.0.1a26.dist-info}/RECORD

@@ -4,8 +4,8 @@ diracx/testing/entrypoints.py,sha256=MbH0VLUQz96XPdHzb7XWFwYtWEitAqPrrGM1H1FzDLo
 diracx/testing/mock_osdb.py,sha256=hHuvmQZ3212SaSGX11dQv4ki3ZWsmqJZFYwdn6kg2MA,6029
 diracx/testing/osdb.py,sha256=m6mUBLnGOoQLTCIBie9P2GhmLMybrgzIrlIYfhF1_Ss,3230
 diracx/testing/routers.py,sha256=UW-TnikMQgcNxF5sUZD5DWoucGiCpP6s8mYmuahDiSc,979
-diracx/testing/utils.py,sha256=Ak0lzWKXgB4Q9oIqr4a3x1IZ6KO9vX8TSeeld8rid7c,24032
-diracx_testing-0.0.1a25.dist-info/METADATA,sha256=5Jottro46jEDYowuLkmLS8P5FvEobp6iS46c45NjJ_I,613
-diracx_testing-0.0.1a25.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-diracx_testing-0.0.1a25.dist-info/top_level.txt,sha256=vJx10tdRlBX3rF2Psgk5jlwVGZNcL3m_7iQWwgPXt-U,7
-diracx_testing-0.0.1a25.dist-info/RECORD,,
+diracx/testing/utils.py,sha256=uPI7UssaOkDvaG3Lq1-A-IJCYvhQ1F_HcQsI6N6O-Bw,24417
+diracx_testing-0.0.1a26.dist-info/METADATA,sha256=Cye4WM98UB1L1irFN3KHkgq4Iqd74Gxi-Ej8U-kc9fw,634
+diracx_testing-0.0.1a26.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+diracx_testing-0.0.1a26.dist-info/top_level.txt,sha256=vJx10tdRlBX3rF2Psgk5jlwVGZNcL3m_7iQWwgPXt-U,7
+diracx_testing-0.0.1a26.dist-info/RECORD,,