Auto merge of #1543 - RalfJung:btree, r=RalfJung

test BTreeMap::drain_filter for leaks
rust-lang · Sep 11, 2020 · 2249c10 · 2249c10
2 parents a6741b0 + bc548d3
commit 2249c10
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/run-pass/btreemap.rs b/tests/run-pass/btreemap.rs
@@ -1,4 +1,6 @@
+#![feature(btree_drain_filter)]
 use std::collections::{BTreeMap, BTreeSet};
+use std::mem;
 
 #[derive(PartialEq, Eq, PartialOrd, Ord)]
 pub enum Foo {
@@ -43,4 +45,9 @@ pub fn main() {
         b.insert(format!("key{}", i), i);
     }
     test_all_refs(&mut 13, b.values_mut());
+
+    // Test forgetting the drain.
+    let mut d = b.drain_filter(|_, i| *i < 30);
+    d.next().unwrap();
+    mem::forget(d);
 }