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 #![feature(panic_handler, const_fn, std_panic)]
  11
  12 use std::sync::atomic::{AtomicUsize, Ordering};
  13 use std::panic;
  14 use std::thread;
  15
  16 static A: AtomicUsize = AtomicUsize::new(0);
  17 static B: AtomicUsize = AtomicUsize::new(0);
  18
  19 fn main() {
  20     panic::set_hook(Box::new(|_| { A.fetch_add(1, Ordering::SeqCst); }));
  21     let hook = panic::take_hook();
  22     panic::set_hook(Box::new(move |info| {
  23         B.fetch_add(1, Ordering::SeqCst);
  24         hook(info);
  25     }));
  26
  27     let _ = thread::spawn(|| {
  28         panic!();
  29     }).join();
  30
  31     assert_eq!(1, A.load(Ordering::SeqCst));
  32     assert_eq!(1, B.load(Ordering::SeqCst));
  33 }