api layer v0.2 - /devices

test/test_devices_endpoints.py

@@ -0,0 +1,70 @@
+import sys
+import pathlib
+import sqlite3
+import random
+import string
+import uuid
+import pytest
+
+INSTALL_PATH = "/app"
+sys.path.extend([f"{INSTALL_PATH}/front/plugins", f"{INSTALL_PATH}/server"])
+
+from helper import timeNowTZ, get_setting_value
+from api_server.api_server_start import app
+
+@pytest.fixture(scope="session")
+def api_token():
+    return get_setting_value("API_TOKEN")
+
+@pytest.fixture
+def client():
+    with app.test_client() as client:
+        yield client
+
+@pytest.fixture
+def test_mac():
+    # Generate a unique MAC for each test run
+    return "AA:BB:CC:" + ":".join(f"{random.randint(0,255):02X}" for _ in range(3))
+
+def auth_headers(token):
+    return {"Authorization": f"Bearer {token}"}
+
+def test_delete_devices_with_macs(client, api_token, test_mac):
+
+    # First create device so it exists
+    payload = {
+        "createNew": True,
+        "name": "Test Device",
+        "owner": "Unit Test",
+        "type": "Router",
+        "vendor": "TestVendor",
+    }
+    resp = client.post(f"/device/{test_mac}", json=payload, headers=auth_headers(api_token))
+    
+    client.post(f"/device/{test_mac}", json={"createNew": True}, headers=auth_headers(api_token))
+
+    # Delete by MAC
+    resp = client.delete("/devices", json={"macs": [test_mac]}, headers=auth_headers(api_token))
+    assert resp.status_code == 200
+    assert resp.json.get("success") is True
+
+def test_delete_test_devices(client, api_token, test_mac):
+
+    # Delete by MAC
+    resp = client.delete("/devices", json={"macs": ["AA:BB:CC:*"]}, headers=auth_headers(api_token))
+    assert resp.status_code == 200
+    assert resp.json.get("success") is True
+
+
+def test_delete_all_empty_macs(client, api_token):
+    resp = client.delete("/devices/empty-macs", headers=auth_headers(api_token))
+    assert resp.status_code == 200
+    # Expect success flag in response
+    assert resp.json.get("success") is True
+
+
+def test_delete_unknown_devices(client, api_token):
+    resp = client.delete("/devices/unknown", headers=auth_headers(api_token))
+    assert resp.status_code == 200
+    assert resp.json.get("success") is True
+