88 lines
2.6 KiB
PowerShell
88 lines
2.6 KiB
PowerShell
param(
|
|
[ValidateRange(1, 1000000)]
|
|
[int]$Rounds = 1000
|
|
)
|
|
|
|
$ErrorActionPreference = "Stop"
|
|
$utf8Encoding = New-Object System.Text.UTF8Encoding($false)
|
|
[Console]::OutputEncoding = $utf8Encoding
|
|
$OutputEncoding = $utf8Encoding
|
|
|
|
$mutex = New-Object System.Threading.Mutex($false, "Local\AiChat-Stability-Suite")
|
|
$ownsMutex = $false
|
|
try {
|
|
try {
|
|
$ownsMutex = $mutex.WaitOne(0)
|
|
} catch [System.Threading.AbandonedMutexException] {
|
|
$ownsMutex = $true
|
|
}
|
|
|
|
if (!$ownsMutex) {
|
|
Write-Host ">> Another stability run is already active; skipping this overlapping run." -ForegroundColor Yellow
|
|
exit 0
|
|
}
|
|
|
|
$root = Split-Path -Parent $PSScriptRoot
|
|
$logRoot = Join-Path $root "logs\stability"
|
|
if (!(Test-Path $logRoot)) {
|
|
New-Item -ItemType Directory -Path $logRoot -Force | Out-Null
|
|
}
|
|
|
|
$timestamp = Get-Date -Format "yyyyMMdd-HHmmss"
|
|
$logFile = Join-Path $logRoot "stability-$timestamp.log"
|
|
$summaryFile = Join-Path $logRoot "stability-$timestamp.summary.json"
|
|
|
|
function Invoke-StabilitySuite {
|
|
param(
|
|
[string]$SuiteName
|
|
)
|
|
|
|
$testScript = Join-Path $root "backend\tests\stability_1000_round.php"
|
|
Write-Host ">> Running $SuiteName suite ($Rounds rounds)" -ForegroundColor Cyan
|
|
Add-Content -Path $logFile -Value "===== $SuiteName =====" -Encoding UTF8
|
|
|
|
# Windows PowerShell wraps native stderr as ErrorRecord objects. With the
|
|
# script-wide Stop preference, a failing suite used to abort this runner
|
|
# before the exit code, remaining suites, and JSON summary were recorded.
|
|
$previousErrorActionPreference = $ErrorActionPreference
|
|
try {
|
|
$ErrorActionPreference = "Continue"
|
|
& php $testScript $Rounds $SuiteName 2>&1 | ForEach-Object {
|
|
$line = [string]$_
|
|
Write-Host $line
|
|
Add-Content -Path $logFile -Value $line -Encoding UTF8
|
|
}
|
|
$exitCode = $LASTEXITCODE
|
|
} finally {
|
|
$ErrorActionPreference = $previousErrorActionPreference
|
|
}
|
|
Add-Content -Path $logFile -Value "" -Encoding UTF8
|
|
|
|
return [PSCustomObject]@{
|
|
suite = $SuiteName
|
|
requestedRounds = $Rounds
|
|
exitCode = $exitCode
|
|
completedAt = (Get-Date).ToString("s")
|
|
}
|
|
}
|
|
|
|
$summary = @()
|
|
foreach ($suite in @("agent", "image", "llm")) {
|
|
$summary += Invoke-StabilitySuite -SuiteName $suite
|
|
}
|
|
|
|
$summary | ConvertTo-Json -Depth 4 | Set-Content -Path $summaryFile -Encoding UTF8
|
|
$failedSuites = @($summary | Where-Object { $_.exitCode -ne 0 })
|
|
if ($failedSuites.Count -gt 0) {
|
|
Write-Error "Stability run failed in $($failedSuites.Count) suite(s). Log: $logFile"
|
|
exit 1
|
|
}
|
|
|
|
Write-Host ">> Stability suites completed. Log: $logFile" -ForegroundColor Green
|
|
} finally {
|
|
if ($ownsMutex) {
|
|
$mutex.ReleaseMutex()
|
|
}
|
|
$mutex.Dispose()
|
|
}
|