[Windows,Ubuntu] Update disk space pester test (#13128)

Author: shamil-mubarakshin

images/ubuntu/scripts/tests/System.Tests.ps1

@@ -2,8 +2,11 @@
 # If the test is running on the ADO pipeline or on the GitHub actions, the test will be skipped
 Describe "Disk free space" -Skip:(-not [String]::IsNullOrEmpty($env:AGENT_NAME) -or -not [String]::IsNullOrEmpty($env:RUNNER_NAME)) {
     It "Image has enough free space" {
-        $freeSpace = (Get-PSDrive "/").Free
-        $freeSpace | Should -BeGreaterOrEqual 17GB
+        $diskInfo = Get-PSDrive "/"
+        $totalSpaceGB = [math]::Floor(($diskInfo.Used + $diskInfo.Free) / 1GB)
+        $freeSpaceGB = [math]::Floor($diskInfo.Free / 1GB)
+        Write-Host "  [i] Disk size: ${totalSpaceGB} GB; Free space: ${freeSpaceGB} GB"
+        $freeSpaceGB | Should -BeGreaterOrEqual 17
     }
 }
 

images/windows/scripts/tests/WindowsFeatures.Tests.ps1

@@ -20,11 +20,12 @@ Describe "WindowsFeatures" {
 }
 
 Describe "DiskSpace" {
-    It "The image has enough disk space"{
-        $availableSpaceMB = [math]::Round((Get-PSDrive -Name C).Free / 1MB)
-        $minimumFreeSpaceMB = 18 * 1024
-
-        $availableSpaceMB | Should -BeGreaterThan $minimumFreeSpaceMB
+    It "The image has enough disk space" {
+        $diskInfo = Get-PSDrive -Name C
+        $totalSpaceGB = [math]::Floor(($diskInfo.Used + $diskInfo.Free) / 1GB)
+        $freeSpaceGB = [math]::Floor($diskInfo.Free / 1GB)
+        Write-Host "  [i] Disk size: ${totalSpaceGB} GB; Free space: ${freeSpaceGB} GB"
+        $freeSpaceGB | Should -BeGreaterOrEqual 18
     }
 }