Add comprehensive test suite, fix minor issues, rename function and file names to match approved verbs (#708)

2026-08-23 16:12:08 +00:00 · 2026-07-19 22:06:07 +02:00
parent a7292e4f35
commit 9c033dbf98
116 changed files with 4629 additions and 650 deletions
@@ -0,0 +1,108 @@
+BeforeAll {
+    function Get-UserDirectory { param($userName, $fileName, $exitIfPathNotFound) }
+
+    . (Join-Path $PSScriptRoot '..\Scripts\Features\Replace-StartMenu.ps1')
+    . (Join-Path $PSScriptRoot '..\Scripts\Features\Telemetry-ScheduledTasks.ps1')
+}
+
+Describe 'Get-StartMenuUserNameFromPath' {
+    It 'extracts a user name from a start-menu path and falls back for unknown paths' {
+        Get-StartMenuUserNameFromPath -StartMenuBinFile 'C:\Users\Alice\AppData\Local\Packages\Start\start2.bin' | Should -Be 'Alice'
+        Get-StartMenuUserNameFromPath -StartMenuBinFile 'C:\Temp\start2.bin' | Should -Be 'unknown'
+    }
+
+    It 'returns the latest current-user start-menu backup' {
+        Mock Get-ChildItem {
+            @(
+                [PSCustomObject]@{ Name = 'Win11Debloat-StartBackup-20260101_120000.bak'; FullName = 'C:\Backups\older.bak' }
+                [PSCustomObject]@{ Name = 'Win11Debloat-StartBackup-20260102_120000.bak'; FullName = 'C:\Backups\newer.bak' }
+            )
+        }
+
+        Get-StartMenuBackupPath -Scope CurrentUser | Should -Be 'C:\Backups\newer.bak'
+    }
+
+    It 'returns the first available all-users start-menu backup' {
+        $script:allUsersPath = 'C:\Users\*\AppData\Local\Packages\Start\LocalState'
+        Mock Get-UserDirectory { $script:allUsersPath }
+        Mock Get-ChildItem {
+            param($Path)
+            if ($Path -eq $script:allUsersPath) {
+                return [PSCustomObject]@{ FullName = 'C:\Users\Alice\AppData\Local\Packages\Start\LocalState' }
+            }
+
+            return [PSCustomObject]@{ Name = 'Win11Debloat-StartBackup-20260103_120000.bak'; FullName = 'C:\Users\Alice\backup.bak' }
+        }
+
+        Get-StartMenuBackupPath -Scope AllUsers | Should -Be 'C:\Users\Alice\backup.bak'
+    }
+
+    It 'restores a start menu backup and preserves the replaced file' {
+        $script:Params = @{}
+        $startMenuFile = Join-Path $TestDrive 'start2.bin'
+        $backupFile = Join-Path $TestDrive 'Win11Debloat-StartBackup-20260101_120000.bak'
+        Set-Content -LiteralPath $startMenuFile -Value 'current'
+        Set-Content -LiteralPath $backupFile -Value 'backup'
+
+        $result = Restore-StartMenuFromBackup -StartMenuBinFile $startMenuFile -BackupFilePath $backupFile
+
+        $result.Result | Should -BeTrue
+        Get-Content -LiteralPath $startMenuFile -Raw | Should -Match 'backup'
+        @(Get-ChildItem -LiteralPath $TestDrive -Filter 'Win11Debloat-StartRestore-*.bak').Count | Should -Be 1
+    }
+
+    It 'reports a missing backup without changing the start-menu file' {
+        $script:Params = @{}
+        $startMenuFile = Join-Path $TestDrive 'start2.bin'
+        Set-Content -LiteralPath $startMenuFile -Value 'current'
+
+        $result = Restore-StartMenuFromBackup -StartMenuBinFile $startMenuFile -BackupFilePath (Join-Path $TestDrive 'missing.bak')
+
+        $result.Result | Should -BeFalse
+        Get-Content -LiteralPath $startMenuFile -Raw | Should -Match 'current'
+    }
+
+    It 'delegates current-user restore to the common backup operation' {
+        $script:Params = @{}
+        Mock Restore-StartMenuFromBackup { [PSCustomObject]@{ Result = $true } }
+
+        Restore-StartMenu -BackupFilePath 'C:\Backups\backup.bak' | Should -Not -BeNullOrEmpty
+        Should -Invoke Restore-StartMenuFromBackup -Times 1 -Exactly -ParameterFilter { $BackupFilePath -eq 'C:\Backups\backup.bak' }
+    }
+
+    It 'restores every discovered user and removes the default-profile start menu' {
+        $script:Params = @{}
+        $script:allUsersStartPath = 'C:\Users\*\LocalState'
+        $script:defaultStartPath = Join-Path $TestDrive 'DefaultLocalState'
+        $defaultBin = Join-Path $script:defaultStartPath 'start2.bin'
+        New-Item -ItemType Directory -Path $script:defaultStartPath | Out-Null
+        Set-Content -LiteralPath $defaultBin -Value 'template'
+        Mock Get-UserDirectory { param($userName) if ($userName -eq '*') { $script:allUsersStartPath } else { $script:defaultStartPath } }
+        Mock Get-ChildItem { param($Path) if ($Path -eq $script:allUsersStartPath) { [PSCustomObject]@{ FullName = 'C:\Users\Alice\LocalState' } } }
+        Mock Restore-StartMenuFromBackup { [PSCustomObject]@{ UserName = 'Alice'; Result = $true; Message = 'Restored' } }
+
+        $result = @(Restore-StartMenuForAllUsers -BackupFilePath 'C:\Backups\backup.bak')
+
+        $result.Count | Should -Be 2
+        Should -Invoke Restore-StartMenuFromBackup -Times 1 -Exactly
+        Test-Path -LiteralPath $defaultBin | Should -BeFalse
+    }
+}
+
+Describe 'Get-TelemetryScheduledTasks' {
+    It 'returns the expected telemetry task catalog' {
+        $tasks = @(Get-TelemetryScheduledTasks)
+
+        $tasks.Count | Should -Be 8
+        @($tasks | ForEach-Object { "$($_.Path)|$($_.Name)" }) | Should -Be @(
+            '\Microsoft\Windows\Application Experience\|Microsoft Compatibility Appraiser'
+            '\Microsoft\Windows\Application Experience\|Microsoft Compatibility Appraiser Exp'
+            '\Microsoft\Windows\Application Experience\|ProgramDataUpdater'
+            '\Microsoft\Windows\Application Experience\|StartupAppTask'
+            '\Microsoft\Windows\Customer Experience Improvement Program\|Consolidator'
+            '\Microsoft\Windows\Customer Experience Improvement Program\|UsbCeip'
+            '\Microsoft\Windows\DiskDiagnostic\|Microsoft-Windows-DiskDiagnosticDataCollector'
+            '\Microsoft\Windows\Autochk\|Proxy'
+        )
+    }
+}