# Safe unit regressions: load only function definitions, never dispatch WELA or touch Windows policy.
$ErrorActionPreference = 'Stop'
$sourcePath = Join-Path $PSScriptRoot '../WELA.ps1'
$tokens = $null
$parseErrors = $null
$ast = [System.Management.Automation.Language.Parser]::ParseFile($sourcePath, [ref]$tokens, [ref]$parseErrors)
if ($parseErrors.Count) { throw ($parseErrors | Out-String) }
foreach ($functionName in @('Get-WelaOutgoingNtlmPolicySource', 'Get-WelaOutgoingNtlmState', 'Set-WelaOutgoingNtlmPolicy')) {
    $definition = $ast.Find({ param($node) $node -is [System.Management.Automation.Language.FunctionDefinitionAst] -and $node.Name -eq $functionName }, $true)
    if (-not $definition) { throw "Missing function $functionName" }
    . ([scriptblock]::Create($definition.Extent.Text))
}
function Assert-Equal($Actual, $Expected, [string]$Message) {
    if ($Actual -cne $Expected) { throw "$Message. Expected '$Expected', got '$Actual'." }
    $script:assertions++
}
function Assert-Throws([scriptblock]$Action, [string]$Message) {
    $threw = $false
    try { & $Action } catch { $threw = $true }
    Assert-Equal $threw $true $Message
}
function Reset-Policy($Value, [string]$Type = 'DWord') {
    $script:value = $Value
    $script:type = $Type
    $script:keyExists = $true
    $script:writes = 0
    $script:prompts = 0
    $script:readFails = $false
    $script:typeReadFails = $false
    $script:writeFails = $false
    $script:ignoreWrite = $false
    $script:ignoreTypeWrite = $false
    $script:response = 'Y'
    $script:rsop = @()
    $script:onPrompt = $null
    $script:onRead = $null
    $script:reads = 0
}
function Test-Path { param($LiteralPath, $ErrorAction) return $script:keyExists }
function Get-ItemProperty {
    param($LiteralPath, $ErrorAction)
    $script:reads++
    if ($script:onRead) { & $script:onRead }
    if ($script:readFails) { throw 'Access denied' }
    if ($null -eq $script:value) { return [pscustomobject]@{} }
    return [pscustomobject]@{ RestrictSendingNTLMTraffic = $script:value }
}
function Get-Item {
    param($LiteralPath, $ErrorAction)
    $key = [pscustomobject]@{}
    $key | Add-Member ScriptMethod GetValueKind {
        param($Name)
        if ($script:typeReadFails) { throw 'Value kind unavailable' }
        return [Microsoft.Win32.RegistryValueKind]$script:type
    }
    return $key
}
function New-Item { param($Path, [switch]$Force, $ErrorAction) $script:keyExists = $true }
function Set-ItemProperty {
    param($LiteralPath, $Name, $Value, $Type, $ErrorAction)
    if ($script:writeFails) { throw 'Access denied' }
    $script:writes++
    if (-not $script:ignoreWrite) {
        $script:value = $Value
        if (-not $script:ignoreTypeWrite) { $script:type = $Type }
    }
}
function Get-CimInstance { param($Namespace, $ClassName, $ErrorAction) if ($ClassName -eq 'RSOP_RegistryPolicySetting') { return $script:rsop } }
function Read-Host { param($Prompt) $script:prompts++; if ($script:onPrompt) { & $script:onPrompt }; return $script:response }

