tests/ui/copy_iterator.rs

   1 // Copyright 2014-2018 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution.
   3 //
   4 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   5 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   6 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   7 // option. This file may not be copied, modified, or distributed
   8 // except according to those terms.
   9
  10 #![warn(clippy::copy_iterator)]
  11
  12 #[derive(Copy, Clone)]
  13 struct Countdown(u8);
  14
  15 impl Iterator for Countdown {
  16     type Item = u8;
  17
  18     fn next(&mut self) -> Option<u8> {
  19         self.0.checked_sub(1).map(|c| {
  20             self.0 = c;
  21             c
  22         })
  23     }
  24 }
  25
  26 fn main() {
  27     let my_iterator = Countdown(5);
  28     let a: Vec<_> = my_iterator.take(1).collect();
  29     assert_eq!(a.len(), 1);
  30     let b: Vec<_> = my_iterator.collect();
  31     assert_eq!(b.len(), 5);
  32 }