add test coverage for tier_fastcom

tests/test_speed_fastcom.py

@@ -0,0 +1,128 @@
+import httpx
+import pytest
+
+from netsleuth.config import Speedtest
+from netsleuth.speed import tier_fastcom
+
+
+async def test_tier_fastcom_happy_path(httpx_mock):
+    config = Speedtest()
+    api_url = config.fastcom_api_url
+
+    httpx_mock.add_response(
+        url=f"{api_url}?https=true&token=YXNkZmFzZGxmbnNkYWZoYXNkZmhrYWxm&urlCount=3",
+        json={
+            "targets": [
+                {"url": "https://server1.example.com/speedtest", "location": {"country": "US"}},
+                {"url": "https://server2.example.com/speedtest", "location": {"country": "US"}},
+            ]
+        },
+    )
+
+    httpx_mock.add_response(
+        url="https://server1.example.com/speedtest",
+        content=b"A" * 125000, # 1 Mbit
+    )
+
+    httpx_mock.add_response(
+        url="https://server2.example.com/speedtest",
+        content=b"B" * 250000, # 2 Mbit
+    )
+
+    async with httpx.AsyncClient() as client:
+        result = await tier_fastcom(client, config, timeout=5.0)
+
+    assert result.method == "fastcom"
+    assert result.download_mbps is not None
+    assert result.download_mbps > 0
+    assert result.upload_mbps is None
+    assert result.server == "server1.example.com"
+    assert result.netflix_oca_onnet is True
+
+
+async def test_tier_fastcom_no_targets(httpx_mock):
+    config = Speedtest()
+    api_url = config.fastcom_api_url
+
+    httpx_mock.add_response(
+        url=f"{api_url}?https=true&token=YXNkZmFzZGxmbnNkYWZoYXNkZmhrYWxm&urlCount=3",
+        json={"targets": []},
+    )
+
+    async with httpx.AsyncClient() as client:
+        result = await tier_fastcom(client, config, timeout=5.0)
+
+    assert result.method == "fastcom"
+    assert result.download_mbps == 0.0
+    assert result.server is None
+    assert result.netflix_oca_onnet is None
+
+
+async def test_tier_fastcom_missing_urls(httpx_mock):
+    config = Speedtest()
+    api_url = config.fastcom_api_url
+
+    httpx_mock.add_response(
+        url=f"{api_url}?https=true&token=YXNkZmFzZGxmbnNkYWZoYXNkZmhrYWxm&urlCount=3",
+        json={
+            "targets": [
+                {"location": {"country": "US"}}, # missing URL
+                {"url": "https://server1.example.com/speedtest", "location": {"country": "US"}},
+            ]
+        },
+    )
+
+    httpx_mock.add_response(
+        url="https://server1.example.com/speedtest",
+        content=b"A" * 125000, # 1 Mbit
+    )
+
+    async with httpx.AsyncClient() as client:
+        result = await tier_fastcom(client, config, timeout=5.0)
+
+    assert result.method == "fastcom"
+    assert result.download_mbps is not None
+    assert result.download_mbps > 0
+    assert result.server is None
+    assert result.netflix_oca_onnet is True
+
+
+async def test_tier_fastcom_api_error(httpx_mock):
+    config = Speedtest()
+    api_url = config.fastcom_api_url
+
+    httpx_mock.add_response(
+        url=f"{api_url}?https=true&token=YXNkZmFzZGxmbnNkYWZoYXNkZmhrYWxm&urlCount=3",
+        status_code=500,
+    )
+
+    async with httpx.AsyncClient() as client:
+        with pytest.raises(httpx.HTTPStatusError) as exc:
+            await tier_fastcom(client, config, timeout=5.0)
+
+    assert exc.value.response.status_code == 500
+
+
+async def test_tier_fastcom_download_error(httpx_mock):
+    config = Speedtest()
+    api_url = config.fastcom_api_url
+
+    httpx_mock.add_response(
+        url=f"{api_url}?https=true&token=YXNkZmFzZGxmbnNkYWZoYXNkZmhrYWxm&urlCount=3",
+        json={
+            "targets": [
+                {"url": "https://server1.example.com/speedtest", "location": {"country": "US"}},
+            ]
+        },
+    )
+
+    httpx_mock.add_response(
+        url="https://server1.example.com/speedtest",
+        status_code=404,
+    )
+
+    async with httpx.AsyncClient() as client:
+        with pytest.raises(httpx.HTTPStatusError) as exc:
+            await tier_fastcom(client, config, timeout=5.0)
+
+    assert exc.value.response.status_code == 404