$script:assertions = 0
foreach ($initial in @($null, 0, 1)) {
    Reset-Policy $initial
    Set-WelaOutgoingNtlmPolicy -Auto
    Assert-Equal $script:value 1 "Default audits initial value '$initial'"
    $expectedWrites = if ($initial -eq 1) { 0 } else { 1 }
    Assert-Equal $script:writes $expectedWrites 'Already audited hosts are idempotent'
}
Reset-Policy $null
$script:keyExists = $false
Set-WelaOutgoingNtlmPolicy -Auto
Assert-Equal $script:keyExists $true 'Missing key is created'
Assert-Equal $script:value 1 'Missing key gets audit mode'
Reset-Policy 2
Set-WelaOutgoingNtlmPolicy -Auto
Assert-Equal $script:value 2 'Auto preserves intentional deny'
Assert-Equal $script:writes 0 'Auto does not rewrite deny'
Assert-Equal ((Get-WelaOutgoingNtlmState).Description -like 'Deny all*authentication restriction*') $true 'Deny is reported as enforcement'
Set-WelaOutgoingNtlmPolicy -Mode Audit -Auto
Assert-Equal $script:value 1 'Explicit Audit may replace deny'
Set-WelaOutgoingNtlmPolicy -Mode Deny -Auto
Assert-Equal $script:value 2 'Only explicit Deny opts into enforcement'
Reset-Policy 42
Set-WelaOutgoingNtlmPolicy -Auto
Assert-Equal $script:writes 0 'Unknown value is preserved'
Assert-Equal ((Get-WelaOutgoingNtlmState).Description) 'Unknown registry value (42)' 'Unknown values are reported honestly'
Reset-Policy 0
$script:readFails = $true
Assert-Throws { Set-WelaOutgoingNtlmPolicy -Auto } 'Unreadable policy fails visibly'
Assert-Equal $script:writes 0 'Unreadable policy is never overwritten'
Reset-Policy 0
Set-WelaOutgoingNtlmPolicy -WhatIf
Assert-Equal $script:writes 0 'WhatIf does not mutate policy'
Assert-Equal $script:prompts 0 'WhatIf does not prompt without Auto'
Reset-Policy 2
$script:response = 'n'
Set-WelaOutgoingNtlmPolicy -Mode Audit
Assert-Equal $script:writes 0 'Declining preserves deny'
$script:response = ''
Set-WelaOutgoingNtlmPolicy -Mode Audit
Assert-Equal $script:value 1 'Confirmed explicit override succeeds'
# A user confirmation must not authorize replacing enforcement that appeared during the prompt.
foreach ($changed in @(2, 42)) {
    Reset-Policy 0
    $script:changedDuringPrompt = $changed
    $script:onPrompt = { $script:value = $script:changedDuringPrompt }
    Set-WelaOutgoingNtlmPolicy
    Assert-Equal $script:prompts 1 'State changes while the user is confirming'
    Assert-Equal $script:value $changed 'Default mode preserves newly applied deny or unknown policy'
    Assert-Equal $script:writes 0 'A stale initial read never authorizes replacing new enforcement'
}
Reset-Policy 0
$script:onPrompt = { $script:readFails = $true }
Assert-Throws { Set-WelaOutgoingNtlmPolicy } 'State becoming unreadable during confirmation aborts'
Assert-Equal $script:writes 0 'Unreadable refreshed state is not overwritten'
Reset-Policy 0
$script:onPrompt = { $script:value = 2 }
Set-WelaOutgoingNtlmPolicy -Mode Audit
Assert-Equal $script:value 1 'Explicit Audit still authorizes replacing deny introduced during confirmation'
Assert-Equal $script:writes 1 'Explicit override writes once after a fresh readable state'
Reset-Policy 0
$script:onPrompt = { $script:value = 1 }
Set-WelaOutgoingNtlmPolicy
Assert-Equal $script:writes 0 'A concurrently applied audit setting is not redundantly rewritten'
Reset-Policy 0
$script:onRead = { if ($script:reads -eq 2) { $script:value = 2 } }
Set-WelaOutgoingNtlmPolicy -Auto
Assert-Equal $script:value 2 'Auto also preserves deny introduced after the initial read'
Assert-Equal $script:writes 0 'Auto rechecks immediately before writing'
Reset-Policy 0
$script:writeFails = $true
Assert-Throws { Set-WelaOutgoingNtlmPolicy -Auto } 'Write failure propagates'
Reset-Policy 0
$script:ignoreWrite = $true
Assert-Throws { Set-WelaOutgoingNtlmPolicy -Auto } 'Read-back mismatch propagates'
Reset-Policy 0
Assert-Equal ((Get-WelaOutgoingNtlmState).PolicySource -like 'Unknown*') $true 'No RSoP does not imply local provenance'
$script:rsop = @(
    [pscustomobject]@{ keyName = 'SYSTEM\CurrentControlSet\Control\Lsa\MSV1_0'; valueName = 'RestrictSendingNTLMTraffic'; precedence = 2; GPOID = 'Lower priority GPO' },
    [pscustomobject]@{ keyName = 'HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Control\Lsa\MSV1_0'; valueName = 'RestrictSendingNTLMTraffic'; precedence = 1; GPOID = 'Winning GPO' },
    [pscustomobject]@{ keyName = 'SYSTEM\Other'; valueName = 'RestrictSendingNTLMTraffic'; precedence = 0; GPOID = 'Unrelated GPO' }
)
$source = (Get-WelaOutgoingNtlmState).PolicySource
Assert-Equal ($source -like 'Last-applied RSoP GPO: Winning GPO*may be stale*') $true 'Matching RSoP priority and freshness limits are reported'
foreach ($kind in @('String', 'QWord')) {
    foreach ($initial in @('0', '1', '2')) {
        Reset-Policy $initial $kind
        $state = Get-WelaOutgoingNtlmState
        Assert-Equal $state.Type $kind 'Outgoing state retains registry kind'
        Assert-Equal ($state.Description -like 'Unknown registry type*expected DWord*') $true 'A non-DWORD mode is reported as unknown'
        Set-WelaOutgoingNtlmPolicy -Auto
        Assert-Equal $script:writes 0 'Default mode preserves malformed outgoing types'
        Assert-Equal $script:type $kind 'Default mode preserves the original registry type'
    }
    foreach ($mode in @('Audit', 'Deny')) {
        $desired = if ($mode -eq 'Audit') { 1 } else { 2 }
        Reset-Policy ([string]$desired) $kind
        Set-WelaOutgoingNtlmPolicy -Mode $mode -Auto
        Assert-Equal $script:writes 1 'Explicit mode repairs a matching value with the wrong type'
        Assert-Equal $script:type 'DWord' 'Explicit mode writes DWORD'
        Assert-Equal $script:value $desired 'Explicit repair preserves the requested policy value'
    }
}
Reset-Policy '1' 'String'
$script:ignoreTypeWrite = $true
Assert-Throws { Set-WelaOutgoingNtlmPolicy -Mode Audit -Auto } 'Outgoing read-back rejects the right value with the wrong type'
Assert-Equal $script:writes 1 'Outgoing read-back type failure occurs after an attempted repair'
Reset-Policy 1
$script:typeReadFails = $true
Assert-Equal ((Get-WelaOutgoingNtlmState).Readable) $false 'A registry kind read failure is not a readable outgoing state'
Assert-Throws { Set-WelaOutgoingNtlmPolicy -Mode Audit -Auto } 'Explicit mode fails closed when registry kind cannot be read'
Assert-Equal $script:writes 0 'Unknown outgoing registry kind is never overwritten'
Reset-Policy 0
$script:onPrompt = { $script:value = '1'; $script:type = 'String' }
Set-WelaOutgoingNtlmPolicy
Assert-Equal $script:writes 0 'Default mode preserves malformed types introduced during confirmation'
Assert-Equal $script:type 'String' 'The final pre-write check retains a newly introduced malformed type'
Reset-Policy 0
$script:onPrompt = { $script:value = '1'; $script:type = 'String' }
Set-WelaOutgoingNtlmPolicy -Mode Audit
Assert-Equal $script:writes 1 'Explicit mode repairs a malformed type introduced during confirmation'
Assert-Equal $script:type 'DWord' 'Explicit pre-write decision checks the registry type'
Write-Host "PASS: $script:assertions outgoing NTLM assertions (mocked; no host changes)."