"""
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