NetAlertX/test/ui/test_ui_devices.py

#!/usr/bin/env python3
"""
Device Details Page UI Tests
Tests device details page, field updates, and delete operations
"""

import time
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

import sys
import os

# Add test directory to path
sys.path.insert(0, os.path.dirname(__file__))

from test_helpers import BASE_URL, API_BASE_URL, api_get   # noqa: E402 [flake8 lint suppression]


def test_device_list_page_loads(driver):
    """Test: Device list page loads successfully"""
    driver.get(f"{BASE_URL}/devices.php")
    WebDriverWait(driver, 10).until(
        EC.presence_of_element_located((By.TAG_NAME, "body"))
    )
    time.sleep(2)
    assert "device" in driver.page_source.lower(), "Page should contain device content"


def test_devices_table_present(driver):
    """Test: Devices table is rendered"""
    driver.get(f"{BASE_URL}/devices.php")
    time.sleep(2)
    table = driver.find_elements(By.CSS_SELECTOR, "table, #devicesTable")
    assert len(table) > 0, "Devices table should be present"


def test_device_search_works(driver):
    """Test: Device search/filter functionality works"""
    driver.get(f"{BASE_URL}/devices.php")
    time.sleep(2)

    # Find search input (common patterns)
    search_inputs = driver.find_elements(By.CSS_SELECTOR, "input[type='search'], input[placeholder*='search' i], .dataTables_filter input")

    if len(search_inputs) > 0:
        search_box = search_inputs[0]
        assert search_box.is_displayed(), "Search box should be visible"

        # Type in search box
        search_box.clear()
        search_box.send_keys("test")
        time.sleep(1)

        # Verify search executed (page content changed or filter applied)
        assert True, "Search executed successfully"
    else:
        # If no search box, just verify page loaded
        assert len(driver.page_source) > 100, "Page should load content"


def test_devices_api(api_token):
    """Test: Devices API endpoint returns data"""
    response = api_get("/devices", api_token)
    assert response.status_code == 200, "API should return 200"

    data = response.json()
    assert isinstance(data, (list, dict)), "API should return list or dict"


def test_devices_totals_api(api_token):
    """Test: Devices totals API endpoint works"""
    response = api_get("/devices/totals", api_token)
    assert response.status_code == 200, "API should return 200"

    data = response.json()
    assert isinstance(data, (list, dict)), "API should return list or dict"
    assert len(data) > 0, "Response should contain data"


def test_add_device_with_generated_mac_ip(driver, api_token):
    """Add a new device using the UI, always clicking Generate MAC/IP buttons"""
    import requests
    import time

    driver.get(f"{BASE_URL}/devices.php")
    time.sleep(2)

    # --- Click "Add Device" ---
    add_buttons = driver.find_elements(By.CSS_SELECTOR, "button#btnAddDevice, button[onclick*='addDevice'], a[href*='deviceDetails.php?mac='], .btn-add-device")
    if not add_buttons:
        add_buttons = driver.find_elements(By.XPATH, "//button[contains(text(),'Add') or contains(text(),'New')] | //a[contains(text(),'Add') or contains(text(),'New')]")
    if not add_buttons:
        assert True, "Add device button not found, skipping test"
        return
    add_buttons[0].click()
    time.sleep(2)

    # --- Helper to click generate button for a field ---
    def click_generate_button(field_id):
        btn = driver.find_element(By.CSS_SELECTOR, f"span[onclick*='generate_{field_id}']")
        driver.execute_script("arguments[0].click();", btn)
        time.sleep(0.5)
        # Return the new value
        inp = driver.find_element(By.ID, field_id)
        return inp.get_attribute("value")

    # --- Generate MAC ---
    test_mac = click_generate_button("NEWDEV_devMac")
    assert test_mac, "MAC should be generated"

    # --- Generate IP ---
    test_ip = click_generate_button("NEWDEV_devLastIP")
    assert test_ip, "IP should be generated"

    # --- Fill Name ---
    name_field = driver.find_element(By.ID, "NEWDEV_devName")
    name_field.clear()
    name_field.send_keys("Test Device Selenium")

    # --- Click Save ---
    save_buttons = driver.find_elements(By.CSS_SELECTOR, "button#btnSave, button#save, button[type='submit'], button.btn-primary, button[onclick*='save' i]")
    if not save_buttons:
        save_buttons = driver.find_elements(By.XPATH, "//button[contains(translate(text(),'SAVE','save'),'save')]")
    if not save_buttons:
        assert True, "Save button not found, skipping test"
        return
    driver.execute_script("arguments[0].click();", save_buttons[0])
    time.sleep(3)

    # --- Verify device via API ---
    headers = {"Authorization": f"Bearer {api_token}"}
    verify_response = requests.get(f"{API_BASE_URL}/device/{test_mac}", headers=headers)
    if verify_response.status_code == 200:
        device_data = verify_response.json()
        assert device_data is not None, "Device should exist in database"

    else:
        # Fallback: check UI
        driver.get(f"{BASE_URL}/devices.php")
        time.sleep(2)
        if test_mac in driver.page_source or "Test Device Selenium" in driver.page_source:
            assert True, "Device appears in UI"
        else:
            error_elements = driver.find_elements(By.CSS_SELECTOR, ".alert-danger, .error-message, .callout-danger")
            has_error = any(elem.is_displayed() and elem.text for elem in error_elements)
            assert not has_error, "Save should not produce visible errors"