Test for _get_check

Author: Snuffy2

tests/test_pyopnsense.py

@@ -146,6 +146,46 @@ async def test_safe_dict_post_and_list_post(monkeypatch, make_client) -> None:
     assert result_empty_list == []
 
 
+@pytest.mark.asyncio
+async def test_get_check(make_client) -> None:
+    """Test _get_check method returns True for ok responses, False otherwise."""
+    session = MagicMock(spec=aiohttp.ClientSession)
+
+    # Fake response class for testing
+    class FakeResp:
+        def __init__(self, status=500, ok=False):
+            self.status = status
+            self.reason = "Test"
+            self.ok = ok
+            self.request_info = MagicMock()
+            self.history = []
+            self.headers = {}
+
+        async def __aenter__(self):
+            return self
+
+        async def __aexit__(self, exc_type, exc, tb):
+            return False
+
+    # Test successful response (ok=True)
+    session.get = lambda *a, **k: FakeResp(status=200, ok=True)
+    client = make_client(session=session)
+    result = await client._get_check("/api/test")
+    assert result is True
+
+    # Test failed response (ok=False)
+    session.get = lambda *a, **k: FakeResp(status=404, ok=False)
+    client = make_client(session=session)
+    result = await client._get_check("/api/test")
+    assert result is False
+
+    # Test 403 response specifically
+    session.get = lambda *a, **k: FakeResp(status=403, ok=False)
+    client = make_client(session=session)
+    result = await client._get_check("/api/test")
+    assert result is False
+
+
 @pytest.mark.asyncio
 async def test_get_ip_key_sorting(make_client) -> None:
     """Sort IP-like items using get_ip_key ordering."""