Quick integration tests of terraform modules using powershell

run-tests.ps1

Function TC-Escape ([string] $text){
    return $text.Replace("|","||").Replace("'","|'").Replace("\n","|n").Replace("\r","|r").Replace("[","|[").Replace("]","|}")
}

Function Invoke-Exec
{
    [CmdletBinding()] param([Parameter(Position=0,Mandatory=1)][scriptblock]$cmd)

    $scriptExpanded =  $ExecutionContext.InvokeCommand.ExpandString($cmd).Trim()
    Write-Host "Executing command: $scriptExpanded"

    & $cmd

    if ($lastexitcode -ne 0) {
        throw ("Invoke-Exec failed with exit code '$lastexitcode' executing command " + $scriptExpanded)
    }
}

Function Invoke-TerraformModuleTests {
    # parameters
    $TestScriptName = "main.tf"
    $modulesTestPath = "$PSScriptRoot\modules\*\test\"
    $passed = 0; $failed = 0;
    $IsTeamcity = If(Test-Path env:\TEAMCITY_VERSION) { $true } Else { $false }

    # Recurse through all child folders to find all the tests and execute each one as a seperate team city test
    Get-ChildItem $modulesTestPath -Filter $TestScriptName -Recurse | 
    Foreach-Object {
        $testName = ($_.FullName | Resolve-Path -Relative).Trim(".","\").Replace("\$TestScriptName", "").Replace("\"," - ")
        $folder = Split-Path $_.FullName -Parent

        if($IsTeamcity){
            Write-Host "##teamcity[testStarted name='$testName' captureStandardOutput='true']"
        }
        $elapsed = [System.Diagnostics.Stopwatch]::StartNew()

        try{
            Push-Location $folder
            Invoke-Exec { terraform init }
            Invoke-Exec { terraform plan }
            Write-Host "Test passed: $testName" -foregroundcolor "green"
            $passed++
        }
        catch {
            Write-Host "Test failed: $($_.Exception)" -foregroundcolor "red"
            if($IsTeamcity){
                Write-Host "##teamcity[testFailed name='$(TC-Escape($testName))' message='$(TC-Escape($_.Exception.Message))']" -foregroundcolor "red"
            } 
            $failed++
        }
        finally{
            $elapsed.Stop()
            if($IsTeamcity){
                Write-Host "##teamcity[testFinished name='$(TC-Escape($testName))' duration='$($elapsed.ElapsedMilliseconds)']" -foregroundcolor "green"
            }
            Pop-Location
        }
    }

    $color = if($failed -eq 0) {"green"} else {"red"}
    Write-Host "Test complete $passed passed and $failed failed" -foregroundcolor $color
}

Invoke-TerraformModuleTests