Add tests

Author: adityapatwardhan

tests/container.tests.ps1

@@ -240,6 +240,21 @@ Describe "Linux Containers" -Tags 'Behavior', 'Linux' {
             $gssNtlmSspPath = Get-LinuxGssNtlmSsp -Name $Name
             $gssNtlmSspPath | Should -Not -BeNullOrEmpty
         }
+
+        it "Has POWERSHELL_DISTRIBUTION_CHANNEL environment variable defined" -TestCases $runTestCases -Skip:$script:skipLinuxRun {
+            param(
+                [Parameter(Mandatory=$true)]
+                [string]
+                $name,
+
+                [Parameter(Mandatory=$true)]
+                [string]
+                $ExpectedVersion
+            )
+
+            $psDistChannel = Get-PowerShellDistibutionChannel -TestContext $testContext -Name $Name
+            $psDistChannel | Should -BeLike "PSDocker-*"
+        }
     }
 
     Context "Labels" {
@@ -591,6 +606,21 @@ Describe "Windows Containers" -Tags 'Behavior', 'Windows' {
 
             $path | should -Match ([System.Text.RegularExpressions.Regex]::Escape("C:\Windows\system32"))
         }
+
+        it "Has POWERSHELL_DISTRIBUTION_CHANNEL environment variable defined" -TestCases $runTestCases -Skip:$script:skipLinuxRun {
+            param(
+                [Parameter(Mandatory=$true)]
+                [string]
+                $name,
+
+                [Parameter(Mandatory=$true)]
+                [string]
+                $ExpectedVersion
+            )
+
+            $psDistChannel = Get-PowerShellDistibutionChannel -TestContext $testContext -Name $Name
+            $psDistChannel | Should -BeLike "PSDocker-*"
+        }
     }
 
     Context "Labels" {

tests/containerTestCommon.psm1

@@ -311,6 +311,28 @@ function Get-ContainerPath
     return (Invoke-Docker -Command run -Params $runParams -SuppressHostOutput -PassThru)
 }
 
+function Get-PowerShellDistibutionChannelVariable
+{
+    param(
+        [HashTable] $TestContext,
+        [string] $Name
+    )
+
+    $imageTag = ${Name}
+
+    $runParams = @()
+    $runParams += '--rm'
+
+    $runParams += $imageTag
+    $runParams += 'pwsh'
+    $runParams += '-nologo'
+    $runParams += '-noprofile'
+    $runParams += '-c'
+    $runParams += '$env:POWERSHELL_DISTRIBUTION_CHANNEL'
+
+    return (Invoke-Docker -Command run -Params $runParams -SuppressHostOutput -PassThru)
+}
+
 function Get-MetadataUsingContainer
 {
     param(