Add Pester tests for streams on Update-AUPackages

tests/Update-AUPackages.Streams.Tests.ps1

@@ -0,0 +1,144 @@
+remove-module AU -ea ignore
+import-module $PSScriptRoot\..\AU
+
+Describe 'Update-AUPackages' -Tag updateallstreams {
+    $saved_pwd = $pwd
+
+    function global:nuspec_file() { [xml](gc $PSScriptRoot/test_package_with_streams/test_package_with_streams.nuspec) }
+    $pkg_no = 2
+    $streams_no = $pkg_no * 3
+
+    BeforeEach {
+        $global:au_Root      = "TestDrive:\packages"
+        $global:au_NoPlugins = $true
+
+        rm -Recurse $global:au_root -ea ignore
+        foreach ( $i in 1..$pkg_no ) {
+            $name = "test_package_with_streams_$i"
+            $path = "$au_root\$name"
+
+            cp -Recurse -Force $PSScriptRoot\test_package_with_streams $path
+            $nu = nuspec_file
+            $nu.package.metadata.id = $name
+            rm "$path\*.nuspec"
+            $nu.OuterXml | sc "$path\$name.nuspec"
+            mv "$path\test_package_with_streams.json" "$path\$name.json"
+
+            $module_path = Resolve-Path $PSScriptRoot\..\AU
+            "import-module '$module_path' -Force", (gc $path\update.ps1 -ea ignore) | sc $path\update.ps1
+        }
+
+        $Options = [ordered]@{}
+    }
+
+    Context 'Plugins' {
+
+        It 'should ignore the package that returns "ignore"' {
+            gc $global:au_Root\test_package_with_streams_1\update.ps1 | set content
+            $content -replace 'update', "Write-Host 'test ignore'; 'ignore'" | set content
+            $content | sc $global:au_Root\test_package_with_streams_1\update.ps1
+
+            $res = updateall -Options $Options -NoPlugins:$false 6>$null
+
+            $res[0].Ignored | Should Be $true
+            $res[0].IgnoreMessage | Should Be 'test ignore'
+        }
+
+        It 'should repeat and ignore on specific error' {
+            gc $global:au_Root\test_package_with_streams_1\update.ps1 | set content
+            $content -replace 'update', "1|Out-File -Append $TestDrive\tmp_test; throw 'test ignore'; update" | set content
+            $content | sc $global:au_Root\test_package_with_streams_1\update.ps1
+
+            $Options.RepeatOn = @('test ignore')
+            $Options.RepeatCount = 2
+            $Options.IgnoreOn = @('test ignore')
+
+            $res = updateall -Options $Options -NoPlugins:$false 6>$null
+
+            $res[0].Ignored | Should Be $true
+            $res[0].IgnoreMessage | Should BeLike 'AU ignored on*test ignore'
+
+            (gc $TestDrive\tmp_test).Count | Should be 3
+        }
+
+        It 'should execute Report plugin' {
+            $Options.Report = @{
+                Type = 'markdown'
+                Path = "$global:au_Root\report.md"
+                Params = @{ Github_UserRepo = 'majkinetor/chocolatey' }
+            }
+
+            updateall -NoPlugins:$false -Options $Options  6> $null
+
+            Test-Path $Options.Report.Path | Should Be $true
+
+            $report = gc $Options.Report.Path 
+            ($report -match "test_package_with_streams_[1-$pkg_no]").Count | Should Be (3 * $pkg_no)
+        }
+
+        It 'should execute RunInfo plugin' {
+            $Options.RunInfo = @{
+                Path    = "$global:au_Root\update_info.xml"
+                Exclude = 'password'
+            }
+            $Options.Test = @{
+                MyPassword = 'password'
+                Parameter2 = 'p2'`
+            }
+
+            $res = updateall -NoPlugins:$false -Options $Options  6> $null
+
+            Test-Path $Options.RunInfo.Path | Should Be $true
+            $info = Import-Clixml $Options.RunInfo.Path
+            $info.plugin_results.RunInfo -match 'Test.MyPassword' | Should Be $true
+            $info.Options.Test.MyPassword | Should Be '*****' 
+        }
+    }
+
+    It 'should update package with checksum verification mode' {
+
+        $choco_path = gcm choco.exe | % Source
+        $choco_hash = Get-FileHash $choco_path -Algorithm SHA256 | % Hash
+        gc $global:au_Root\test_package_with_streams_1\update.ps1 | set content
+        $content -replace '@\{.+\}', "@{ Version = '1.3'; ChecksumType32 = 'sha256'; Checksum32 = '$choco_hash'}" | set content
+        $content -replace 'update', "update -ChecksumFor 32" | set content
+        $content | sc $global:au_Root\test_package_with_streams_1\update.ps1
+
+        $res = updateall -Options $Options 6> $null
+        $res.Count | Should Be $pkg_no
+        $res[0].Updated | Should Be $true
+    }
+
+    It 'should limit update time' {
+        gc $global:au_Root\test_package_with_streams_1\update.ps1 | set content
+        $content -replace 'update', "sleep 10; update" | set content
+        $content | sc $global:au_Root\test_package_with_streams_1\update.ps1
+        $Options.UpdateTimeout = 5
+
+        $res = updateall -Options $Options 3>$null 6> $null
+        $res[0].Error -eq "Job termintated due to the 5s UpdateTimeout" | Should Be $true
+    }
+
+    It 'should update all packages when forced' {
+        $Options.Force = $true
+
+        $res = updateall -Options $Options 6> $null
+
+        lsau | measure | % Count | Should Be $pkg_no
+        $res.Count | Should Be $pkg_no
+        ($res.Result -match 'update is forced').Count | Should Be $pkg_no
+        ($res | ? Updated).Count | Should Be $pkg_no
+    }
+
+    It 'should update no packages when none is newer' {
+        $res = updateall 6> $null
+
+        lsau | measure | % Count | Should Be $pkg_no
+        $res.Count | Should Be $pkg_no
+        ($res.Result -match 'No new version found').Count | Should Be $streams_no
+        ($res | ? Updated).Count | Should Be 0
+    }
+
+    $saved_pwd = $pwd
+}
+

tests/Update-AUPackages.Tests.ps1

@@ -4,25 +4,23 @@ import-module $PSScriptRoot\..\AU
 Describe 'Update-AUPackages' -Tag updateall {
     $saved_pwd = $pwd
 
-    $pkg_no = 5
+    function global:nuspec_file() { [xml](gc $PSScriptRoot/test_package/test_package.nuspec) }
+    $pkg_no = 3
 
     BeforeEach {
-        $stream_no = 0
         $global:au_Root      = "TestDrive:\packages"
         $global:au_NoPlugins = $true
 
         rm -Recurse $global:au_root -ea ignore
         foreach ( $i in 1..$pkg_no ) {
-            $base = if ($i -le 3) {$stream_no++; 'test_package'} else {$stream_no += 3; 'test_package_with_streams'}
-            $name = "${base}_$i"
+            $name = "test_package_$i"
             $path = "$au_root\$name"
 
-            cp -Recurse -Force "$PSScriptRoot\$base" $path
-            $nu = [xml](gc "$PSScriptRoot\$base\$base.nuspec")
+            cp -Recurse -Force $PSScriptRoot\test_package $path
+            $nu = nuspec_file
             $nu.package.metadata.id = $name
             rm "$au_root\$name\*.nuspec"
             $nu.OuterXml | sc "$path\$name.nuspec"
-            if (Test-Path "$path\$base.json") { mv "$path\$base.json" "$path\$name.json" }
 
             $module_path = Resolve-Path $PSScriptRoot\..\AU
             "import-module '$module_path' -Force", (gc $path\update.ps1 -ea ignore) | sc $path\update.ps1
@@ -180,7 +178,7 @@ Describe 'Update-AUPackages' -Tag updateall {
 
         lsau | measure | % Count | Should Be $pkg_no
         $res.Count | Should Be $pkg_no
-        ($res.Result -match 'No new version found').Count | Should Be $stream_no
+        ($res.Result -match 'No new version found').Count | Should Be $pkg_no
         ($res | ? Updated).Count | Should Be 0
     }