src/test/run-pass-valgrind/cast-enum-with-dtor.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 #![allow(dead_code)]
  12
  13 // check dtor calling order when casting enums.
  14
  15 use std::sync::atomic;
  16 use std::sync::atomic::Ordering;
  17 use std::mem;
  18
  19 enum E {
  20     A = 0,
  21     B = 1,
  22     C = 2
  23 }
  24
  25 static FLAG: atomic::AtomicUsize = atomic::ATOMIC_USIZE_INIT;
  26
  27 impl Drop for E {
  28     fn drop(&mut self) {
  29         // avoid dtor loop
  30         unsafe { mem::forget(mem::replace(self, E::B)) };
  31
  32         FLAG.store(FLAG.load(Ordering::SeqCst)+1, Ordering::SeqCst);
  33     }
  34 }
  35
  36 fn main() {
  37     assert_eq!(FLAG.load(Ordering::SeqCst), 0);
  38     {
  39         let e = E::C;
  40         assert_eq!(e as u32, 2);
  41         assert_eq!(FLAG.load(Ordering::SeqCst), 0);
  42     }
  43     assert_eq!(FLAG.load(Ordering::SeqCst), 1);
  44 }