PowerShell · TravisEz13 · Jan 16, 2019 · Dec 9, 2018 · Dec 9, 2018 · Dec 9, 2018
diff --git a/test/powershell/Modules/PSDiagnostics/PSDiagnostics.Tests.ps1 b/test/powershell/Modules/PSDiagnostics/PSDiagnostics.Tests.ps1
@@ -0,0 +1,93 @@
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License.
+
+Describe "PSDiagnostics cmdlets tests." -Tag "CI", "RequireAdminOnWindows" {
+    BeforeAll {
+        $LogType = 'Analytic'
+        $OriginalDefaultParameterValues = $PSDefaultParameterValues.Clone()
+        if (-not $IsWindows) {
+            $PSDefaultParameterValues["it:skip"] = $true
+        }
+        else{
+            $LogSettingBak = Get-LogProperties -Name Microsoft-Windows-PowerShell/$LogType
+        }
+    }
+    AfterAll {
+        if ($IsWindows) {
+            Set-LogProperties -LogDetails $LogSettingBak -Force
+        }
+        $Global:PSDefaultParameterValues = $OriginalDefaultParameterValues
+    }
+
+    Context "Test for Enable-PSTrace and Disable-PSTrace cmdlets." {
+        it "Should enable $LogType logs for Microsoft-Windows-PowerShell." {
+            [XML]$CurrentSetting = & wevtutil gl Microsoft-Windows-PowerShell/$LogType /f:xml
+            if($CurrentSetting.Channel.Enabled -eq 'true'){
+                & wevtutil sl Microsoft-Windows-PowerShell/$LogType /e:false /q
+            }
+
+            Enable-PSTrace -Force
+
+            [XML]$ExpectedOutput = & wevtutil gl Microsoft-Windows-PowerShell/$LogType /f:xml
+
+            $ExpectedOutput.Channel.enabled | Should -BeExactly 'true'
+        }
+
+        it "Should disable $LogType logs for Microsoft-Windows-PowerShell." {
+            [XML]$CurrentState = & wevtutil gl Microsoft-Windows-PowerShell/$LogType /f:xml
+            if($CurrentState.channel.enabled -eq 'false'){
+                & wevtutil sl Microsoft-Windows-PowerShell/$LogType /e:true /q
+            }
+            Disable-PSTrace
+
+            [XML]$ExpectedOutput = & wevtutil gl Microsoft-Windows-PowerShell/$LogType /f:xml
+
+            $ExpectedOutput.Channel.enabled | Should -Be 'false'
+        }
+    }
+
+    Context "Test for Get-LogProperties cmdlet." {
+        it "Should return properties of $LogType logs for 'Microsoft-Windows-PowerShell'." {
+            [XML]$ExpectedOutput = wevtutil gl Microsoft-Windows-PowerShell/$LogType /f:xml
+
+            $LogProperty = Get-LogProperties -Name Microsoft-Windows-PowerShell/$LogType
+
+            $LogProperty.Name       | Should -Be $ExpectedOutput.channel.Name
+            $LogProperty.Enabled    | Should -Be $ExpectedOutput.channel.Enabled
+            $LogProperty.Retention  | Should -Be $ExpectedOutput.channel.Logging.Retention
+            $LogProperty.AutoBackup | Should -Be $ExpectedOutput.channel.Logging.AutoBackup
+            $LogProperty.MaxLogSize | Should -Be $ExpectedOutput.channel.Logging.MaxSize
+
+            #Verifying the property count. Adding 2 to count from the wevtutil output as the Enabled and Name property as the count is taken only for Logging property.
+            ($LogProperty | Get-Member -MemberType Property | Measure-Object).Count |
+            Should -Be (($ExpectedOutput.Channel.Logging | Get-Member -MemberType Property | Measure-Object).Count + 2)
+        }
+    }
+
+    Context "Test for Set-LogProperties cmdlet." {
+        BeforeAll {
+            if ($IsWindows) {
+                [XML]$WevtUtilBefore = wevtutil gl Microsoft-Windows-PowerShell/$LogType /f:xml
+                $LogPropertyToSet = [Microsoft.PowerShell.Diagnostics.LogDetails]::new($WevtUtilBefore.channel.Name,
+                    [bool]::Parse($WevtUtilBefore.channel.Enabled),
+                    $LogType,
+                    [bool]::Parse($WevtUtilBefore.channel.Logging.Retention),
+                    [bool]::Parse($WevtUtilBefore.channel.Logging.AutoBackup),
+                    $WevtUtilBefore.channel.Logging.MaxSize -as [int]
+                )
+            }
+        }
+
+        it "Should invert AutoBackup setting of $LogType logs for 'Microsoft-Windows-PowerShell'." {
+            $LogPropertyToSet.AutoBackup = -not $LogPropertyToSet.AutoBackup
+            Set-LogProperties -LogDetails $LogPropertyToSet -Force
+
+            [XML]$ExpectedOutput = & wevtutil gl Microsoft-Windows-PowerShell/$LogType /f:xml
+            (Get-LogProperties -Name Microsoft-Windows-PowerShell/$LogType).AutoBackup | Should -Be ([bool]::Parse($ExpectedOutput.Channel.Logging.AutoBackup))
+        }
+
+        it "Should throw exception for invalid LogName." {
+            {Set-LogProperties -LogDetails 'Foo' -Force } | Should -Throw -ErrorId 'ParameterArgumentTransformationError'
+        }
+    }
+}