Test CARGO environment replacement

Author: ChrisDenton

Diff for: src/test/clitools.rs

@@ -684,6 +684,7 @@ async fn setup_test_state(test_dist_dir: tempfile::TempDir) -> (tempfile::TempDi
     //   **in this process** when the tests are still running.
     unsafe {
         // Unset env variables that will break our testing
+        env::remove_var("CARGO");
         env::remove_var("RUSTUP_UPDATE_ROOT");
         env::remove_var("RUSTUP_TOOLCHAIN");
         env::remove_var("SHELL");

Diff for: src/test/mock_bin_src.rs

@@ -94,6 +94,14 @@ fn main() {
             let mut out = io::stderr();
             writeln!(out, "{}", std::env::var("PATH").unwrap()).unwrap();
         }
+        Some("--echo-cargo-env") => {
+            let mut out = io::stderr();
+            if let Ok(cargo) = std::env::var("CARGO") {
+                writeln!(out, "{cargo}").unwrap();
+            } else {
+                panic!("CARGO environment variable not set");
+            }
+        }
         arg => panic!("bad mock proxy commandline: {:?}", arg),
     }
 }

Diff for: tests/suite/cli_misc.rs

@@ -1233,3 +1233,47 @@ async fn toolchain_install_multi_components_comma() {
             .await;
     }
 }
+
+#[tokio::test]
+async fn rustup_updates_cargo_env_if_proxy() {
+    let mut cx = CliTestContext::new(Scenario::SimpleV2).await;
+    cx.config.expect_ok(&["rustup", "default", "stable"]).await;
+
+    use std::env::consts::EXE_SUFFIX;
+    let proxy_path = cx
+        .config
+        .workdir
+        .borrow()
+        .join("bin")
+        .join(format!("cargo{EXE_SUFFIX}"));
+    let real_path = cx.config.run("rustup", &["which", "cargo"], &[]).await;
+    assert!(real_path.ok);
+    let real_path = real_path.stdout;
+
+    fs::create_dir_all(proxy_path.parent().unwrap()).unwrap();
+    #[cfg(windows)]
+    if std::os::windows::fs::symlink_file("rustup", &proxy_path).is_err() {
+        // skip this test on Windows if symlinking isn't enabled.
+        return;
+    }
+    #[cfg(unix)]
+    std::os::unix::fs::symlink("rustup", &proxy_path).unwrap();
+
+    // If CARGO isn't set then we should not set it.
+    cx.config
+        .expect_err(
+            &["cargo", "--echo-cargo-env"],
+            "CARGO environment variable not set",
+        )
+        .await;
+
+    // If CARGO is set to a proxy then change it to the real CARGO path
+    cx.config
+        .expect_ok_ex_env(
+            &["cargo", "--echo-cargo-env"],
+            &[("CARGO", proxy_path.to_str().unwrap())],
+            "",
+            &real_path,
+        )
+        .await;
+}