src/test/ui/generator/control-flow.rs

   1 // run-pass
   2
   3 // revisions: default nomiropt
   4 //[nomiropt]compile-flags: -Z mir-opt-level=0
   5
   6 #![feature(generators, generator_trait)]
   7
   8 use std::marker::Unpin;
   9 use std::ops::{GeneratorState, Generator};
  10 use std::pin::Pin;
  11
  12 fn finish<T>(mut amt: usize, mut t: T) -> T::Return
  13     where T: Generator<(), Yield = ()> + Unpin,
  14 {
  15     loop {
  16         match Pin::new(&mut t).resume(()) {
  17             GeneratorState::Yielded(()) => amt = amt.checked_sub(1).unwrap(),
  18             GeneratorState::Complete(ret) => {
  19                 assert_eq!(amt, 0);
  20                 return ret
  21             }
  22         }
  23     }
  24
  25 }
  26
  27 fn main() {
  28     finish(1, || yield);
  29     finish(8, || {
  30         for _ in 0..8 {
  31             yield;
  32         }
  33     });
  34     finish(1, || {
  35         if true {
  36             yield;
  37         } else {
  38         }
  39     });
  40     finish(1, || {
  41         if false {
  42         } else {
  43             yield;
  44         }
  45     });
  46     finish(2, || {
  47         if { yield; false } {
  48             yield;
  49             panic!()
  50         }
  51         yield
  52     });
  53 }