From d39e30cba9afe5bc9094dd7584496dad17c07a5e Mon Sep 17 00:00:00 2001
From: Nir Bar <nir.bar@panel-sw.co.il>
Date: Wed, 23 Feb 2022 10:54:42 +0200
Subject: [PATCH] MONGOSH-1103: Reload previous selection of user/machine
 installation context

---
 packages/build/test/fixtures/msi-template/Product.wxs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/packages/build/test/fixtures/msi-template/Product.wxs b/packages/build/test/fixtures/msi-template/Product.wxs
index 1055ffc715..71cdf12ef2 100644
--- a/packages/build/test/fixtures/msi-template/Product.wxs
+++ b/packages/build/test/fixtures/msi-template/Product.wxs
@@ -19,6 +19,14 @@
     <SetProperty Id="ARPINSTALLLOCATION" Value="[INSTALLFOLDER]" Sequence="execute" Before="InstallInitialize"/>
     <SetDirectory Id="INSTALLFOLDER" Sequence="first" Value="[PREV_INSTALLFOLDER_]"><![CDATA[Not INSTALLFOLDER And PREV_INSTALLFOLDER_]]></SetDirectory>
 
+    <!-- Restore per user/machine context on upgrade -->
+    <Property Id="IS_PER_USER_">
+      <RegistrySearch Type="raw" Root="HKCU" Key="Software\Microsoft\Windows\CurrentVersion\Uninstall\[WIX_UPGRADE_DETECTED]" Name="InstallLocation" Id="HKCU.upgrade.InstallLocation1"/>
+      <RegistrySearch Type="raw" Root="HKCU" Key="Software\Microsoft\Windows\CurrentVersion\Uninstall\[ProductCode]" Name="InstallLocation" Id="HKCU.curr.InstallLocation1"/>
+    </Property>
+    <SetProperty Id="MSIINSTALLPERUSER" Value="1" Sequence="first" After="AppSearch"><![CDATA[IS_PER_USER_]]></SetProperty>
+    <SetProperty Id="MSIINSTALLPERUSER" Action="UnsetMSIINSTALLPERUSER" Value="[nothing]" Sequence="first" After="AppSearch"><![CDATA[Not IS_PER_USER_ And WIX_UPGRADE_DETECTED]]></SetProperty>
+
     <Feature Id="Mongosh" Title="Mongosh" Level="1">
       <ComponentGroupRef Id="Files" />
     </Feature>