add tests for tier_cloudflare

tests/test_speed_cloudflare.py

@@ -0,0 +1,72 @@
+import pytest
+import httpx
+from pytest_httpx import HTTPXMock
+
+from netsleuth.config import Speedtest
+from netsleuth.speed import tier_cloudflare
+
+
+@pytest.mark.asyncio
+async def test_tier_cloudflare_happy_path(httpx_mock: HTTPXMock):
+    cfg = Speedtest(
+        download_sizes_bytes=[100, 200],
+        upload_sizes_bytes=[50],
+        cloudflare_base_url="https://speed.cloudflare.com"
+    )
+
+    httpx_mock.add_response(
+        url="https://speed.cloudflare.com/__down?bytes=100",
+        method="GET",
+        content=b"x" * 100,
+        headers={"Server-Timing": 'cfL4;desc="?rtt=12345&min_rtt=11000"'}
+    )
+
+    httpx_mock.add_response(
+        url="https://speed.cloudflare.com/__down?bytes=200",
+        method="GET",
+        content=b"x" * 200,
+        headers={"Server-Timing": 'cfL4;desc="?rtt=12345&min_rtt=11000"'}
+    )
+
+    httpx_mock.add_response(
+        url="https://speed.cloudflare.com/__up",
+        method="POST",
+        content=b""
+    )
+
+    async with httpx.AsyncClient() as client:
+        result = await tier_cloudflare(client, cfg, timeout=2.0)
+
+    assert result.method == "cloudflare"
+    assert result.server == "speed.cloudflare.com"
+
+    # We used lengths 100 and 200 for download, so some throughput should be calculated
+    assert result.download_mbps is not None
+    assert result.download_mbps > 0.0
+
+    # We used length 50 for upload
+    assert result.upload_mbps is not None
+    assert result.upload_mbps > 0.0
+
+    assert result.cfL4_stats is not None
+    assert result.cfL4_stats.rtt_ms == 12.345
+    assert result.cfL4_stats.min_rtt_ms == 11.0
+
+
+@pytest.mark.asyncio
+async def test_tier_cloudflare_http_error(httpx_mock: HTTPXMock):
+    cfg = Speedtest(
+        download_sizes_bytes=[100],
+        upload_sizes_bytes=[],
+        cloudflare_base_url="https://speed.cloudflare.com"
+    )
+
+    httpx_mock.add_response(
+        url="https://speed.cloudflare.com/__down?bytes=100",
+        method="GET",
+        status_code=500
+    )
+
+    async with httpx.AsyncClient() as client:
+        with pytest.raises(httpx.HTTPStatusError):
+            await tier_cloudflare(client, cfg, timeout=2.0)