Fix testing

Author: CoMPaTech

airos/airos8data.py

@@ -18,7 +18,7 @@ class WirelessMode(Enum):
     """Enum definition."""
 
     AccessPoint_PointToPoint = "ap-ptp"
-    Station_PointToPoint = "ap-ptp"
+    Station_PointToPoint = "sta-ptp"
     # More to be added when known
 
 

tests/test_stations.py

@@ -11,6 +11,8 @@
 import aiofiles
 import aiohttp
 
+from . import AirOSData
+
 
 async def _read_fixture(fixture: str = "ap-ptp"):
     """Read fixture file per device type."""
@@ -27,7 +29,7 @@ async def _read_fixture(fixture: str = "ap-ptp"):
 
 @pytest.mark.parametrize("mode", ["ap-ptp", "sta-ptp"])
 @pytest.mark.asyncio
-async def test_ap(airos_device, base_url, mode):
+async def test_ap_json(airos_device, base_url, mode):
     """Test device operation."""
     cookie = SimpleCookie()
     cookie["session_id"] = "test-cookie"
@@ -54,12 +56,47 @@ async def test_ap(airos_device, base_url, mode):
         patch.object(airos_device.session, "get", return_value=mock_status_response),
     ):
         assert await airos_device.login()
-        status = await airos_device.status()
+        status = await airos_device.status(return_json=True)
 
         # Verify the fixture returns the correct mode
         assert status.get("wireless", {}).get("mode") == mode
 
 
+@pytest.mark.parametrize("mode", ["ap-ptp", "sta-ptp"])
+@pytest.mark.asyncio
+async def test_ap_object(airos_device, base_url, mode):
+    """Test device operation."""
+    cookie = SimpleCookie()
+    cookie["session_id"] = "test-cookie"
+    cookie["AIROS_TOKEN"] = "abc123"
+
+    # --- Prepare fake POST /api/auth response with cookies ---
+    mock_login_response = MagicMock()
+    mock_login_response.__aenter__.return_value = mock_login_response
+    mock_login_response.text = AsyncMock(return_value="{}")
+    mock_login_response.status = 200
+    mock_login_response.cookies = cookie
+    mock_login_response.headers = {"X-CSRF-ID": "test-csrf-token"}
+    # --- Prepare fake GET /api/status response ---
+    fixture_data = await _read_fixture(mode)
+    mock_status_payload = fixture_data
+    mock_status_response = MagicMock()
+    mock_status_response.__aenter__.return_value = mock_status_response
+    mock_status_response.text = AsyncMock(return_value=json.dumps(fixture_data))
+    mock_status_response.status = 200
+    mock_status_response.json = AsyncMock(return_value=mock_status_payload)
+
+    with (
+        patch.object(airos_device.session, "post", return_value=mock_login_response),
+        patch.object(airos_device.session, "get", return_value=mock_status_response),
+    ):
+        assert await airos_device.login()
+        status: AirOSData = await airos_device.status()  # Implies return_json = False
+
+        # Verify the fixture returns the correct mode
+        assert status.wireless.mode.value == mode
+
+
 @pytest.mark.asyncio
 async def test_reconnect(airos_device, base_url):
     """Test reconnect client."""