Address review comments from PR #1544

test/api_endpoints/test_devices_endpoints.py

@@ -195,11 +195,14 @@ def test_devices_by_status(client, api_token, test_mac):
 
         # 3. Check favorite formatting if devFavorite = 1
         # Update dummy device to favorite
-        client.post(
+        update_resp = client.post(
             f"/device/{test_mac}",
             json={"devFavorite": 1},
             headers=auth_headers(api_token)
         )
+        assert update_resp.status_code == 200
+        assert update_resp.json.get("success") is True
+
         resp_fav = client.get("/devices/by-status?status=my", headers=auth_headers(api_token))
         fav_data = next((d for d in resp_fav.json if d["id"] == test_mac), None)
         assert fav_data is not None

test/api_endpoints/test_mcp_tools_endpoints.py

@@ -30,11 +30,21 @@ def create_dummy(client, api_token, test_mac):
         "devType": "Router",
         "devVendor": "TestVendor",
     }
-    client.post(f"/device/{test_mac}", json=payload, headers=auth_headers(api_token))
+    response = client.post(f"/device/{test_mac}", json=payload, headers=auth_headers(api_token))
+    assert response.status_code in [200, 201], (
+        f"Expected status 200/201 for device creation, got {response.status_code}. "
+        f"Response body: {response.get_data(as_text=True)}"
+    )
+    return response
 
 
 def delete_dummy(client, api_token, test_mac):
-    client.delete("/devices", json={"macs": [test_mac]}, headers=auth_headers(api_token))
+    response = client.delete("/devices", json={"macs": [test_mac]}, headers=auth_headers(api_token))
+    assert response.status_code == 200, (
+        f"Expected status 200 for device deletion, got {response.status_code}. "
+        f"Response body: {response.get_data(as_text=True)}"
+    )
+    return response
 
 
 # --- Device Search Tests ---