From 37d68093da7c3b35d5516d3cbe966b2c0d0d17c2 Mon Sep 17 00:00:00 2001
From: Guillaume Gomez <guillaume.gomez@huawei.com>
Date: Mon, 27 Nov 2023 13:47:43 +0100
Subject: [PATCH] Add ui tests for `--env` option

---
 tests/ui/extenv/extenv-env-overload.rs | 9 +++++++++
 tests/ui/extenv/extenv-env.rs          | 5 +++++
 tests/ui/extenv/extenv-not-env.rs      | 7 +++++++
 3 files changed, 21 insertions(+)
 create mode 100644 tests/ui/extenv/extenv-env-overload.rs
 create mode 100644 tests/ui/extenv/extenv-env.rs
 create mode 100644 tests/ui/extenv/extenv-not-env.rs

diff --git a/tests/ui/extenv/extenv-env-overload.rs b/tests/ui/extenv/extenv-env-overload.rs
new file mode 100644
index 0000000000000..b82bb2fe9661a
--- /dev/null
+++ b/tests/ui/extenv/extenv-env-overload.rs
@@ -0,0 +1,9 @@
+// run-pass
+// rustc-env:MY_VAR=tadam
+// compile-flags: --env MY_VAR=123abc -Zunstable-options
+
+// This test ensures that variables provided with `--env` take precedence over
+// variables from environment.
+fn main() {
+    assert_eq!(env!("MY_VAR"), "123abc");
+}
diff --git a/tests/ui/extenv/extenv-env.rs b/tests/ui/extenv/extenv-env.rs
new file mode 100644
index 0000000000000..9fda52b894111
--- /dev/null
+++ b/tests/ui/extenv/extenv-env.rs
@@ -0,0 +1,5 @@
+// compile-flags: --env FOO=123abc -Zunstable-options
+// run-pass
+fn main() {
+    assert_eq!(env!("FOO"), "123abc");
+}
diff --git a/tests/ui/extenv/extenv-not-env.rs b/tests/ui/extenv/extenv-not-env.rs
new file mode 100644
index 0000000000000..d6c4a43b0032a
--- /dev/null
+++ b/tests/ui/extenv/extenv-not-env.rs
@@ -0,0 +1,7 @@
+// run-pass
+// rustc-env:MY_ENV=/
+// Ensures that variables not defined through `--env` are still available.
+
+fn main() {
+    assert!(!env!("MY_ENV").is_empty());
+}