PublicStudy/NetworkModuleTest/run_test_auto.ps1 at main · leedg/PublicStudy · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
# ==============================================================================
# run_test_auto.ps1
# 역할: DBServer + TestServer + TestClient 1개를 자동으로 기동하고,
#       지정한 시간 실행 후 Graceful Shutdown 시키는 단일 클라이언트 자동 테스트 스크립트.
#       run_test_2clients.ps1 의 단순화 버전으로, CI/CD 나 단독 검증에 적합하다.
#
# 실행 순서:
#   1. (선택) 서버 구조 동기화 검증 스크립트 실행
#      (ModuleTest/ServerStructureSync/validate_server_structure_sync.ps1)
#      → 실패 시 테스트 중단 (-SkipStructureSyncCheck 로 건너뛸 수 있음)
#   2. TestDBServer 기동 (포트 8002)
#   3. TestServer 기동 (포트 9000, DB: 127.0.0.1:8002)
#   4. TestClient 기동
#   5. $RunSeconds 초 대기
#   6. Named Event 로 Graceful Shutdown 신호 전송 (타임아웃 시 Kill 폴백)
#   7. 각 프로세스 출력 결과 콘솔에 표시
#
# 매개변수:
#   -RunSeconds            : 테스트 실행 시간(초), 기본값 5
#   -SkipStructureSyncCheck: 지정 시 구조 동기화 검증 단계를 건너뜀
#
# 빌드 경로:
#   스크립트 위치 기준 .\x64\Debug (이 스크립트는 NetworkModuleTest\ 루트에 위치)
# ==============================================================================
param(
    [int]$RunSeconds = 5,
    [switch]$SkipStructureSyncCheck,
    [ValidateSet('Debug','Release')]
    [string]$Config = 'Debug'
)

$binDir = Join-Path $PSScriptRoot "x64\$Config"
$dbOut  = "$env:TEMP\dbserver_out.txt"
$srvOut = "$env:TEMP\server_out.txt"
$cliOut = "$env:TEMP\client_out.txt"

if (-not $SkipStructureSyncCheck)
{
    $syncCheckScript = Join-Path $PSScriptRoot "ModuleTest\ServerStructureSync\validate_server_structure_sync.ps1"
    if (Test-Path $syncCheckScript)
    {
        Write-Host "=== Server Structure Sync Check ===" -ForegroundColor Cyan
        & $syncCheckScript
        if (-not $?)
        {
            throw "Server structure sync check failed. Fix docs/comments/tests before runtime test."
        }
    }
    else
    {
        Write-Host "[WARN] Sync check script not found: $syncCheckScript" -ForegroundColor Yellow
    }
}

# English: Helper — signal a Named Event for graceful shutdown, then wait with Kill fallback.
# 한글: Named Event로 정상 종료 신호 → 대기 → 타임아웃 시 Kill 폴백.
function Stop-Gracefully {
    param(
        [System.Diagnostics.Process]$Proc,
        [string]$EventName,
        [int]$TimeoutMs = 5000
    )
    if (-not $Proc -or $Proc.HasExited) { return }

    $sig = [IntPtr]::Zero
    try {
        Add-Type -TypeDefinition @"
using System;
using System.Runtime.InteropServices;
public class Win32Event {
    [DllImport("kernel32.dll", SetLastError=true)]
    public static extern IntPtr OpenEvent(uint dwDesiredAccess, bool bInheritHandle, string lpName);
    [DllImport("kernel32.dll", SetLastError=true)]
    public static extern bool SetEvent(IntPtr hEvent);
    [DllImport("kernel32.dll", SetLastError=true)]
    public static extern bool CloseHandle(IntPtr hObject);
}
"@ -ErrorAction SilentlyContinue

        $sig = [Win32Event]::OpenEvent(0x0002, $false, $EventName)
        if ($sig -ne [IntPtr]::Zero) {
            [Win32Event]::SetEvent($sig) | Out-Null
            $Proc.WaitForExit($TimeoutMs) | Out-Null
        }
    } catch {}
    finally {
        if ($sig -ne [IntPtr]::Zero) { [Win32Event]::CloseHandle($sig) | Out-Null }
    }

    if (-not $Proc.HasExited) { try { $Proc.Kill() } catch {} }
}

$dbProc = Start-Process -FilePath "$binDir\TestDBServer.exe" `
    -ArgumentList '-p','8002' -WorkingDirectory $binDir -PassThru `
    -RedirectStandardOutput $dbOut -RedirectStandardError "$env:TEMP\dbserver_err.txt"
Start-Sleep -Milliseconds 1000

$serverProc = Start-Process -FilePath "$binDir\TestServer.exe" `
    -ArgumentList '-p','9000','--db-host','127.0.0.1','--db-port','8002' `
    -WorkingDirectory $binDir -PassThru `
    -RedirectStandardOutput $srvOut -RedirectStandardError "$env:TEMP\server_err.txt"
Start-Sleep -Milliseconds 1000

$clientProc = Start-Process -FilePath "$binDir\TestClient.exe" `
    -ArgumentList '--host','127.0.0.1','--port','9000' `
    -WorkingDirectory $binDir -PassThru `
    -RedirectStandardOutput $cliOut -RedirectStandardError "$env:TEMP\client_err.txt"

Start-Sleep -Seconds $RunSeconds

# English: Graceful shutdown via Named Events (Kill fallback on timeout)
# 한글: Named Event로 정상 종료 (타임아웃 시 Kill 폴백)
Stop-Gracefully -Proc $clientProc -EventName "TestClient_GracefulShutdown_$($clientProc.Id)" -TimeoutMs 4000
Stop-Gracefully -Proc $serverProc -EventName "TestServer_GracefulShutdown"   -TimeoutMs 6000
Stop-Gracefully -Proc $dbProc     -EventName "TestDBServer_GracefulShutdown" -TimeoutMs 6000
Start-Sleep -Milliseconds 500

Write-Host "=== DBServer Output ===" -ForegroundColor Cyan
Get-Content $dbOut  -ErrorAction SilentlyContinue
Write-Host "=== Server Output ===" -ForegroundColor Cyan
Get-Content $srvOut -ErrorAction SilentlyContinue
Write-Host "=== Client Output ===" -ForegroundColor Cyan
Get-Content $cliOut -ErrorAction SilentlyContinue