Add build staging tests for library, miri and a bootstrap tool

Author: Kobzol

src/bootstrap/src/core/builder/tests.rs

@@ -1248,6 +1248,7 @@ mod staging {
     }
 
     #[test]
+    #[should_panic]
     fn build_compiler_stage_0() {
         insta::assert_snapshot!(get_steps("compiler", "build", Some(0)), @"");
     }
@@ -1270,6 +1271,97 @@ mod staging {
         ");
     }
 
+    #[test]
+    fn build_library_no_stage() {
+        insta::assert_snapshot!(get_steps("library", "build", None), @r"
+        [build] llvm <target1>
+        [build] rustc 0 <target1> -> rustc 1 <target1>
+        [build] rustc 1 <target1> -> std 1 <target1>
+        ");
+    }
+
+    #[test]
+    #[should_panic]
+    fn build_library_stage_0() {
+        insta::assert_snapshot!(get_steps("library", "build", Some(0)), @"");
+    }
+
+    #[test]
+    fn build_library_stage_1() {
+        insta::assert_snapshot!(get_steps("library", "build", Some(1)), @r"
+        [build] llvm <target1>
+        [build] rustc 0 <target1> -> rustc 1 <target1>
+        [build] rustc 1 <target1> -> std 1 <target1>
+        ");
+    }
+
+    #[test]
+    fn build_library_stage_2() {
+        insta::assert_snapshot!(get_steps("library", "build", Some(2)), @r"
+        [build] llvm <target1>
+        [build] rustc 0 <target1> -> rustc 1 <target1>
+        [build] rustc 1 <target1> -> std 1 <target1>
+        [build] rustc 1 <target1> -> rustc 2 <target1>
+        [build] rustc 2 <target1> -> std 2 <target1>
+        ");
+    }
+
+    #[test]
+    fn build_miri_no_stage() {
+        insta::assert_snapshot!(get_steps("miri", "build", None), @r"
+        [build] llvm <target1>
+        [build] rustc 0 <target1> -> rustc 1 <target1>
+        [build] rustc 1 <target1> -> miri 1 <target1>
+        ");
+    }
+
+    #[test]
+    #[should_panic]
+    fn build_miri_stage_0() {
+        insta::assert_snapshot!(get_steps("miri", "build", Some(0)), @"");
+    }
+
+    #[test]
+    fn build_miri_stage_1() {
+        insta::assert_snapshot!(get_steps("miri", "build", Some(1)), @r"
+        [build] llvm <target1>
+        [build] rustc 0 <target1> -> rustc 1 <target1>
+        [build] rustc 1 <target1> -> miri 1 <target1>
+        ");
+    }
+
+    #[test]
+    fn build_miri_stage_2() {
+        insta::assert_snapshot!(get_steps("miri", "build", Some(2)), @r"
+        [build] llvm <target1>
+        [build] rustc 0 <target1> -> rustc 1 <target1>
+        [build] rustc 1 <target1> -> std 1 <target1>
+        [build] rustc 1 <target1> -> rustc 2 <target1>
+        [build] rustc 2 <target1> -> miri 2 <target1>
+        ");
+    }
+
+    #[test]
+    fn build_bootstrap_tool_no_stage() {
+        insta::assert_snapshot!(get_steps("opt-dist", "build", None), @"[build] rustc 0 <target1> -> opt-dist 0 <target1>");
+    }
+
+    #[test]
+    #[should_panic]
+    fn build_bootstrap_tool_stage_0() {
+        insta::assert_snapshot!(get_steps("opt-dist", "build", Some(0)), @"");
+    }
+
+    #[test]
+    fn build_bootstrap_tool_stage_1() {
+        insta::assert_snapshot!(get_steps("opt-dist", "build", Some(1)), @"[build] rustc 0 <target1> -> opt-dist 0 <target1>");
+    }
+
+    #[test]
+    fn build_bootstrap_tool_stage_2() {
+        insta::assert_snapshot!(get_steps("opt-dist", "build", Some(2)), @"[build] rustc 0 <target1> -> opt-dist 0 <target1>");
+    }
+
     fn get_steps(path: &str, kind: &str, stage: Option<u32>) -> String {
         let mut args = vec![kind, path];
         let stage_str = stage.map(|v| v.to_string()).unwrap_or_default();
@@ -1308,6 +1400,21 @@ fn render_steps(steps: &[(Box<dyn Any>, Box<dyn Any>)]) -> String {
                     std.compiler.stage,
                     std.target
                 ))
+            } else if let Some((miri, output)) = downcast_step::<tool::Miri>(step, output) {
+                Some(format!(
+                    "[build] {} -> miri {} <{}>",
+                    render_compiler(miri.compiler),
+                    miri.compiler.stage,
+                    miri.target
+                ))
+            } else if let Some((tool, output)) = downcast_step::<tool::OptimizedDist>(step, output)
+            {
+                Some(format!(
+                    "[build] {} -> opt-dist {} <{}>",
+                    render_compiler(tool.compiler),
+                    tool.compiler.stage,
+                    tool.target
+                ))
             } else if let Some((llvm, output)) = downcast_step::<llvm::Llvm>(step, output) {
                 Some(format!("[build] llvm <{}>", llvm.target))
             } else {