# nl-video-api 系统测试脚本
# 执行完整的功能测试、性能测试和错误处理测试

param(
    [string]$BaseUrl = "http://localhost:8000",
    [string]$TestMode = "all"  # all, function, performance, error
)

# 测试结果统计
$TestResults = @{
    Total = 0
    Passed = 0
    Failed = 0
    Errors = @()
}

# 颜色输出函数
function Write-ColorOutput {
    param(
        [string]$Message,
        [string]$Color = "White"
    )
    
    switch ($Color) {
        "Green" { Write-Host $Message -ForegroundColor Green }
        "Red" { Write-Host $Message -ForegroundColor Red }
        "Yellow" { Write-Host $Message -ForegroundColor Yellow }
        "Blue" { Write-Host $Message -ForegroundColor Blue }
        "Cyan" { Write-Host $Message -ForegroundColor Cyan }
        default { Write-Host $Message }
    }
}

# HTTP请求函数
function Invoke-ApiRequest {
    param(
        [string]$Method,
        [string]$Url,
        [hashtable]$Headers = @{},
        [string]$Body = $null,
        [string]$TestName
    )
    
    $TestResults.Total++
    
    try {
        $params = @{
            Uri = $Url
            Method = $Method
            Headers = $Headers
            ContentType = "application/json"
        }
        
        if ($Body) {
            $params.Body = $Body
        }
        
        $startTime = Get-Date
        $response = Invoke-RestMethod @params
        $endTime = Get-Date
        $duration = ($endTime - $startTime).TotalMilliseconds
        
        if ($response.code -eq 0) {
            Write-ColorOutput "✓ $TestName - 通过 (${duration}ms)" "Green"
            $TestResults.Passed++
            return @{ Success = $true; Response = $response; Duration = $duration }
        } else {
            Write-ColorOutput "✗ $TestName - 失败: $($response.message)" "Red"
            $TestResults.Failed++
            $TestResults.Errors += "$TestName - $($response.message)"
            return @{ Success = $false; Response = $response; Duration = $duration }
        }
    }
    catch {
        Write-ColorOutput "✗ $TestName - 错误: $($_.Exception.Message)" "Red"
        $TestResults.Failed++
        $TestResults.Errors += "$TestName - $($_.Exception.Message)"
        return @{ Success = $false; Error = $_.Exception.Message; Duration = 0 }
    }
}

# 获取管理员Token
function Get-AdminToken {
    Write-ColorOutput "`n=== 获取管理员Token ===" "Blue"
    
    $loginData = @{
        username = "admin"
        password = "123456"
    } | ConvertTo-Json
    
    $result = Invoke-ApiRequest -Method "POST" -Url "$BaseUrl/api/v1/auth/admin/login" -Body $loginData -TestName "管理员登录"
    
    if ($result.Success) {
        return $result.Response.data.token
    }
    return $null
}

