src/test/run-pass/panic-handler-chain.rs

   1 // Copyright 2015 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 // ignore-emscripten no threads support
  12
  13 #![feature(panic_handler, const_fn, std_panic)]
  14
  15 use std::sync::atomic::{AtomicUsize, Ordering};
  16 use std::panic;
  17 use std::thread;
  18
  19 static A: AtomicUsize = AtomicUsize::new(0);
  20 static B: AtomicUsize = AtomicUsize::new(0);
  21
  22 fn main() {
  23     panic::set_hook(Box::new(|_| { A.fetch_add(1, Ordering::SeqCst); }));
  24     let hook = panic::take_hook();
  25     panic::set_hook(Box::new(move |info| {
  26         B.fetch_add(1, Ordering::SeqCst);
  27         hook(info);
  28     }));
  29
  30     let _ = thread::spawn(|| {
  31         panic!();
  32     }).join();
  33
  34     assert_eq!(1, A.load(Ordering::SeqCst));
  35     assert_eq!(1, B.load(Ordering::SeqCst));
  36 }