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
  11 #![feature(tool_lints)]
  12
  13 #![warn(clippy::copy_iterator)]
  14
  15 #[derive(Copy, Clone)]
  16 struct Countdown(u8);
  17
  18 impl Iterator for Countdown {
  19     type Item = u8;
  20
  21     fn next(&mut self) -> Option<u8> {
  22         self.0.checked_sub(1).map(|c| {
  23             self.0 = c;
  24             c
  25         })
  26     }
  27 }
  28
  29 fn main() {
  30     let my_iterator = Countdown(5);
  31     let a: Vec<_> = my_iterator.take(1).collect();
  32     assert_eq!(a.len(), 1);
  33     let b: Vec<_> = my_iterator.collect();
  34     assert_eq!(b.len(), 5);
  35 }