"""Tests for registry.redact — the safety net for diagnostics downloads."""

import json
from pathlib import Path

from custom_components.localthings.registry.batch import parse_device0_batch
from custom_components.localthings.registry.redact import REDACTED, redact_resources

FIXTURES = Path(__file__).resolve().parent / "fixtures"


def _load(name: str) -> dict:
    data = json.loads((FIXTURES / name).read_text())
    return parse_device0_batch(data["device0"])


def test_redacts_known_sensitive_fields_in_dishwasher_dump():
    resources = _load("dishwasher_device.json")
    redacted = redact_resources(resources)

    info = redacted["/information/vs/0"]
    assert info["x.com.samsung.da.serialNum"] == REDACTED
    assert info["x.com.samsung.da.otnDUID"] == REDACTED

    wireless = redacted["/wirelessinfo/vs/0"]
    assert wireless["macaddressWiFi"] == REDACTED
    assert wireless["macaddressBLE"] == REDACTED

    provisioning = redacted["/voice/provisioning/vs/0"]
    headers = provisioning["voice.provisioning.headers"]
    assert headers["login_id"] == REDACTED
    deviceinfo = provisioning["voice.provisioning.deviceinfo"]
    assert deviceinfo["voice.provisioning.deviceinfo.accesstoken"] == REDACTED
    assert deviceinfo["voice.provisioning.deviceinfo.deviceid"] == REDACTED
    assert deviceinfo["voice.provisioning.deviceinfo.userid"] == REDACTED


def test_ordinary_state_fields_survive_untouched():
    resources = _load("dishwasher_device.json")
    redacted = redact_resources(resources)

    op_state = redacted["/operational/state/vs/0"]
    assert op_state["x.com.samsung.da.state"] == "Run"
    assert op_state["x.com.samsung.da.progress"] == "Finish"

    power = redacted["/power/vs/0"]
    assert power["x.com.samsung.da.power"] == "On"

    dishwasher = redacted["/dishwasher/vs/0"]
    assert dishwasher["x.com.samsung.da.sanitize"] == "On"
    assert dishwasher["x.com.samsung.da.rinseLevel"] == "4"

    alarms = redacted["/alarms/vs/0"]["x.com.samsung.da.items"]
    assert alarms[0]["x.com.samsung.da.code"] == "SNSF_Reached"


def test_redacts_known_sensitive_fields_in_refrigerator_dump():
    resources = _load("refrigerator_device.json")
    redacted = redact_resources(resources)

    info = redacted["/information/vs/0"]
    assert info["x.com.samsung.da.serialNum"] == REDACTED

    wireless = redacted["/wirelessinfo/vs/0"]
    assert wireless["macaddressWiFi"] == REDACTED
    assert wireless["macaddressBLE"] == REDACTED


def test_redact_resources_does_not_mutate_input():
    resources = _load("dishwasher_device.json")
    original_serial = resources["/information/vs/0"]["x.com.samsung.da.serialNum"]

    redact_resources(resources)

    assert resources["/information/vs/0"]["x.com.samsung.da.serialNum"] == original_serial


def test_redacts_bare_ocf_identity_keys():
    """/oic/d and /oic/p identify the unit with two-letter keys ('di', 'pi')
    that the substring rules can't see."""
    redacted = redact_resources(
        {
            "/oic/d": {
                "di": "ab-cd-ef",
                "n": "Marc's Fridge",
                "rt": ["oic.wk.d", "oic.d.refrigerator"],
            },
            "/oic/p": {"pi": "12-34-56", "mnmo": "RF9000B"},
        }
    )

    assert redacted["/oic/d"]["di"] == REDACTED
    assert redacted["/oic/p"]["pi"] == REDACTED
    # 'n' is free text the owner sets from the SmartThings app, so it can
    # carry a person's name -- redacted too. `rt`, the device-type signal
    # we actually want out of /oic/d, is not.
    assert redacted["/oic/d"]["n"] == REDACTED
    assert redacted["/oic/d"]["rt"] == ["oic.wk.d", "oic.d.refrigerator"]
    assert redacted["/oic/p"]["mnmo"] == "RF9000B"


def test_bare_key_redaction_does_not_leak_into_substring_matching():
    """The bare keys are whole-key matches only -- plenty of ordinary
    appliance fields contain those letters and must survive untouched."""
    redacted = redact_resources(
        {
            "/x": {
                "condition": "Normal",
                "display": "On",
                "dispenser": "Cubed",
                "humidity": "45",
                "spinSpeed": "1200",
                "name": "FilterProgress",
            },
        }
    )

    assert redacted["/x"] == {
        "condition": "Normal",
        "display": "On",
        "dispenser": "Cubed",
        "humidity": "45",
        "spinSpeed": "1200",
        "name": "FilterProgress",
    }