feat(self-update): add with_saved_programs_display_version()

Author: rami3l

src/cli/self_update/test.rs

@@ -48,6 +48,18 @@ fn restore_path(p: Option<RegValue>) {
 #[cfg(unix)]
 fn restore_path(_: Option<()>) {}
 
+#[cfg(windows)]
+pub fn with_saved_programs_display_version(f: &mut dyn FnMut()) {
+    let root = &RegKey::predef(HKEY_CURRENT_USER);
+    let key = super::windows::RUSTUP_UNINSTALL_ENTRY;
+    let name = "DisplayVersion";
+    with_saved_global_state(
+        || get_reg_value(root, key, name),
+        |p| restore_reg_value(root, key, name, p),
+        f,
+    )
+}
+
 #[cfg(windows)]
 fn get_reg_value(root: &RegKey, subkey: &str, name: &str) -> io::Result<Option<RegValue>> {
     let subkey = root

src/cli/self_update/windows.rs

@@ -498,7 +498,8 @@ pub(crate) fn do_remove_from_path() -> Result<()> {
     _apply_new_path(new_path)
 }
 
-const RUSTUP_UNINSTALL_ENTRY: &str = r"Software\Microsoft\Windows\CurrentVersion\Uninstall\Rustup";
+pub(super) const RUSTUP_UNINSTALL_ENTRY: &str =
+    r"Software\Microsoft\Windows\CurrentVersion\Uninstall\Rustup";
 
 fn rustup_uninstall_reg_key() -> Result<RegKey> {
     Ok(RegKey::predef(HKEY_CURRENT_USER)

src/test.rs

@@ -19,6 +19,9 @@ pub use crate::cli::self_update::test::{get_path, with_saved_path};
 use crate::currentprocess;
 use crate::dist::dist::TargetTriple;
 
+#[cfg(windows)]
+pub use crate::cli::self_update::test::with_saved_programs_display_version;
+
 // Things that can have environment variables applied to them.
 pub trait Env {
     fn env<K, V>(&mut self, key: K, val: V)