1 | /* Selftests for typed-splay-tree.h. |
2 | Copyright (C) 2016-2023 Free Software Foundation, Inc. |
3 | |
4 | This file is part of GCC. |
5 | |
6 | GCC is free software; you can redistribute it and/or modify it under |
7 | the terms of the GNU General Public License as published by the Free |
8 | Software Foundation; either version 3, or (at your option) any later |
9 | version. |
10 | |
11 | GCC is distributed in the hope that it will be useful, but WITHOUT ANY |
12 | WARRANTY; without even the implied warranty of MERCHANTABILITY or |
13 | FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
14 | for more details. |
15 | |
16 | You should have received a copy of the GNU General Public License |
17 | along with GCC; see the file COPYING3. If not see |
18 | <http://www.gnu.org/licenses/>. */ |
19 | |
20 | #include "config.h" |
21 | #include "system.h" |
22 | #include "coretypes.h" |
23 | #include "typed-splay-tree.h" |
24 | #include "selftest.h" |
25 | |
26 | #if CHECKING_P |
27 | |
28 | namespace selftest { |
29 | |
30 | /* Callback for use by test_str_to_int. */ |
31 | |
32 | static int |
33 | append_cb (const char *, int value, void *user_data) |
34 | { |
35 | auto_vec <int> *vec = (auto_vec <int> *)user_data; |
36 | vec->safe_push (obj: value); |
37 | return 0; |
38 | } |
39 | |
40 | /* Test of typed_splay_tree <const char *, int>. */ |
41 | |
42 | static void |
43 | test_str_to_int () |
44 | { |
45 | typed_splay_tree <const char *, int> t (strcmp, NULL, NULL); |
46 | |
47 | t.insert (key: "a" , value: 1); |
48 | t.insert (key: "b" , value: 2); |
49 | t.insert (key: "c" , value: 3); |
50 | t.insert (key: "d" , value: 4); |
51 | |
52 | t.remove (key: "d" ); |
53 | |
54 | ASSERT_EQ (1, t.lookup ("a" )); |
55 | ASSERT_EQ (2, t.lookup ("b" )); |
56 | ASSERT_EQ (3, t.lookup ("c" )); |
57 | |
58 | ASSERT_EQ (2, t.predecessor ("c" )); |
59 | ASSERT_EQ (3, t.successor ("b" )); |
60 | ASSERT_EQ (1, t.min ()); |
61 | ASSERT_EQ (3, t.max ()); |
62 | |
63 | /* Test foreach by appending to a vec, and verifying the vec. */ |
64 | auto_vec <int> v; |
65 | t.foreach (foreach_fn: append_cb, user_data: &v); |
66 | ASSERT_EQ (3, v.length ()); |
67 | ASSERT_EQ (1, v[0]); |
68 | ASSERT_EQ (2, v[1]); |
69 | ASSERT_EQ (3, v[2]); |
70 | } |
71 | |
72 | /* Run all of the selftests within this file. */ |
73 | |
74 | void |
75 | typed_splay_tree_cc_tests () |
76 | { |
77 | test_str_to_int (); |
78 | } |
79 | |
80 | } // namespace selftest |
81 | |
82 | #endif /* #if CHECKING_P */ |
83 | |