# 功能测试
function Test-Functions {
    Write-ColorOutput "`n=== 开始功能测试 ===" "Cyan"
    
    # 获取管理员Token
    $adminToken = Get-AdminToken
    if (-not $adminToken) {
        Write-ColorOutput "无法获取管理员Token，跳过需要认证的测试" "Yellow"
        return
    }
    
    $authHeaders = @{ "Authorization" = "Bearer $adminToken" }
    
    # 测试影片管理
    Write-ColorOutput "`n--- 影片管理测试 ---" "Blue"
    
    # 获取影片列表
    Invoke-ApiRequest -Method "GET" -Url "$BaseUrl/api/v1/movies?page=1&page_size=10" -TestName "获取影片列表"
    
    # 创建影片
    $movieData = @{
        title = "测试电影_$(Get-Date -Format 'yyyyMMddHHmmss')"
        type = 1
        category_id = 1
        year = 2024
        country = "中国"
        director = "测试导演"
        actors = "测试演员"
        description = "这是一部测试电影"
        duration = 120
        status = 1
    } | ConvertTo-Json
    
    $createResult = Invoke-ApiRequest -Method "POST" -Url "$BaseUrl/api/v1/admin/movies" -Headers $authHeaders -Body $movieData -TestName "创建影片"
    
    if ($createResult.Success) {
        $movieId = $createResult.Response.data.id
        
        # 获取影片详情
        Invoke-ApiRequest -Method "GET" -Url "$BaseUrl/api/v1/movies/$movieId" -TestName "获取影片详情"
        
        # 更新影片
        $updateData = @{
            title = "更新后的测试电影"
            description = "更新后的描述"
        } | ConvertTo-Json
        
        Invoke-ApiRequest -Method "PUT" -Url "$BaseUrl/api/v1/admin/movies/$movieId" -Headers $authHeaders -Body $updateData -TestName "更新影片"
        
        # 删除影片
        Invoke-ApiRequest -Method "DELETE" -Url "$BaseUrl/api/v1/admin/movies/$movieId" -Headers $authHeaders -TestName "删除影片"
    }
    
    # 测试用户管理
    Write-ColorOutput "`n--- 用户管理测试 ---" "Blue"
    
    # 获取用户列表
    Invoke-ApiRequest -Method "GET" -Url "$BaseUrl/api/v1/admin/users?page=1&page_size=10" -Headers $authHeaders -TestName "获取用户列表"
    
    # 创建用户
    $userData = @{
        username = "testuser_$(Get-Date -Format 'yyyyMMddHHmmss')"
        phone = "138$(Get-Random -Minimum 10000000 -Maximum 99999999)"
        email = "test$(Get-Date -Format 'yyyyMMddHHmmss')@example.com"
        password = "123456"
        nickname = "测试用户"
        gender = 1
        status = 1
    } | ConvertTo-Json
    
    $userResult = Invoke-ApiRequest -Method "POST" -Url "$BaseUrl/api/v1/admin/users" -Headers $authHeaders -Body $userData -TestName "创建用户"
    
    if ($userResult.Success) {
        $userId = $userResult.Response.data.id
        
        # 获取用户详情
        Invoke-ApiRequest -Method "GET" -Url "$BaseUrl/api/v1/admin/users/$userId" -Headers $authHeaders -TestName "获取用户详情"
        
        # 更新用户VIP
        $vipData = @{
            vip_level = 2
            days = 30
        } | ConvertTo-Json
        
        Invoke-ApiRequest -Method "POST" -Url "$BaseUrl/api/v1/admin/users/$userId/vip" -Headers $authHeaders -Body $vipData -TestName "升级用户VIP"
        
        # 更新用户余额
        $balanceData = @{
            amount = 100.50
            type = 1
            remark = "测试充值"
        } | ConvertTo-Json
        
        Invoke-ApiRequest -Method "PUT" -Url "$BaseUrl/api/v1/admin/users/$userId/balance" -Headers $authHeaders -Body $balanceData -TestName "更新用户余额"
    }
    
    # 测试权限管理
    Write-ColorOutput "`n--- 权限管理测试 ---" "Blue"
    
    # 获取角色列表
    Invoke-ApiRequest -Method "GET" -Url "$BaseUrl/api/v1/admin/roles" -Headers $authHeaders -TestName "获取角色列表"
    
    # 获取权限列表
    Invoke-ApiRequest -Method "GET" -Url "$BaseUrl/api/v1/admin/permissions" -Headers $authHeaders -TestName "获取权限列表"
    
    # 获取权限树
    Invoke-ApiRequest -Method "GET" -Url "$BaseUrl/api/v1/admin/permissions/tree" -Headers $authHeaders -TestName "获取权限树"
}

