src/test/ui/threads-sendsync/send_str_treemap.rs

   1 // run-pass
   2 use std::collections::BTreeMap;
   3 use std::borrow::Cow;
   4
   5 use std::borrow::Cow::{Owned as O, Borrowed as B};
   6
   7 type SendStr = Cow<'static, str>;
   8
   9 fn main() {
  10     let mut map: BTreeMap<SendStr, usize> = BTreeMap::new();
  11     assert!(map.insert(B("foo"), 42).is_none());
  12     assert!(map.insert(O("foo".to_string()), 42).is_some());
  13     assert!(map.insert(B("foo"), 42).is_some());
  14     assert!(map.insert(O("foo".to_string()), 42).is_some());
  15
  16     assert!(map.insert(B("foo"), 43).is_some());
  17     assert!(map.insert(O("foo".to_string()), 44).is_some());
  18     assert!(map.insert(B("foo"), 45).is_some());
  19     assert!(map.insert(O("foo".to_string()), 46).is_some());
  20
  21     let v = 46;
  22
  23     assert_eq!(map.get(&O("foo".to_string())), Some(&v));
  24     assert_eq!(map.get(&B("foo")), Some(&v));
  25
  26     let (a, b, c, d) = (50, 51, 52, 53);
  27
  28     assert!(map.insert(B("abc"), a).is_none());
  29     assert!(map.insert(O("bcd".to_string()), b).is_none());
  30     assert!(map.insert(B("cde"), c).is_none());
  31     assert!(map.insert(O("def".to_string()), d).is_none());
  32
  33     assert!(map.insert(B("abc"), a).is_some());
  34     assert!(map.insert(O("bcd".to_string()), b).is_some());
  35     assert!(map.insert(B("cde"), c).is_some());
  36     assert!(map.insert(O("def".to_string()), d).is_some());
  37
  38     assert!(map.insert(O("abc".to_string()), a).is_some());
  39     assert!(map.insert(B("bcd"), b).is_some());
  40     assert!(map.insert(O("cde".to_string()), c).is_some());
  41     assert!(map.insert(B("def"), d).is_some());
  42
  43     assert_eq!(map.get(&B("abc")), Some(&a));
  44     assert_eq!(map.get(&B("bcd")), Some(&b));
  45     assert_eq!(map.get(&B("cde")), Some(&c));
  46     assert_eq!(map.get(&B("def")), Some(&d));
  47
  48     assert_eq!(map.get(&O("abc".to_string())), Some(&a));
  49     assert_eq!(map.get(&O("bcd".to_string())), Some(&b));
  50     assert_eq!(map.get(&O("cde".to_string())), Some(&c));
  51     assert_eq!(map.get(&O("def".to_string())), Some(&d));
  52
  53     assert!(map.remove(&B("foo")).is_some());
  54     assert_eq!(map.into_iter().map(|(k, v)| format!("{}{}", k, v))
  55                               .collect::<Vec<String>>()
  56                               .concat(),
  57                "abc50bcd51cde52def53".to_string());
  58 }