65324a16dc
Introduce proper command line arguments so we can control the script Add appveyor support Add an ability to run all other (non db_tests) test executables in parallel Use .NET HashSet instead of empty valued hashtable to improve the looks TODO: Some of the tests do not use GTests and need to improve log parsing
252 lines
6.6 KiB
PowerShell
252 lines
6.6 KiB
PowerShell
# This script enables you running RocksDB tests by running
|
|
# All the tests in paralell and utilizing all the cores
|
|
# For db_test the script first lists and parses the tests
|
|
# and then fires them up in parallel using async PS Job functionality
|
|
# Run the script from the enlistment
|
|
Param(
|
|
[switch]$EnableJE = $false, # Use je executable
|
|
[string]$WorkFolder = "", # Direct tests to use that folder
|
|
[int]$Limit = -1, # -1 means run all otherwise limit for testing purposes
|
|
[string]$Exclude = "", # Expect a comma separated list, no spaces
|
|
[string]$Run = "db_test" # Run db_test|tests
|
|
)
|
|
|
|
# Folders and commands must be fullpath to run assuming
|
|
# the current folder is at the root of the git enlistment
|
|
Get-Date
|
|
|
|
# If running under Appveyor assume that root
|
|
[string]$Appveyor = $Env:APPVEYOR_BUILD_FOLDER
|
|
if($Appveyor -ne "") {
|
|
$RootFolder = $Appveyor
|
|
} else {
|
|
$RootFolder = $PSScriptRoot -replace '\\build_tools', ''
|
|
}
|
|
|
|
$LogFolder = -Join($RootFolder, "\db_logs\")
|
|
$BinariesFolder = -Join($RootFolder, "\build\Debug\")
|
|
|
|
if($WorkFolder -eq "") {
|
|
|
|
# If TEST_TMPDIR is set use it
|
|
[string]$var = $Env:TEST_TMPDIR
|
|
if($var -eq "") {
|
|
$WorkFolder = -Join($RootFolder, "\db_tests\")
|
|
$Env:TEST_TMPDIR = $WorkFolder
|
|
} else {
|
|
$WorkFolder = $var
|
|
}
|
|
} else {
|
|
# Override from a command line
|
|
$Env:TEST_TMPDIR = $WorkFolder
|
|
}
|
|
|
|
# Use JEMALLOC executables
|
|
if($EnableJE) {
|
|
$db_test = -Join ($BinariesFolder, "db_test_je.exe")
|
|
} else {
|
|
$db_test = -Join ($BinariesFolder, "db_test.exe")
|
|
}
|
|
|
|
Write-Output "Root: $RootFolder, WorkFolder: $WorkFolder"
|
|
Write-Output "Binaries: $BinariesFolder exe: $db_test"
|
|
|
|
#Exclusions that we do not want to run
|
|
$ExcludeTests = New-Object System.Collections.Generic.HashSet[string]
|
|
|
|
|
|
if($Exclude -ne "") {
|
|
Write-Host "Exclude: $Exclude"
|
|
$l = $Exclude -split ','
|
|
ForEach($t in $l) { $ExcludeTests.Add($t) | Out-Null }
|
|
}
|
|
|
|
# Create test directories in the current folder
|
|
md -Path $WorkFolder -ErrorAction Ignore | Out-Null
|
|
md -Path $LogFolder -ErrorAction Ignore | Out-Null
|
|
|
|
# Extract the names of its tests by running db_test with --gtest_list_tests.
|
|
# This filter removes the "#"-introduced comments, and expands to
|
|
# fully-qualified names by changing input like this:
|
|
#
|
|
# DBTest.
|
|
# Empty
|
|
# WriteEmptyBatch
|
|
# MultiThreaded/MultiThreadedDBTest.
|
|
# MultiThreaded/0 # GetParam() = 0
|
|
# MultiThreaded/1 # GetParam() = 1
|
|
#
|
|
# into this:
|
|
#
|
|
# DBTest.Empty
|
|
# DBTest.WriteEmptyBatch
|
|
# MultiThreaded/MultiThreadedDBTest.MultiThreaded/0
|
|
# MultiThreaded/MultiThreadedDBTest.MultiThreaded/1
|
|
# Output into the parameter in a form TestName -> Log File Name
|
|
function Normalize-DbTests($HashTable) {
|
|
|
|
$Tests = @()
|
|
# Run db_test to get a list of tests and store it into $a array
|
|
&$db_test --gtest_list_tests | tee -Variable Tests | Out-Null
|
|
|
|
# Current group
|
|
$Group=""
|
|
|
|
ForEach( $l in $Tests) {
|
|
# Trailing dot is a test group
|
|
if( $l -match "\.$") {
|
|
$Group = $l
|
|
} else {
|
|
# Otherwise it is a test name, remove leading space
|
|
$test = $l -replace '^\s+',''
|
|
# remove trailing comment if any and create a log name
|
|
$test = $test -replace '\s+\#.*',''
|
|
$test = "$Group$test"
|
|
|
|
if($ExcludeTests.Contains($test)) {
|
|
continue
|
|
}
|
|
|
|
$test_log = $test -replace '[\./]','_'
|
|
$test_log += ".log"
|
|
|
|
# Add to a hashtable
|
|
$HashTable.Add($test, $test_log);
|
|
}
|
|
}
|
|
}
|
|
|
|
# The function scans build\Debug folder to discover
|
|
# Test executables. It then populates a table with
|
|
# Test executable name -> Log file
|
|
function Discover-TestBinaries($HashTable) {
|
|
|
|
$Exclusions = @("db_test*", "db_sanity_test*")
|
|
$p = -join ($BinariesFolder, "*_test*.exe")
|
|
|
|
dir -Path $p -Exclude $Exclusions | ForEach-Object {
|
|
$t = ($_.Name) -replace '.exe$', ''
|
|
$test_log = -join ($t, ".log")
|
|
$HashTable.Add($t, $test_log)
|
|
}
|
|
}
|
|
|
|
$TestToLog = [ordered]@{}
|
|
|
|
if($Run -ceq "db_test") {
|
|
Normalize-DbTests -HashTable $TestToLog
|
|
} elseif($Run -ceq "tests") {
|
|
Discover-TestBinaries -HashTable $TestToLog
|
|
}
|
|
|
|
|
|
Write-Host "Attempting to start: " ($TestToLog.Count) " tests"
|
|
|
|
# Invoke a test with a filter and redirect all output
|
|
$InvokeTestCase = {
|
|
param($exe, $test, $log);
|
|
&$exe --gtest_filter=$test > $log 2>&1
|
|
}
|
|
|
|
# Invoke all tests and redirect output
|
|
$InvokeTestAsync = {
|
|
param($exe, $log)
|
|
&$exe > $log 2>&1
|
|
}
|
|
|
|
$jobs = @()
|
|
$JobToLog = @{}
|
|
# Test limiting factor here
|
|
$count = 0
|
|
|
|
ForEach($k in $TestToLog.keys) {
|
|
|
|
Write-Host "Starting $k"
|
|
$log_path = -join ($LogFolder, ($TestToLog.$k))
|
|
|
|
if($Run -ceq "db_test") {
|
|
$job = Start-Job -Name $k -ScriptBlock $InvokeTestCase -ArgumentList @($db_test,$k,$log_path)
|
|
} else {
|
|
[string]$Exe = -Join ($BinariesFolder, $k)
|
|
$job = Start-Job -Name $k -ScriptBlock $InvokeTestAsync -ArgumentList @($exe,$log_path)
|
|
}
|
|
|
|
$JobToLog.Add($job, $log_path)
|
|
|
|
# Limiting trial runs
|
|
if(($Limit -gt 0) -and (++$count -ge $Limit)) {
|
|
break
|
|
}
|
|
}
|
|
|
|
[bool]$success = $true;
|
|
|
|
# Wait for all to finish and get the results
|
|
while($JobToLog.Count -gt 0) {
|
|
|
|
$jobs = @()
|
|
foreach($k in $JobToLog.Keys) { $jobs += $k }
|
|
|
|
<#
|
|
if(!$success) {
|
|
break
|
|
}
|
|
#>
|
|
|
|
$completed = Wait-Job -Job $jobs -Any
|
|
$log = $JobToLog[$completed]
|
|
$JobToLog.Remove($completed)
|
|
|
|
$message = -join @($completed.Name, " State: ", ($completed.State))
|
|
|
|
$log_content = @(Get-Content $log)
|
|
|
|
if($completed.State -ne "Completed") {
|
|
$success = $false
|
|
Write-Warning $message
|
|
$log_content | Write-Warning
|
|
} else {
|
|
# Scan the log. If we find PASSED and no occurence of FAILED
|
|
# then it is a success
|
|
[bool]$pass_found = $false
|
|
ForEach($l in $log_content) {
|
|
|
|
if(($l -match "^\[\s+FAILED") -or
|
|
($l -match "Assertion failed:")) {
|
|
$pass_found = $false
|
|
break
|
|
}
|
|
|
|
if(($l -match "^\[\s+PASSED") -or
|
|
($l -match " : PASSED$") -or
|
|
($l -match "^PASSED") -or
|
|
($l -match "Passed all tests!") ) {
|
|
$pass_found = $true
|
|
}
|
|
}
|
|
|
|
if(!$pass_found) {
|
|
$success = $false;
|
|
Write-Warning $message
|
|
$log_content | Write-Warning
|
|
} else {
|
|
Write-Host $message
|
|
}
|
|
}
|
|
|
|
# Remove cached job info from the system
|
|
# Should be no output
|
|
Receive-Job -Job $completed | Out-Null
|
|
}
|
|
|
|
Get-Date
|
|
|
|
if(!$success) {
|
|
# This does not succeed killing off jobs quick
|
|
# So we simply exit
|
|
# Remove-Job -Job $jobs -Force
|
|
# indicate failure using this exit code
|
|
exit 12345
|
|
}
|
|
|
|
|