# 性能测试
function Test-Performance {
    Write-ColorOutput "`n=== 开始性能测试 ===" "Cyan"
    
    $concurrentRequests = 10
    $testDuration = 30 # 秒
    $requestCount = 0
    $successCount = 0
    $failureCount = 0
    $responseTimes = @()
    
    Write-ColorOutput "并发请求数: $concurrentRequests" "Yellow"
    Write-ColorOutput "测试持续时间: $testDuration 秒" "Yellow"
    
    $startTime = Get-Date
    $endTime = $startTime.AddSeconds($testDuration)
    
    # 并发测试
    $jobs = @()
    for ($i = 1; $i -le $concurrentRequests; $i++) {
        $job = Start-Job -ScriptBlock {
            param($BaseUrl, $EndTime)
            
            $results = @{
                RequestCount = 0
                SuccessCount = 0
                FailureCount = 0
                ResponseTimes = @()
            }
            
            while ((Get-Date) -lt $EndTime) {
                try {
                    $startRequest = Get-Date
                    $response = Invoke-RestMethod -Uri "$BaseUrl/api/v1/movies?page=1&page_size=10" -Method GET
                    $endRequest = Get-Date
                    $duration = ($endRequest - $startRequest).TotalMilliseconds
                    
                    $results.RequestCount++
                    $results.ResponseTimes += $duration
                    
                    if ($response.code -eq 0) {
                        $results.SuccessCount++
                    } else {
                        $results.FailureCount++
                    }
                }
                catch {
                    $results.RequestCount++
                    $results.FailureCount++
                }
                
                Start-Sleep -Milliseconds 100
            }
            
            return $results
        } -ArgumentList $BaseUrl, $endTime
        
        $jobs += $job
    }
    
    # 等待所有任务完成
    Write-ColorOutput "等待性能测试完成..." "Yellow"
    $allResults = $jobs | Wait-Job | Receive-Job
    $jobs | Remove-Job
    
    # 汇总结果
    $totalRequests = ($allResults | Measure-Object -Property RequestCount -Sum).Sum
    $totalSuccess = ($allResults | Measure-Object -Property SuccessCount -Sum).Sum
    $totalFailure = ($allResults | Measure-Object -Property FailureCount -Sum).Sum
    $allResponseTimes = $allResults | ForEach-Object { $_.ResponseTimes } | Where-Object { $_ -ne $null }
    
    if ($allResponseTimes.Count -gt 0) {
        $avgResponseTime = ($allResponseTimes | Measure-Object -Average).Average
        $minResponseTime = ($allResponseTimes | Measure-Object -Minimum).Minimum
        $maxResponseTime = ($allResponseTimes | Measure-Object -Maximum).Maximum
        $qps = [math]::Round($totalRequests / $testDuration, 2)
        $successRate = [math]::Round(($totalSuccess / $totalRequests) * 100, 2)
        
        Write-ColorOutput "`n--- 性能测试结果 ---" "Blue"
        Write-ColorOutput "总请求数: $totalRequests" "White"
        Write-ColorOutput "成功请求: $totalSuccess" "Green"
        Write-ColorOutput "失败请求: $totalFailure" "Red"
        Write-ColorOutput "成功率: $successRate%" "White"
        Write-ColorOutput "QPS: $qps" "White"
        Write-ColorOutput "平均响应时间: $([math]::Round($avgResponseTime, 2))ms" "White"
        Write-ColorOutput "最小响应时间: $([math]::Round($minResponseTime, 2))ms" "White"
        Write-ColorOutput "最大响应时间: $([math]::Round($maxResponseTime, 2))ms" "White"
        
        # 性能评估
        if ($avgResponseTime -lt 200) {
            Write-ColorOutput "性能评估: 优秀" "Green"
        } elseif ($avgResponseTime -lt 500) {
            Write-ColorOutput "性能评估: 良好" "Yellow"
        } else {
            Write-ColorOutput "性能评估: 需要优化" "Red"
        }
    }
}

# 错误处理测试
function Test-ErrorHandling {
    Write-ColorOutput "`n=== 开始错误处理测试 ===" "Cyan"
    
    # 测试无效的API端点
    Write-ColorOutput "`n--- 无效端点测试 ---" "Blue"
    try {
        Invoke-RestMethod -Uri "$BaseUrl/api/v1/invalid-endpoint" -Method GET
        Write-ColorOutput "✗ 无效端点测试 - 应该返回404错误" "Red"
    }
    catch {
        if ($_.Exception.Response.StatusCode -eq 404) {
            Write-ColorOutput "✓ 无效端点测试 - 正确返回404错误" "Green"
        } else {
            Write-ColorOutput "✗ 无效端点测试 - 返回了意外的错误: $($_.Exception.Message)" "Red"
        }
    }
    
    # 测试无效的请求参数
    Write-ColorOutput "`n--- 无效参数测试 ---" "Blue"
    $invalidData = @{
        invalid_field = "invalid_value"
    } | ConvertTo-Json
    
    try {
        $response = Invoke-RestMethod -Uri "$BaseUrl/api/v1/auth/admin/login" -Method POST -Body $invalidData -ContentType "application/json"
        if ($response.code -ne 0) {
            Write-ColorOutput "✓ 无效参数测试 - 正确返回参数错误" "Green"
        } else {
            Write-ColorOutput "✗ 无效参数测试 - 应该返回参数错误" "Red"
        }
    }
    catch {
        Write-ColorOutput "✓ 无效参数测试 - 正确处理了无效参数" "Green"
    }
    
    # 测试未授权访问
    Write-ColorOutput "`n--- 未授权访问测试 ---" "Blue"
    try {
        $response = Invoke-RestMethod -Uri "$BaseUrl/api/v1/admin/users" -Method GET
        Write-ColorOutput "✗ 未授权访问测试 - 应该返回认证错误" "Red"
    }
    catch {
        if ($_.Exception.Response.StatusCode -eq 401) {
            Write-ColorOutput "✓ 未授权访问测试 - 正确返回401错误" "Green"
        } else {
            Write-ColorOutput "✗ 未授权访问测试 - 返回了意外的错误: $($_.Exception.Message)" "Red"
        }
    }
    
    # 测试SQL注入防护
    Write-ColorOutput "`n--- SQL注入防护测试 ---" "Blue"
    $sqlInjectionData = @{
        username = "admin'; DROP TABLE users; --"
        password = "123456"
    } | ConvertTo-Json
    
    try {
        $response = Invoke-RestMethod -Uri "$BaseUrl/api/v1/auth/admin/login" -Method POST -Body $sqlInjectionData -ContentType "application/json"
        if ($response.code -ne 0) {
            Write-ColorOutput "✓ SQL注入防护测试 - 正确阻止了SQL注入攻击" "Green"
        } else {
            Write-ColorOutput "✗ SQL注入防护测试 - 可能存在SQL注入漏洞" "Red"
        }
    }
    catch {
        Write-ColorOutput "✓ SQL注入防护测试 - 正确处理了恶意输入" "Green"
    }
}

