From bc548d30041a3ac247d264d6b49904b25f232f7a Mon Sep 17 00:00:00 2001
From: Ralf Jung <post@ralfj.de>
Date: Fri, 11 Sep 2020 12:20:08 +0200
Subject: [PATCH] test BTreeMap::drain_filter for leaks

---
 tests/run-pass/btreemap.rs | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/run-pass/btreemap.rs b/tests/run-pass/btreemap.rs
index 603674cc450..e639ba6225c 100644
--- 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);
 }
-- 
2.44.0