src/tools/clippy/tests/ui/get_unwrap.fixed

   1 // run-rustfix
   2
   3 #![allow(unused_mut, clippy::from_iter_instead_of_collect, clippy::get_first)]
   4 #![warn(clippy::unwrap_used)]
   5 #![deny(clippy::get_unwrap)]
   6
   7 use std::collections::BTreeMap;
   8 use std::collections::HashMap;
   9 use std::collections::VecDeque;
  10
  11 struct GetFalsePositive {
  12     arr: [u32; 3],
  13 }
  14
  15 impl GetFalsePositive {
  16     fn get(&self, pos: usize) -> Option<&u32> {
  17         self.arr.get(pos)
  18     }
  19     fn get_mut(&mut self, pos: usize) -> Option<&mut u32> {
  20         self.arr.get_mut(pos)
  21     }
  22 }
  23
  24 fn main() {
  25     let mut boxed_slice: Box<[u8]> = Box::new([0, 1, 2, 3]);
  26     let mut some_slice = &mut [0, 1, 2, 3];
  27     let mut some_vec = vec![0, 1, 2, 3];
  28     let mut some_vecdeque: VecDeque<_> = some_vec.iter().cloned().collect();
  29     let mut some_hashmap: HashMap<u8, char> = HashMap::from_iter(vec![(1, 'a'), (2, 'b')]);
  30     let mut some_btreemap: BTreeMap<u8, char> = BTreeMap::from_iter(vec![(1, 'a'), (2, 'b')]);
  31     let mut false_positive = GetFalsePositive { arr: [0, 1, 2] };
  32
  33     {
  34         // Test `get().unwrap()`
  35         let _ = &boxed_slice[1];
  36         let _ = &some_slice[0];
  37         let _ = &some_vec[0];
  38         let _ = &some_vecdeque[0];
  39         let _ = &some_hashmap[&1];
  40         let _ = &some_btreemap[&1];
  41         #[allow(clippy::unwrap_used)]
  42         let _ = false_positive.get(0).unwrap();
  43         // Test with deref
  44         let _: u8 = boxed_slice[1];
  45     }
  46
  47     {
  48         // Test `get_mut().unwrap()`
  49         boxed_slice[0] = 1;
  50         some_slice[0] = 1;
  51         some_vec[0] = 1;
  52         some_vecdeque[0] = 1;
  53         // Check false positives
  54         #[allow(clippy::unwrap_used)]
  55         {
  56             *some_hashmap.get_mut(&1).unwrap() = 'b';
  57             *some_btreemap.get_mut(&1).unwrap() = 'b';
  58             *false_positive.get_mut(0).unwrap() = 1;
  59         }
  60     }
  61
  62     {
  63         // Test `get().unwrap().foo()` and `get_mut().unwrap().bar()`
  64         let _ = some_vec[0..1].to_vec();
  65         let _ = some_vec[0..1].to_vec();
  66     }
  67 }