From 3f2f4a35ed89ed02b7192bf3b35b02a4654bb2f0 Mon Sep 17 00:00:00 2001 From: Keita Nonaka Date: Fri, 15 Apr 2022 01:12:00 -0700 Subject: test: add try_insert() test cases for BTreeSet --- library/alloc/src/collections/btree/map/tests.rs | 15 +++++++++++++++ 1 file changed, 15 insertions(+) (limited to 'library/alloc') diff --git a/library/alloc/src/collections/btree/map/tests.rs b/library/alloc/src/collections/btree/map/tests.rs index d14f2cd034e..b9db98ffae9 100644 --- a/library/alloc/src/collections/btree/map/tests.rs +++ b/library/alloc/src/collections/btree/map/tests.rs @@ -1996,6 +1996,21 @@ fn test_insert_into_full_height_1() { } } +#[test] +fn test_try_insert() { + let mut map = BTreeMap::new(); + + assert!(map.is_empty()); + + assert_eq!(map.try_insert(1, 10).unwrap(), &10); + assert_eq!(map.try_insert(2, 20).unwrap(), &20); + + let err = map.try_insert(2, 200).unwrap_err(); + assert_eq!(err.entry.key(), &2); + assert_eq!(err.entry.get(), &20); + assert_eq!(err.value, 200); +} + macro_rules! create_append_test { ($name:ident, $len:expr) => { #[test] -- cgit 1.4.1-3-g733a5