SERVER-81436 Read multiversion config from file instead of resmoke output (#69)

Author: Trevor159

.gitignore

@@ -1,3 +1,4 @@
 target/
 .emm-local.yml
 .vscode
+multiversion-config.yml

CHANGELOG.md

@@ -1,5 +1,8 @@
 # Changelog
 
+## 0.7.10 - 2022-09-25
+* SERVER-81436 Read multiversion config from file instead of resmoke output
+
 ## 0.7.9 - 2022-07-31
 * DAG-2777: Added better logging around yaml failures
 

Cargo.lock

@@ -2,7 +2,7 @@
 name = "mongo-task-generator"
 description = "Dynamically split evergreen tasks into subtasks for testing the mongodb/mongo project."
 license = "Apache-2.0"
-version = "0.7.9"
+version = "0.7.10"
 repository = "https://github.com/mongodb/mongo-task-generator"
 authors = ["Decision Automation Group <[email protected]>"]
 edition = "2018"

Cargo.toml

@@ -156,15 +156,20 @@ impl MultiversionConfig {
     /// Query the multiversion configuration from resmoke.
     pub fn from_resmoke(cmd: &str, script: &[String]) -> Result<MultiversionConfig> {
         let mut cmd = vec![cmd];
+        let file_name = "multiversion-config.yml";
         cmd.append(&mut script.iter().map(|s| s.as_str()).collect());
         cmd.append(&mut vec!["multiversion-config"]);
-        let cmd_output = run_command(&cmd).unwrap();
+        let file_arg = format!("--config-file-output={}", file_name);
+        cmd.append(&mut vec![&file_arg]);
+        run_command(&cmd).unwrap();
+        let multiversion_config_output =
+            std::fs::read_to_string(file_name).expect("Multiversion config file not found.");
         let multiversion_config: Result<MultiversionConfig, serde_yaml::Error> =
-            serde_yaml::from_str(&cmd_output);
+            serde_yaml::from_str(&multiversion_config_output);
         if multiversion_config.is_err() {
             error!(
                 command = cmd.join(" "),
-                command_output = &cmd_output,
+                command_output = &multiversion_config_output,
                 "Failed to parse yaml from multiversion config command output",
             );
         }

src/resmoke/resmoke_proxy.rs

@@ -1,14 +1,19 @@
 """Mock of resmoke.py for testing task generation."""
 import sys
+import os
 
 
 def multiversion_config():
-    print("""
+    file_contents =  """
 last_versions:
 - last_lts
 - last_continuous
 requires_fcv_tag: requires_fcv_51,requires_fcv_52,requires_fcv_53,requires_fcv_60
-    """)
+    """
+    file_name = "multiversion-config.yml"
+    if not os.path.exists(file_name):
+        with open(file_name, "w") as file:
+            file.write(file_contents)
 
 
 def suiteconfig():