Add basic unittests

Author: fabaff

netdata/tests/__init__.py

@@ -0,0 +1 @@
+"""Tests for the Python Netdata client."""

netdata/tests/test_chart.py

@@ -0,0 +1,34 @@
+"""Test the chart data retrieval."""
+import pytest
+from pytest_httpx import HTTPXMock
+
+from netdata import Netdata, exceptions
+
+RESPONSE_VALID = {
+    "labels": [
+        "time",
+        "guest_nice",
+        "guest",
+        "steal",
+        "softirq",
+        "irq",
+        "user",
+        "system",
+        "nice",
+        "iowait",
+    ],
+    "data": [
+        [1634256154, 0, 0, 0, 0.2506266, 3.759398, 32.45614, 6.015038, 0, 0.1253133]
+    ],
+}
+
+
+@pytest.mark.asyncio
+async def test_chart_valid(httpx_mock: HTTPXMock):
+    """Test a valid response."""
+    httpx_mock.add_response(json=RESPONSE_VALID)
+
+    client = Netdata("localhost")
+    await client.get_chart("system.cpu")
+
+    assert client.values.get("system") == 6.015038

netdata/tests/test_connection.py

@@ -0,0 +1,23 @@
+"""Test the connection."""
+import pytest
+from pytest_httpx import HTTPXMock
+
+from netdata import Netdata
+import httpx
+
+
+@pytest.mark.asyncio
+async def test_timeout(httpx_mock: HTTPXMock):
+    """Test if the connection is hitting the timeout."""
+
+    def raise_timeout(request, extensions: dict):
+        """Set the timeout for the requests."""
+        raise httpx.ReadTimeout(
+            f"Unable to read within {extensions['timeout']}", request=request
+        )
+
+    httpx_mock.add_callback(raise_timeout)
+
+    with pytest.raises(httpx.ReadTimeout):
+        client = Netdata("localhost")
+        await client.get_info()

netdata/tests/test_info.py

@@ -0,0 +1,28 @@
+"""Test the chart data retrieval."""
+import pytest
+from pytest_httpx import HTTPXMock
+
+from netdata import Netdata
+
+RESPONSE_VALID = {
+    "version": "v1.31.0",
+    "uid": "792bb46a-fb11-11e7-b935-e6a17492adc8",
+    "mirrored_hosts": ["london3"],
+    "mirrored_hosts_status": [
+        {
+            "guid": "792bb46a-fb11-11e7-b935-e6a17492adc8",
+            "claim_id": "792bb46a-fb11-11e7-b935-e6a17492adc8",
+        }
+    ],
+}
+
+
+@pytest.mark.asyncio
+async def test_info_valid(httpx_mock: HTTPXMock):
+    """Test a valid response."""
+    httpx_mock.add_response(json=RESPONSE_VALID)
+
+    client = Netdata("localhost")
+    await client.get_info()
+
+    assert client.info.get("version") == "v1.31.0"