1
This commit is contained in:
@@ -0,0 +1,171 @@
|
||||
param(
|
||||
[string]$ApiBaseUrl = "http://192.168.200.61:18121/api/v1",
|
||||
[string]$ClusterID = "cfc0743d-d960-49fb-9de8-96e063d5e4aa",
|
||||
[string]$ActorUserID = "f67d943f-5397-4b3a-a229-695fe67ad700",
|
||||
[string]$EntryNodeName = "test-1",
|
||||
[string]$ExitNodeName = "test-2",
|
||||
[string]$EntryBaseUrl = "http://192.168.200.61:19131",
|
||||
[string]$ResultPath = "artifacts\c19z2-remote-workspace-mailbox-preflight-telemetry-smoke-result.json"
|
||||
)
|
||||
|
||||
Set-StrictMode -Version Latest
|
||||
$ErrorActionPreference = "Stop"
|
||||
|
||||
$repoRoot = (Resolve-Path (Join-Path $PSScriptRoot "..\..")).ProviderPath
|
||||
$sourceResultPath = "artifacts\c19z2-remote-workspace-mailbox-preflight-source-result.json"
|
||||
$consumerID = "rdp-worker-probe"
|
||||
|
||||
function Invoke-Api {
|
||||
param([string]$Method, [string]$Path, [object]$Body = $null)
|
||||
$uri = "$ApiBaseUrl$Path"
|
||||
if ($null -eq $Body) { return Invoke-RestMethod -Method $Method -Uri $uri -TimeoutSec 30 }
|
||||
return Invoke-RestMethod -Method $Method -Uri $uri -ContentType "application/json" -Body ($Body | ConvertTo-Json -Depth 80) -TimeoutSec 30
|
||||
}
|
||||
|
||||
function Get-PropertyValue {
|
||||
param([object]$Item, [string]$Name, [object]$Default = $null)
|
||||
if ($null -eq $Item) { return $Default }
|
||||
$property = $Item.PSObject.Properties[$Name]
|
||||
if ($null -eq $property) { return $Default }
|
||||
return $property.Value
|
||||
}
|
||||
|
||||
function Get-NodeByName {
|
||||
param([string]$Name)
|
||||
$nodes = (Invoke-Api -Method GET -Path "/clusters/$ClusterID/nodes?actor_user_id=$ActorUserID").nodes
|
||||
$node = @($nodes | Where-Object { $_.name -eq $Name }) | Select-Object -First 1
|
||||
if ($null -eq $node) { throw "Node '$Name' was not found in cluster $ClusterID" }
|
||||
return $node
|
||||
}
|
||||
|
||||
function Get-RemoteWorkspaceSinkReports {
|
||||
param([string]$NodeID)
|
||||
$workloadStatus = $null
|
||||
$workloadSink = $null
|
||||
$telemetry = $null
|
||||
$telemetrySink = $null
|
||||
$statuses = (Invoke-Api -Method GET -Path "/clusters/$ClusterID/nodes/$NodeID/workloads/status?actor_user_id=$ActorUserID").workload_statuses
|
||||
$workloadStatus = @($statuses | Where-Object { $_.service_type -eq "rdp-worker" } | Select-Object -First 1)
|
||||
if ($null -ne $workloadStatus) {
|
||||
$workloadSink = Get-PropertyValue -Item $workloadStatus.status_payload -Name "remote_workspace_adapter_sink" -Default $null
|
||||
}
|
||||
$telemetryItems = @((Invoke-Api -Method GET -Path "/clusters/$ClusterID/nodes/$NodeID/telemetry?actor_user_id=$ActorUserID&limit=10").telemetry)
|
||||
$telemetry = $telemetryItems | Select-Object -First 1
|
||||
if ($null -ne $telemetry) {
|
||||
$telemetryPayload = Get-PropertyValue -Item $telemetry -Name "payload" -Default $null
|
||||
$telemetrySink = Get-PropertyValue -Item $telemetryPayload -Name "remote_workspace_adapter_sink_report" -Default $null
|
||||
}
|
||||
return [ordered]@{
|
||||
workload_status = $workloadStatus
|
||||
workload_sink = $workloadSink
|
||||
telemetry = $telemetry
|
||||
telemetry_sink = $telemetrySink
|
||||
}
|
||||
}
|
||||
|
||||
function Test-PreflightTelemetry {
|
||||
param(
|
||||
[object]$Sink,
|
||||
[object]$BaselineSink,
|
||||
[string]$AdapterSessionID,
|
||||
[int64]$ResumeSequence,
|
||||
[int]$AvailableCount,
|
||||
[int]$ReturnedCount,
|
||||
[int]$SkippedCount
|
||||
)
|
||||
if ($null -eq $Sink) { return $false }
|
||||
$baselineTotal = [int64](Get-PropertyValue -Item $BaselineSink -Name "mailbox_preflight_total" -Default 0)
|
||||
$baselineAck = [int64](Get-PropertyValue -Item $BaselineSink -Name "mailbox_preflight_ack_total" -Default 0)
|
||||
$baselineCheckpoint = [int64](Get-PropertyValue -Item $BaselineSink -Name "mailbox_preflight_checkpoint_total" -Default 0)
|
||||
$readiness = Get-PropertyValue -Item $Sink -Name "adapter_runtime_readiness" -Default $null
|
||||
return (
|
||||
[string](Get-PropertyValue -Item $Sink -Name "schema_version" -Default "") -eq "rap.remote_workspace_adapter_sink_report.v1" -and
|
||||
[int64](Get-PropertyValue -Item $Sink -Name "mailbox_preflight_total" -Default 0) -ge ($baselineTotal + 2) -and
|
||||
[int64](Get-PropertyValue -Item $Sink -Name "mailbox_preflight_ack_total" -Default 0) -ge ($baselineAck + 1) -and
|
||||
[int64](Get-PropertyValue -Item $Sink -Name "mailbox_preflight_checkpoint_total" -Default 0) -ge ($baselineCheckpoint + 1) -and
|
||||
[string](Get-PropertyValue -Item $Sink -Name "last_mailbox_preflight_adapter_session_id" -Default "") -eq $AdapterSessionID -and
|
||||
[string](Get-PropertyValue -Item $Sink -Name "last_mailbox_preflight_consumer_id" -Default "") -eq $consumerID -and
|
||||
[string](Get-PropertyValue -Item $Sink -Name "last_mailbox_preflight_resume_from" -Default "") -eq "checkpoint" -and
|
||||
[int64](Get-PropertyValue -Item $Sink -Name "last_mailbox_preflight_resume_sequence" -Default -1) -eq $ResumeSequence -and
|
||||
[int](Get-PropertyValue -Item $Sink -Name "last_mailbox_preflight_available_count" -Default -1) -eq $AvailableCount -and
|
||||
[int](Get-PropertyValue -Item $Sink -Name "last_mailbox_preflight_returned_count" -Default -1) -eq $ReturnedCount -and
|
||||
[int](Get-PropertyValue -Item $Sink -Name "last_mailbox_preflight_skipped_count" -Default -1) -eq $SkippedCount -and
|
||||
$null -ne $readiness -and
|
||||
[int64](Get-PropertyValue -Item $readiness -Name "mailbox_preflight_total" -Default 0) -ge 2
|
||||
)
|
||||
}
|
||||
|
||||
$entryNode = Get-NodeByName -Name $EntryNodeName
|
||||
$baseline = Get-RemoteWorkspaceSinkReports -NodeID $entryNode.id
|
||||
$baselineWorkloadSink = $baseline.workload_sink
|
||||
$baselineTelemetrySink = $baseline.telemetry_sink
|
||||
|
||||
$source = & powershell -ExecutionPolicy Bypass -File (Join-Path $PSScriptRoot "c19z1-remote-workspace-mailbox-preflight-smoke.ps1") `
|
||||
-ApiBaseUrl $ApiBaseUrl `
|
||||
-ClusterID $ClusterID `
|
||||
-ActorUserID $ActorUserID `
|
||||
-EntryNodeName $EntryNodeName `
|
||||
-ExitNodeName $ExitNodeName `
|
||||
-EntryBaseUrl $EntryBaseUrl `
|
||||
-ResultPath $sourceResultPath
|
||||
|
||||
$sourceFile = Join-Path $repoRoot $sourceResultPath
|
||||
$sourceResult = Get-Content -Raw -Path $sourceFile | ConvertFrom-Json
|
||||
$adapterSessionID = [string](Get-PropertyValue -Item $sourceResult -Name "adapter_session_id" -Default "")
|
||||
$checkpoint = Get-PropertyValue -Item $sourceResult -Name "preflight_checkpoint" -Default $null
|
||||
$checkpointJson = Get-PropertyValue -Item $checkpoint -Name "json" -Default $null
|
||||
$resumeSequence = [int64](Get-PropertyValue -Item $checkpointJson -Name "resume_sequence" -Default 0)
|
||||
$availableCount = [int](Get-PropertyValue -Item $checkpointJson -Name "expected_available_count" -Default -1)
|
||||
$returnedCount = [int](Get-PropertyValue -Item $checkpointJson -Name "expected_returned_count" -Default -1)
|
||||
$skippedCount = [int](Get-PropertyValue -Item $checkpointJson -Name "expected_skipped_count" -Default -1)
|
||||
|
||||
$observed = $null
|
||||
$deadline = (Get-Date).AddSeconds(90)
|
||||
while ((Get-Date) -lt $deadline) {
|
||||
Start-Sleep -Seconds 5
|
||||
$observed = Get-RemoteWorkspaceSinkReports -NodeID $entryNode.id
|
||||
if (
|
||||
(Test-PreflightTelemetry -Sink $observed.workload_sink -BaselineSink $baselineWorkloadSink -AdapterSessionID $adapterSessionID -ResumeSequence $resumeSequence -AvailableCount $availableCount -ReturnedCount $returnedCount -SkippedCount $skippedCount) -and
|
||||
(Test-PreflightTelemetry -Sink $observed.telemetry_sink -BaselineSink $baselineTelemetrySink -AdapterSessionID $adapterSessionID -ResumeSequence $resumeSequence -AvailableCount $availableCount -ReturnedCount $returnedCount -SkippedCount $skippedCount)
|
||||
) {
|
||||
break
|
||||
}
|
||||
}
|
||||
if ($null -eq $observed) {
|
||||
$observed = Get-RemoteWorkspaceSinkReports -NodeID $entryNode.id
|
||||
}
|
||||
|
||||
$checks = [ordered]@{
|
||||
source_smoke_passed = ([bool]$sourceResult.passed)
|
||||
adapter_session_id_present = ($adapterSessionID -match "^rap-rw-adapter-session-[0-9a-f]{24}$")
|
||||
checkpoint_preflight_shape_visible = ([string](Get-PropertyValue -Item $checkpointJson -Name "resume_from" -Default "") -eq "checkpoint" -and $resumeSequence -gt 0 -and $availableCount -eq 0 -and $returnedCount -eq 0 -and $skippedCount -eq 3)
|
||||
workload_preflight_telemetry_visible = (Test-PreflightTelemetry -Sink $observed.workload_sink -BaselineSink $baselineWorkloadSink -AdapterSessionID $adapterSessionID -ResumeSequence $resumeSequence -AvailableCount $availableCount -ReturnedCount $returnedCount -SkippedCount $skippedCount)
|
||||
telemetry_preflight_telemetry_visible = (Test-PreflightTelemetry -Sink $observed.telemetry_sink -BaselineSink $baselineTelemetrySink -AdapterSessionID $adapterSessionID -ResumeSequence $resumeSequence -AvailableCount $availableCount -ReturnedCount $returnedCount -SkippedCount $skippedCount)
|
||||
}
|
||||
$failed = @($checks.GetEnumerator() | Where-Object { -not $_.Value } | ForEach-Object { $_.Key })
|
||||
|
||||
$result = [ordered]@{
|
||||
schema_version = "c19z2.remote_workspace_mailbox_preflight_telemetry_smoke.v1"
|
||||
source_result_path = $sourceFile
|
||||
cluster_id = $ClusterID
|
||||
entry_node = [ordered]@{ id = $entryNode.id; name = $entryNode.name }
|
||||
adapter_session_id = $adapterSessionID
|
||||
baseline = $baseline
|
||||
source = $sourceResult
|
||||
observed = $observed
|
||||
checks = $checks
|
||||
failed_checks = $failed
|
||||
passed = ($failed.Count -eq 0)
|
||||
}
|
||||
|
||||
$fullResultPath = Join-Path $repoRoot $ResultPath
|
||||
$resultDir = Split-Path -Parent $fullResultPath
|
||||
if ($resultDir) { New-Item -ItemType Directory -Force -Path $resultDir | Out-Null }
|
||||
$result | ConvertTo-Json -Depth 100 | Set-Content -Encoding UTF8 -Path $fullResultPath
|
||||
|
||||
if (-not $result.passed) {
|
||||
throw "C19Z2 remote workspace mailbox preflight telemetry smoke failed. Result: $fullResultPath Failed: $($failed -join ', ')"
|
||||
}
|
||||
|
||||
Write-Host "C19Z2 remote workspace mailbox preflight telemetry smoke passed. Result: $fullResultPath"
|
||||
$result
|
||||
Reference in New Issue
Block a user