NetAlertX/test/test_device_endpoints.py

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_create_device(client, api_token, test_mac):
    payload = {
        "createNew": True,
        "devType": "Test Device",
        "devOwner": "Unit Test",
        "devType": "Router",
        "devVendor": "TestVendor",
    }
    resp = client.post(f"/device/{test_mac}", json=payload, headers=auth_headers(api_token))
    assert resp.status_code == 200
    assert resp.json.get("success") is True

def test_get_device(client, api_token, test_mac):
    # First create it
    client.post(f"/device/{test_mac}", json={"createNew": True}, headers=auth_headers(api_token))
    # Then retrieve it
    resp = client.get(f"/device/{test_mac}", headers=auth_headers(api_token))
    assert resp.status_code == 200
    assert resp.json.get("devMac") == test_mac

def test_reset_device_props(client, api_token, test_mac):
    client.post(f"/device/{test_mac}", json={"createNew": True}, headers=auth_headers(api_token))
    resp = client.post(f"/device/{test_mac}/reset-props", json={}, headers=auth_headers(api_token))
    assert resp.status_code == 200
    assert resp.json.get("success") is True

def test_delete_device_events(client, api_token, test_mac):
    client.post(f"/device/{test_mac}", json={"createNew": True}, headers=auth_headers(api_token))
    resp = client.delete(f"/device/{test_mac}/events/delete", headers=auth_headers(api_token))
    assert resp.status_code == 200
    assert resp.json.get("success") is True

def test_delete_device(client, api_token, test_mac):
    client.post(f"/device/{test_mac}", json={"createNew": True}, headers=auth_headers(api_token))
    resp = client.delete(f"/device/{test_mac}/delete", headers=auth_headers(api_token))
    assert resp.status_code == 200
    assert resp.json.get("success") is True

def test_copy_device(client, api_token, test_mac):
    # Step 1: Create the source device
    payload = {"createNew": True}
    resp = client.post(f"/device/{test_mac}", json=payload, headers=auth_headers(api_token))
    assert resp.status_code == 200
    assert resp.json.get("success") is True

    # Step 2: Generate a target MAC
    target_mac = "AA:BB:CC:" + ":".join(f"{random.randint(0,255):02X}" for _ in range(3))

    # Step 3: Copy device
    copy_payload = {"macFrom": test_mac, "macTo": target_mac}
    resp = client.post("/device/copy", json=copy_payload, headers=auth_headers(api_token))
    assert resp.status_code == 200
    assert resp.json.get("success") is True

    # Step 4: Verify new device exists
    resp = client.get(f"/device/{target_mac}", headers=auth_headers(api_token))
    assert resp.status_code == 200
    assert resp.json.get("devMac") == target_mac

    # Cleanup: delete both devices
    client.delete(f"/device/{test_mac}/delete", headers=auth_headers(api_token))
    client.delete(f"/device/{target_mac}/delete", headers=auth_headers(api_token))

def test_update_device_column(client, api_token, test_mac):
    # First, create the device
    client.post(
        f"/device/{test_mac}",
        json={"createNew": True},
        headers=auth_headers(api_token),
    )

    # Update its parent MAC
    resp = client.post(
        f"/device/{test_mac}/update-column",
        json={"columnName": "devParentMAC", "columnValue": "Internet"},
        headers=auth_headers(api_token),
    )

    assert resp.status_code == 200
    assert resp.json.get("success") is True

    # Try updating a non-existent device
    resp_missing = client.post(
        "/device/11:22:33:44:55:66/update-column",
        json={"columnName": "devParentMAC", "columnValue": "Internet"},
        headers=auth_headers(api_token),
    )

    assert resp_missing.status_code == 404
    assert resp_missing.json.get("success") is False