src/libcoretest/finally.rs

   1 // Copyright 2014 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 use core::finally::{try_finally, Finally};
  12 use std::thread::Thread;
  13
  14 #[test]
  15 fn test_success() {
  16     let mut i = 0i;
  17     try_finally(
  18         &mut i, (),
  19         |i, ()| {
  20             *i = 10;
  21         },
  22         |i| {
  23             assert!(!Thread::panicking());
  24             assert_eq!(*i, 10);
  25             *i = 20;
  26         });
  27     assert_eq!(i, 20);
  28 }
  29
  30 #[test]
  31 #[should_fail]
  32 fn test_fail() {
  33     let mut i = 0i;
  34     try_finally(
  35         &mut i, (),
  36         |i, ()| {
  37             *i = 10;
  38             panic!();
  39         },
  40         |i| {
  41             assert!(Thread::panicking());
  42             assert_eq!(*i, 10);
  43         })
  44 }
  45
  46 #[test]
  47 fn test_retval() {
  48     let mut closure = |&mut:| 10i;
  49     let i = closure.finally(|| { });
  50     assert_eq!(i, 10);
  51 }
  52
  53 #[test]
  54 fn test_compact() {
  55     fn do_some_fallible_work() {}
  56     fn but_always_run_this_function() { }
  57     let mut f = do_some_fallible_work;
  58     f.finally(but_always_run_this_function);
  59 }