From 6e0160fd7656d614337e4d94518ba44ace671347 Mon Sep 17 00:00:00 2001
From: Medeni Baykal <433724+Haplois@users.noreply.github.com>
Date: Fri, 25 Jun 2021 12:18:55 +0200
Subject: [PATCH] Add missing framework references for WinUI.

 * Added missing framework references
 * Updated license headers on PowerShell files
 * Included a script to install required Windows SDK
---
 scripts/Install-WindowsSDK.ps1                | 40 ++++++-------------
 .../Extension.WinUI/Extension.WinUI.csproj    |  5 +++
 2 files changed, 17 insertions(+), 28 deletions(-)

diff --git a/scripts/Install-WindowsSDK.ps1 b/scripts/Install-WindowsSDK.ps1
index 9241f6bf3..ceed45282 100644
--- a/scripts/Install-WindowsSDK.ps1
+++ b/scripts/Install-WindowsSDK.ps1
@@ -1,34 +1,18 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT license. See LICENSE file in the project root for full license information.
 
-function Write-Log ([string] $message, $messageColor = "Green") {
-    $currentColor = $Host.UI.RawUI.ForegroundColor
-    $Host.UI.RawUI.ForegroundColor = $messageColor
-    if ($message) {
-      Write-Output "... $message"
-    }
-    $Host.UI.RawUI.ForegroundColor = $currentColor
-}
-
-Push-Location
-try {
-    Write-Log "Downloading the Windows SDK 10.0.14393.795..."
-    Invoke-WebRequest -Method Get -Uri https://go.microsoft.com/fwlink/p/?LinkId=838916 -OutFile sdksetup.exe -UseBasicParsing
+Write-Host "Downloading Windows SDK 10.0.14393.795..." -ForegroundColor Green
+Invoke-WebRequest -Method Get -Uri https://go.microsoft.com/fwlink/p/?LinkId=838916 -OutFile sdksetup.exe -UseBasicParsing
 
-    Write-Log "Installing the Windows SDK, setup might request elevation please approve."
-    $process = Start-Process -Wait sdksetup.exe -ArgumentList "/quiet", "/norestart", "/ceip off", "/features OptionId.WindowsSoftwareDevelopmentKit"  -PassThru
-    Remove-Item sdksetup.exe -Force
+Write-Host "Installing Windows SDK, if setup requests elevation please approve." -ForegroundColor Green
+$process = Start-Process -Wait sdksetup.exe -ArgumentList "/quiet", "/norestart", "/ceip off", "/features OptionId.WindowsSoftwareDevelopmentKit"  -PassThru
+Remove-Item sdksetup.exe -Force
 
-    if($process.ExitCode -eq 0) 
-    {
-        Write-Log "Done"
-    }
-    else 
-    {
-        Write-Log "Failed (Exit code: $($process.ExitCode))" -messageColor "Red"
-    }
+if($process.ExitCode -eq 0) 
+{
+    Write-Host "Done" -ForegroundColor Green
 }
-finally {
-    Pop-Location
-}
-
+else 
+{
+    Write-Error "Failed to install Windows SDK (Exit code: $($process.ExitCode))"
+}
\ No newline at end of file
diff --git a/src/TestFramework/Extension.WinUI/Extension.WinUI.csproj b/src/TestFramework/Extension.WinUI/Extension.WinUI.csproj
index a5bbdfdf2..5ba1aef9a 100644
--- a/src/TestFramework/Extension.WinUI/Extension.WinUI.csproj
+++ b/src/TestFramework/Extension.WinUI/Extension.WinUI.csproj
@@ -20,6 +20,11 @@
     <AppendTargetFrameworkToOutputPath>false</AppendTargetFrameworkToOutputPath>
     <RuntimeIdentifiers>win10-x86;win10-x64;win10-arm64</RuntimeIdentifiers>
   </PropertyGroup>
+  
+  <ItemGroup> 
+    <FrameworkReference Update="Microsoft.Windows.SDK.NET.Ref" RuntimeFrameworkVersion="10.0.18362.16" /> 
+    <FrameworkReference Update="Microsoft.Windows.SDK.NET.Ref" TargetingPackVersion="10.0.18362.16" />
+  </ItemGroup>
 
   <ItemGroup>
     <ProjectReference Include="$(RepoRoot)src\TestFramework\MSTest.Core\MSTest.Core.csproj" />