Improve test running scripts

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
main
Dmitri Smirnov 9 years ago
parent 03b08ba9a9
commit 65324a16dc
  1. 174
      build_tools/run_ci_db_test.ps1

@ -1,3 +1,70 @@
# 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. # Extract the names of its tests by running db_test with --gtest_list_tests.
# This filter removes the "#"-introduced comments, and expands to # This filter removes the "#"-introduced comments, and expands to
# fully-qualified names by changing input like this: # fully-qualified names by changing input like this:
@ -15,44 +82,17 @@
# DBTest.WriteEmptyBatch # DBTest.WriteEmptyBatch
# MultiThreaded/MultiThreadedDBTest.MultiThreaded/0 # MultiThreaded/MultiThreadedDBTest.MultiThreaded/0
# MultiThreaded/MultiThreadedDBTest.MultiThreaded/1 # MultiThreaded/MultiThreadedDBTest.MultiThreaded/1
# Output into the parameter in a form TestName -> Log File Name
function Normalize-DbTests($HashTable) {
# Folders and commands must be fullpath to run assuming $Tests = @()
# the current folder is at the root of the git enlistment # Run db_test to get a list of tests and store it into $a array
Get-Date &$db_test --gtest_list_tests | tee -Variable Tests | Out-Null
# Limit the number of tests to start for debugging purposes
$limit = -1
$RootFolder = $pwd -replace '\\build_tools', ''
$LogFolder = -Join($RootFolder, "\db_logs\")
$TmpFolder = -Join($RootFolder, "\db_tests\")
$Env:TEST_TMPDIR = $TmpFolder
$global:db_test = -Join ($RootFolder, "\build\Debug\db_test.exe")
#Exclusions that we do not want to run
$ExcludeTests = @{
<#
"DBTest.HugeNumberOfLevels" = ""
"DBTest.SparseMerge" = ""
"DBTest.RateLimitingTest" = ""
"DBTest.kAbsoluteConsistency" = ""
"DBTest.GroupCommitTest" = ""
"DBTest.FileCreationRandomFailure" = ""
"DBTest.kTolerateCorruptedTailRecords" = ""
"DBTest.kSkipAnyCorruptedRecords" = ""
"DBTest.kPointInTimeRecovery" = ""
"DBTest.Randomized" = ""
#>
}
# Create test directories in the current folder
md -Path $TmpFolder -ErrorAction Ignore
md -Path $LogFolder -ErrorAction Ignore
function Normalize-Tests([System.Array]$Tests, $HashTable) {
# Current group # Current group
$Group="" $Group=""
ForEach( $l in $tests) { ForEach( $l in $Tests) {
# Trailing dot is a test group # Trailing dot is a test group
if( $l -match "\.$") { if( $l -match "\.$") {
$Group = $l $Group = $l
@ -67,7 +107,7 @@ function Normalize-Tests([System.Array]$Tests, $HashTable) {
continue continue
} }
$test_log = $test -replace '[./]','_' $test_log = $test -replace '[\./]','_'
$test_log += ".log" $test_log += ".log"
# Add to a hashtable # Add to a hashtable
@ -76,22 +116,44 @@ function Normalize-Tests([System.Array]$Tests, $HashTable) {
} }
} }
# Run db_test to get a list of tests and store it into $a array # The function scans build\Debug folder to discover
&$db_test --gtest_list_tests | tee -Variable TestList | Out-Null # 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)
}
}
# Parse the tests and store along with the log name into a hash
$TestToLog = [ordered]@{} $TestToLog = [ordered]@{}
Normalize-Tests -Tests $TestList -HashTable $TestToLog 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" Write-Host "Attempting to start: " ($TestToLog.Count) " tests"
# Start jobs async each running a separate test # Invoke a test with a filter and redirect all output
$AsyncScript = { $InvokeTestCase = {
param($exe, $test, $log); param($exe, $test, $log);
&$exe --gtest_filter=$test > $log 2>&1 &$exe --gtest_filter=$test > $log 2>&1
} }
# Invoke all tests and redirect output
$InvokeTestAsync = {
param($exe, $log)
&$exe > $log 2>&1
}
$jobs = @() $jobs = @()
$JobToLog = @{} $JobToLog = @{}
# Test limiting factor here # Test limiting factor here
@ -101,17 +163,23 @@ ForEach($k in $TestToLog.keys) {
Write-Host "Starting $k" Write-Host "Starting $k"
$log_path = -join ($LogFolder, ($TestToLog.$k)) $log_path = -join ($LogFolder, ($TestToLog.$k))
$job = Start-Job -Name $k -ScriptBlock $AsyncScript -ArgumentList @($db_test,$k,$log_path)
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) $JobToLog.Add($job, $log_path)
# Limiting trial runs # Limiting trial runs
if(($limit -gt 0) -and (++$count -ge $limit)) { if(($Limit -gt 0) -and (++$count -ge $Limit)) {
break break
} }
} }
[bool]$success = $true;
$success = 1;
# Wait for all to finish and get the results # Wait for all to finish and get the results
while($JobToLog.Count -gt 0) { while($JobToLog.Count -gt 0) {
@ -134,27 +202,31 @@ while($JobToLog.Count -gt 0) {
$log_content = @(Get-Content $log) $log_content = @(Get-Content $log)
if($completed.State -ne "Completed") { if($completed.State -ne "Completed") {
$success = 0 $success = $false
Write-Warning $message Write-Warning $message
$log_content | Write-Warning $log_content | Write-Warning
} else { } else {
# Scan the log. If we find PASSED and no occurence of FAILED # Scan the log. If we find PASSED and no occurence of FAILED
# then it is a success # then it is a success
$pass_found = 0 [bool]$pass_found = $false
ForEach($l in $log_content) { ForEach($l in $log_content) {
if($l -match "^\[\s+FAILED") { if(($l -match "^\[\s+FAILED") -or
$pass_found = 0 ($l -match "Assertion failed:")) {
$pass_found = $false
break break
} }
if($l -match "^\[\s+PASSED") { if(($l -match "^\[\s+PASSED") -or
$pass_found = 1 ($l -match " : PASSED$") -or
($l -match "^PASSED") -or
($l -match "Passed all tests!") ) {
$pass_found = $true
} }
} }
if(!$pass_found) { if(!$pass_found) {
$success = 0; $success = $false;
Write-Warning $message Write-Warning $message
$log_content | Write-Warning $log_content | Write-Warning
} else { } else {

Loading…
Cancel
Save