Put panic=abort test support behind -Z panic_abort_tests

Author: tmandry

src/librustc/session/config.rs

@@ -1295,6 +1295,8 @@ options! {DebuggingOptions, DebuggingSetter, basic_debugging_options,
         "show extended diagnostic help"),
     terminal_width: Option<usize> = (None, parse_opt_uint, [UNTRACKED],
         "set the current terminal width"),
+    panic_abort_tests: bool = (false, parse_bool, [TRACKED],
+        "support compiling tests with panic=abort"),
     continue_parse_after_error: bool = (false, parse_bool, [TRACKED],
         "attempt to recover from parse errors (experimental)"),
     dep_tasks: bool = (false, parse_bool, [UNTRACKED],

src/librustc_interface/passes.rs

@@ -447,6 +447,7 @@ fn configure_and_expand_inner<'a>(
             sess.diagnostic(),
             &sess.features_untracked(),
             sess.panic_strategy(),
+            sess.opts.debugging_opts.panic_abort_tests,
         )
     });
 

src/libsyntax_ext/test_harness.rs

@@ -43,6 +43,7 @@ pub fn inject(
     span_diagnostic: &errors::Handler,
     features: &Features,
     panic_strategy: PanicStrategy,
+    enable_panic_abort_tests: bool,
 ) {
     // Check for #![reexport_test_harness_main = "some_name"] which gives the
     // main test function the name `some_name` without hygiene. This needs to be
@@ -55,6 +56,10 @@ pub fn inject(
     // even in non-test builds
     let test_runner = get_test_runner(span_diagnostic, &krate);
 
+    let panic_strategy = match (panic_strategy, enable_panic_abort_tests) {
+        (PanicStrategy::Abort, true) => PanicStrategy::Abort,
+        _ => PanicStrategy::Unwind,
+    };
     if should_test {
         generate_test_harness(sess, resolver, reexport_test_harness_main,
                               krate, features, panic_strategy, test_runner)

src/test/ui/test-panic-abort-disabled.rs

@@ -0,0 +1,18 @@
+// no-prefer-dynamic
+// compile-flags: --test -Cpanic=abort
+// run-flags: --test-threads=1
+// run-fail
+// check-run-results
+
+#![cfg(test)]
+
+#[test]
+fn it_works() {
+    assert_eq!(1 + 1, 2);
+}
+
+#[test]
+#[should_panic]
+fn it_panics() {
+    assert_eq!(1 + 1, 4);
+}

src/test/ui/test-panic-abort-disabled.run.stdout

@@ -0,0 +1,3 @@
+
+running 2 tests
+test it_panics ... 

src/test/ui/test-panic-abort.rs

@@ -1,5 +1,5 @@
 // no-prefer-dynamic
-// compile-flags: --test -Cpanic=abort
+// compile-flags: --test -Cpanic=abort -Zpanic_abort_tests
 // run-flags: --test-threads=1
 // run-fail
 // check-run-results