mirror of
https://github.com/jokob-sk/NetAlertX.git
synced 2026-01-22 05:38:32 -05:00
112 lines
4.4 KiB
Python
112 lines
4.4 KiB
Python
"""
|
|
Unit tests for authoritative field update handler.
|
|
"""
|
|
|
|
import pytest
|
|
|
|
from server.db.authoritative_handler import (
|
|
can_overwrite_field,
|
|
get_source_for_field_update,
|
|
FIELD_SOURCE_MAP,
|
|
)
|
|
|
|
|
|
class TestCanOverwriteField:
|
|
"""Test the can_overwrite_field authorization logic."""
|
|
|
|
def test_user_source_prevents_overwrite(self):
|
|
"""USER source should prevent any overwrite."""
|
|
assert not can_overwrite_field(
|
|
"devName", "USER", "UNIFIAPI", {"set_always": [], "set_empty": []}, "NewName"
|
|
)
|
|
|
|
def test_locked_source_prevents_overwrite(self):
|
|
"""LOCKED source should prevent any overwrite."""
|
|
assert not can_overwrite_field(
|
|
"devName", "LOCKED", "ARPSCAN", {"set_always": [], "set_empty": []}, "NewName"
|
|
)
|
|
|
|
def test_empty_value_prevents_overwrite(self):
|
|
"""Empty/None values should prevent overwrite."""
|
|
assert not can_overwrite_field(
|
|
"devName", "", "UNIFIAPI", {"set_always": ["devName"], "set_empty": []}, ""
|
|
)
|
|
assert not can_overwrite_field(
|
|
"devName", "", "UNIFIAPI", {"set_always": ["devName"], "set_empty": []}, None
|
|
)
|
|
|
|
def test_set_always_allows_overwrite(self):
|
|
"""SET_ALWAYS should allow overwrite regardless of current source."""
|
|
assert can_overwrite_field(
|
|
"devName", "ARPSCAN", "UNIFIAPI", {"set_always": ["devName"], "set_empty": []}, "NewName"
|
|
)
|
|
assert can_overwrite_field(
|
|
"devName", "NEWDEV", "UNIFIAPI", {"set_always": ["devName"], "set_empty": []}, "NewName"
|
|
)
|
|
|
|
def test_set_empty_allows_overwrite_only_when_empty(self):
|
|
"""SET_EMPTY should allow overwrite only if field is empty or NEWDEV."""
|
|
assert can_overwrite_field(
|
|
"devName", "", "UNIFIAPI", {"set_always": [], "set_empty": ["devName"]}, "NewName"
|
|
)
|
|
assert can_overwrite_field(
|
|
"devName", "NEWDEV", "UNIFIAPI", {"set_always": [], "set_empty": ["devName"]}, "NewName"
|
|
)
|
|
assert not can_overwrite_field(
|
|
"devName", "ARPSCAN", "UNIFIAPI", {"set_always": [], "set_empty": ["devName"]}, "NewName"
|
|
)
|
|
|
|
def test_default_behavior_overwrites_empty_fields(self):
|
|
"""Without SET_ALWAYS/SET_EMPTY, should overwrite only empty fields."""
|
|
assert can_overwrite_field(
|
|
"devName", "", "UNIFIAPI", {"set_always": [], "set_empty": []}, "NewName"
|
|
)
|
|
assert can_overwrite_field(
|
|
"devName", "NEWDEV", "UNIFIAPI", {"set_always": [], "set_empty": []}, "NewName"
|
|
)
|
|
assert not can_overwrite_field(
|
|
"devName", "ARPSCAN", "UNIFIAPI", {"set_always": [], "set_empty": []}, "NewName"
|
|
)
|
|
|
|
def test_whitespace_value_treated_as_empty(self):
|
|
"""Whitespace-only values should be treated as empty."""
|
|
assert not can_overwrite_field(
|
|
"devName", "", "UNIFIAPI", {"set_always": ["devName"], "set_empty": []}, " "
|
|
)
|
|
|
|
|
|
class TestGetSourceForFieldUpdate:
|
|
"""Test source value determination for field updates."""
|
|
|
|
def test_user_override_sets_user_source(self):
|
|
"""User override should set USER source."""
|
|
assert get_source_for_field_update("devName", "UNIFIAPI", is_user_override=True) == "USER"
|
|
|
|
def test_plugin_update_sets_plugin_prefix(self):
|
|
"""Plugin update should set plugin prefix as source."""
|
|
assert get_source_for_field_update("devName", "UNIFIAPI", is_user_override=False) == "UNIFIAPI"
|
|
assert get_source_for_field_update("devLastIP", "ARPSCAN", is_user_override=False) == "ARPSCAN"
|
|
|
|
|
|
class TestFieldSourceMapping:
|
|
"""Test field source mapping is correct."""
|
|
|
|
def test_all_tracked_fields_have_source_counterpart(self):
|
|
"""All tracked fields should have a corresponding *Source field."""
|
|
expected_fields = {
|
|
"devMac": "devMacSource",
|
|
"devName": "devNameSource",
|
|
"devFQDN": "devFqdnSource",
|
|
"devLastIP": "devLastIpSource",
|
|
"devVendor": "devVendorSource",
|
|
"devSSID": "devSsidSource",
|
|
"devParentMAC": "devParentMacSource",
|
|
"devParentPort": "devParentPortSource",
|
|
"devParentRelType": "devParentRelTypeSource",
|
|
"devVlan": "devVlanSource",
|
|
}
|
|
|
|
for field, source in expected_fields.items():
|
|
assert field in FIELD_SOURCE_MAP
|
|
assert FIELD_SOURCE_MAP[field] == source
|