From 8da5ba4f1c0f27463ce8cc5dfb4ecab334b9e60a Mon Sep 17 00:00:00 2001
From: anteater <65555601+nt8r@users.noreply.github.com>
Date: Fri, 24 Jan 2025 15:14:15 +0000
Subject: [PATCH] Remove `unsafe` by using `LazyLock`

---
 crates/cargo-test-macro/src/lib.rs | 30 ++++++++++++++----------------
 1 file changed, 14 insertions(+), 16 deletions(-)

diff --git a/crates/cargo-test-macro/src/lib.rs b/crates/cargo-test-macro/src/lib.rs
index 01265f8fba0..c3b82fe7695 100644
--- a/crates/cargo-test-macro/src/lib.rs
+++ b/crates/cargo-test-macro/src/lib.rs
@@ -10,7 +10,7 @@
 use proc_macro::*;
 use std::path::Path;
 use std::process::Command;
-use std::sync::Once;
+use std::sync::LazyLock;
 
 /// Replacement for `#[test]`
 ///
@@ -254,23 +254,21 @@ fn to_token_stream(code: &str) -> TokenStream {
     code.parse().unwrap()
 }
 
-static mut VERSION: (u32, bool) = (0, false);
+static VERSION: std::sync::LazyLock<(u32, bool)> = LazyLock::new(|| {
+    let output = Command::new("rustc")
+        .arg("-V")
+        .output()
+        .expect("rustc should run");
+    let stdout = std::str::from_utf8(&output.stdout).expect("utf8");
+    let vers = stdout.split_whitespace().skip(1).next().unwrap();
+    let is_nightly = option_env!("CARGO_TEST_DISABLE_NIGHTLY").is_none()
+        && (vers.contains("-nightly") || vers.contains("-dev"));
+    let minor = vers.split('.').skip(1).next().unwrap().parse().unwrap();
+    (minor, is_nightly)
+});
 
 fn version() -> (u32, bool) {
-    static INIT: Once = Once::new();
-    INIT.call_once(|| {
-        let output = Command::new("rustc")
-            .arg("-V")
-            .output()
-            .expect("rustc should run");
-        let stdout = std::str::from_utf8(&output.stdout).expect("utf8");
-        let vers = stdout.split_whitespace().skip(1).next().unwrap();
-        let is_nightly = option_env!("CARGO_TEST_DISABLE_NIGHTLY").is_none()
-            && (vers.contains("-nightly") || vers.contains("-dev"));
-        let minor = vers.split('.').skip(1).next().unwrap().parse().unwrap();
-        unsafe { VERSION = (minor, is_nightly) }
-    });
-    unsafe { VERSION }
+    LazyLock::force(&VERSION).clone()
 }
 
 fn check_command(command_path: &Path, args: &[&str]) -> bool {