From ee47b15891776c7c98380fcf28f191497c41f45d Mon Sep 17 00:00:00 2001
From: Jan Plhak <jp@chiselstrike.com>
Date: Fri, 25 Mar 2022 13:31:42 +0100
Subject: [PATCH] Allow the specification of env variables for tests.

---
 src/config.rs             | 3 +++
 src/run/test_evaluator.rs | 1 +
 2 files changed, 4 insertions(+)

diff --git a/src/config.rs b/src/config.rs
index 593b194..494478f 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -49,6 +49,8 @@ pub struct Config
     pub always_show_stderr: bool,
     /// Which shell to use (defaults to 'bash').
     pub shell: String,
+    /// List of environment variables to be used on test invocation.
+    pub env_variables: HashMap<String, String>,
 }
 
 /// A function which can dynamically define newly used variables in a test.
@@ -154,6 +156,7 @@ impl Default for Config
             truncate_output_context_to_number_of_lines: Some(DEFAULT_MAX_OUTPUT_CONTEXT_LINE_COUNT),
             extra_executable_search_paths,
             shell: "bash".to_string(),
+            env_variables: HashMap::default(),
         }
     }
 }
diff --git a/src/run/test_evaluator.rs b/src/run/test_evaluator.rs
index dd3b504..f18295d 100644
--- a/src/run/test_evaluator.rs
+++ b/src/run/test_evaluator.rs
@@ -143,6 +143,7 @@ fn build_command(invocation: &Invocation,
 
     let mut cmd = process::Command::new(&config.shell);
     cmd.args(&["-c", &command_line]);
+    cmd.envs(&config.env_variables);
 
     if !config.extra_executable_search_paths.is_empty() {
         let os_path_separator = if cfg!(windows) { ";" } else { ":" };