# 生成测试报告
function Generate-TestReport {
    Write-ColorOutput "`n=== 测试报告 ===" "Cyan"
    Write-ColorOutput "测试时间: $(Get-Date -Format 'yyyy-MM-dd HH:mm:ss')" "White"
    Write-ColorOutput "总测试数: $($TestResults.Total)" "White"
    Write-ColorOutput "通过测试: $($TestResults.Passed)" "Green"
    Write-ColorOutput "失败测试: $($TestResults.Failed)" "Red"
    
    if ($TestResults.Total -gt 0) {
        $passRate = [math]::Round(($TestResults.Passed / $TestResults.Total) * 100, 2)
        Write-ColorOutput "通过率: $passRate%" "White"
        
        if ($passRate -ge 90) {
            Write-ColorOutput "测试评估: 优秀" "Green"
        } elseif ($passRate -ge 80) {
            Write-ColorOutput "测试评估: 良好" "Yellow"
        } else {
            Write-ColorOutput "测试评估: 需要改进" "Red"
        }
    }
    
    if ($TestResults.Errors.Count -gt 0) {
        Write-ColorOutput "`n失败的测试:" "Red"
        foreach ($error in $TestResults.Errors) {
            Write-ColorOutput "  - $error" "Red"
        }
    }
    
    # 保存测试报告到文件
    $reportPath = "test-report-$(Get-Date -Format 'yyyyMMdd-HHmmss').txt"
    $reportContent = @"
nl-video-api 系统测试报告
========================

测试时间: $(Get-Date -Format 'yyyy-MM-dd HH:mm:ss')
测试模式: $TestMode
基础URL: $BaseUrl

测试结果统计:
- 总测试数: $($TestResults.Total)
- 通过测试: $($TestResults.Passed)
- 失败测试: $($TestResults.Failed)
- 通过率: $([math]::Round(($TestResults.Passed / $TestResults.Total) * 100, 2))%

失败的测试:
$($TestResults.Errors -join "`n")

测试建议:
1. 定期执行系统测试以确保功能稳定性
2. 关注性能指标，及时优化慢查询
3. 加强错误处理和异常情况的测试覆盖
4. 建议集成到CI/CD流程中自动执行
"@
    
    $reportContent | Out-File -FilePath $reportPath -Encoding UTF8
    Write-ColorOutput "`n测试报告已保存到: $reportPath" "Blue"
}

# 主函数
function Main {
    Write-ColorOutput "nl-video-api 系统测试工具" "Cyan"
    Write-ColorOutput "========================" "Cyan"
    Write-ColorOutput "基础URL: $BaseUrl" "White"
    Write-ColorOutput "测试模式: $TestMode" "White"
    Write-ColorOutput "开始时间: $(Get-Date -Format 'yyyy-MM-dd HH:mm:ss')" "White"
    
    # 检查服务器是否可访问
    try {
        $response = Invoke-RestMethod -Uri "$BaseUrl/api/v1/movies?page=1&page_size=1" -Method GET -TimeoutSec 10
        Write-ColorOutput "✓ 服务器连接正常" "Green"
    }
    catch {
        Write-ColorOutput "✗ 无法连接到服务器: $BaseUrl" "Red"
        Write-ColorOutput "请确保服务器正在运行并且URL正确" "Yellow"
        return
    }
    
    # 根据测试模式执行相应测试
    switch ($TestMode.ToLower()) {
        "function" { Test-Functions }
        "performance" { Test-Performance }
        "error" { Test-ErrorHandling }
        "all" {
            Test-Functions
            Test-Performance
            Test-ErrorHandling
        }
        default {
            Write-ColorOutput "无效的测试模式: $TestMode" "Red"
            Write-ColorOutput "支持的模式: all, function, performance, error" "Yellow"
            return
        }
    }
    
    # 生成测试报告
    Generate-TestReport
    
    Write-ColorOutput "`n测试完成!" "Green"
}

# 执行主函数
Main