rollup merge of #19301: Gankro/take-fix

Was taking the value out correctly, but then not doing anything to actually fix the table. derp.
This commit is contained in:
Alex Crichton 2014-11-26 09:44:49 -08:00
commit 69e7554a47

View File

@ -1376,7 +1376,7 @@ impl<'a, K, V> OccupiedEntry<'a, K, V> {
/// Takes the value out of the entry, and returns it
pub fn take(self) -> V {
let (_, _, v) = self.elem.take();
let (_, v) = pop_internal(self.elem);
v
}
}
@ -1433,6 +1433,7 @@ mod test_map {
use hash;
use iter::{Iterator,range_inclusive,range_step_inclusive};
use cell::RefCell;
use rand::{weak_rng, Rng};
struct KindaIntLike(int);
@ -2073,4 +2074,37 @@ mod test_map {
assert_eq!(map.get(&10).unwrap(), &1000);
assert_eq!(map.len(), 6);
}
#[test]
fn test_entry_take_doesnt_corrupt() {
// Test for #19292
fn check(m: &HashMap<int, ()>) {
for k in m.keys() {
assert!(m.contains_key(k),
"{} is in keys() but not in the map?", k);
}
}
let mut m = HashMap::new();
let mut rng = weak_rng();
// Populate the map with some items.
for _ in range(0u, 50) {
let x = rng.gen_range(-10, 10);
m.insert(x, ());
}
for i in range(0u, 1000) {
let x = rng.gen_range(-10, 10);
match m.entry(x) {
Vacant(_) => {},
Occupied(e) => {
println!("{}: remove {}", i, x);
e.take();
},
}
check(&m);
}
}
}