adding AlwaysRollback mode

Author: nvarscar

functions/Install-DBOPackage.ps1

@@ -27,6 +27,7 @@
         - SingleTransaction: wrap all the deployment scripts into a single transaction and rollback whole deployment on error
         - TransactionPerScript: wrap each script into a separate transaction; rollback single script deployment in case of error
         - NoTransaction: deploy as is
+        - AlwaysRollback: deploy as a single transaction, but rollback changes after the deployment is done
 
         Default: NoTransaction
 
@@ -149,7 +150,7 @@
         [string]$SqlInstance,
         [Parameter(Position = 3)]
         [string]$Database,
-        [ValidateSet('SingleTransaction', 'TransactionPerScript', 'NoTransaction')]
+        [ValidateSet('SingleTransaction', 'TransactionPerScript', 'NoTransaction', 'AlwaysRollback')]
         [string]$DeploymentMethod = 'NoTransaction',
         [int]$ConnectionTimeout,
         [int]$ExecutionTimeout,

functions/Install-DBOScript.ps1

@@ -27,6 +27,7 @@
         - SingleTransaction: wrap all the deployment scripts into a single transaction and rollback whole deployment on error
         - TransactionPerScript: wrap each script into a separate transaction; rollback single script deployment in case of error
         - NoTransaction: deploy as is
+        - AlwaysRollback: deploy as a single transaction, but rollback changes after the deployment is done
 
         Default: NoTransaction
 
@@ -158,7 +159,7 @@
         [string]$SqlInstance,
         [Parameter(Position = 3)]
         [string]$Database,
-        [ValidateSet('SingleTransaction', 'TransactionPerScript', 'NoTransaction')]
+        [ValidateSet('SingleTransaction', 'TransactionPerScript', 'NoTransaction', 'AlwaysRollback')]
         [string]$DeploymentMethod = 'NoTransaction',
         [int]$ConnectionTimeout,
         [int]$ExecutionTimeout,

internal/functions/Get-DbUpBuilder.ps1

@@ -62,6 +62,9 @@ function Get-DbUpBuilder {
     elseif ($Config.DeploymentMethod -eq 'TransactionPerScript') {
         $dbUp = [StandardExtensions]::WithTransactionPerScript($dbUp)
     }
+    elseif ($Config.DeploymentMethod -eq 'AlwaysRollback') {
+        $dbUp = [StandardExtensions]::WithTransactionAlwaysRollback($dbUp)
+    }
     # Adding execution timeout - defaults to unlimited execution
     $dbUp = [StandardExtensions]::WithExecutionTimeout($dbUp, [timespan]::FromSeconds($config.ExecutionTimeout))
     return $dbUp

tests/functional/Install-DBOPackage.Tests.ps1

@@ -77,6 +77,24 @@ Describe "<type> Install-DBOPackage functional tests" -Tag FunctionalTests -ForE
             'd' | Should -Not -BeIn $after.(Get-ColumnName name)
         }
     }
+    Context "testing rollback deployment" {
+        BeforeAll {
+            if ($Type -in 'MySQL', 'Oracle') {
+                Set-ItResult -Skipped -Because "CREATE TABLE cannot be rolled back in $Type"
+            }
+            $p1 = New-DBOPackage -ScriptPath (Get-PackageScript -Version 1) -Name "$workFolder\pv1" -Build 1.0 -Force
+            Reset-TestDatabase
+        }
+        It "should rollback version 1.0" {
+            $testResults = Install-DBOPackage $p1 -Build '1.0' @dbConnectionParams -SchemaVersionTable $logTable -DeploymentMethod AlwaysRollback
+            $testResults | Test-DeploymentOutput -Version 1 -HasJournal
+            $testResults.SourcePath | Should -Be (Join-PSFPath -Normalize "$workFolder\pv1.zip")
+
+            "Beginning transaction" | Should -BeIn $testResults.DeploymentLog
+            "Success! No errors have occured when executing scripts, transaction will be rolled back" | Should -BeIn $testResults.DeploymentLog
+            Test-DeploymentState -Version 0
+        }
+    }
     Context "testing regular deployment" {
         BeforeAll {
             $p1 = New-DBOPackage -ScriptPath (Get-PackageScript -Version 1) -Name "$workFolder\pv1" -Build 1.0 -Force

tests/functional/Install-DBOScript.Tests.ps1

@@ -37,7 +37,6 @@ Describe "<type> Install-DBOScript functional tests" -Tag FunctionalTests -ForEa
             if ($Type -in 'MySQL', 'Oracle') {
                 Set-ItResult -Skipped -Because "CREATE TABLE cannot be rolled back in $Type"
             }
-            #Running package
             {
                 $null = Install-DBOScript -Path $tranFailScripts @dbConnectionParams -SchemaVersionTable $logTable -DeploymentMethod SingleTransaction
             } | Should -Throw (Get-TableExistsMessage "a")
@@ -49,7 +48,6 @@ Describe "<type> Install-DBOScript functional tests" -Tag FunctionalTests -ForEa
             $null = Invoke-DBOQuery @dbConnectionParams -InputFile $cleanupScript
         }
         It "should throw an error and create one object" {
-            #Running package
             try {
                 $null = Install-DBOScript -Path $tranFailScripts @dbConnectionParams -SchemaVersionTable $logTable -DeploymentMethod NoTransaction
             }
@@ -89,6 +87,23 @@ Describe "<type> Install-DBOScript functional tests" -Tag FunctionalTests -ForEa
             Test-DeploymentState -Script -Version 2 -HasJournal
         }
     }
+    Context "testing rollback deployment" {
+        BeforeAll {
+            if ($Type -in 'MySQL', 'Oracle') {
+                Set-ItResult -Skipped -Because "CREATE TABLE cannot be rolled back in $Type"
+            }
+            Reset-TestDatabase
+        }
+        It "should rollback version 1.0" {
+            $testResults = Install-DBOScript -ScriptPath (Get-PackageScript -Version 1) @dbConnectionParams -SchemaVersionTable $logTable -DeploymentMethod AlwaysRollback
+            $testResults | Test-DeploymentOutput -Version 1 -HasJournal -Script
+            $testResults.SourcePath | Should -Be (Get-PackageScript -Version 1)
+
+            "Beginning transaction" | Should -BeIn $testResults.DeploymentLog
+            "Success! No errors have occured when executing scripts, transaction will be rolled back" | Should -BeIn $testResults.DeploymentLog
+            Test-DeploymentState -Script -Version 0
+        }
+    }
     Context "testing deployment order" {
         BeforeAll {
             Reset-TestDatabase