| // Copyright 2026 The Abseil Authors |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // https://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| // |
| // ----------------------------------------------------------------------------- |
| // File: btree_instantiate_test.cc |
| // ----------------------------------------------------------------------------- |
| // |
| // Various tests for btree containers with diverse value types and |
| // configurations. |
| // NOTE: There are more tests in btree_test.cc. |
| // The main reason to have this file separate is to improve build times. |
| |
| #include <cstdint> |
| #include <functional> |
| #include <map> |
| #include <memory> |
| #include <set> |
| #include <string> |
| #include <utility> |
| #include <vector> |
| |
| #include "gmock/gmock.h" |
| #include "gtest/gtest.h" |
| #include "absl/base/config.h" |
| #include "absl/base/internal/raw_logging.h" |
| #include "absl/container/btree_map.h" |
| #include "absl/container/btree_set.h" |
| #include "absl/container/btree_test.h" |
| #include "absl/flags/flag.h" |
| #include "absl/strings/cord.h" |
| |
| ABSL_FLAG(int, test_values, 10000, "The number of values to use for tests"); |
| |
| namespace absl { |
| ABSL_NAMESPACE_BEGIN |
| namespace container_internal { |
| |
| namespace { |
| |
| template <typename T, typename U> |
| void CheckPairEquals(const T& x, const U& y) { |
| ABSL_INTERNAL_CHECK(x == y, "Values are unequal."); |
| } |
| |
| template <typename T, typename U, typename V, typename W> |
| void CheckPairEquals(const std::pair<T, U>& x, const std::pair<V, W>& y) { |
| CheckPairEquals(x.first, y.first); |
| CheckPairEquals(x.second, y.second); |
| } |
| |
| } // namespace |
| |
| // The base class for a sorted associative container checker. TreeType is the |
| // container type to check and CheckerType is the container type to check |
| // against. TreeType is expected to be btree_{set,map,multiset,multimap} and |
| // CheckerType is expected to be {set,map,multiset,multimap}. |
| template <typename TreeType, typename CheckerType> |
| class base_checker { |
| public: |
| using key_type = typename TreeType::key_type; |
| using value_type = typename TreeType::value_type; |
| using key_compare = typename TreeType::key_compare; |
| using pointer = typename TreeType::pointer; |
| using const_pointer = typename TreeType::const_pointer; |
| using reference = typename TreeType::reference; |
| using const_reference = typename TreeType::const_reference; |
| using size_type = typename TreeType::size_type; |
| using difference_type = typename TreeType::difference_type; |
| using iterator = typename TreeType::iterator; |
| using const_iterator = typename TreeType::const_iterator; |
| using reverse_iterator = typename TreeType::reverse_iterator; |
| using const_reverse_iterator = typename TreeType::const_reverse_iterator; |
| |
| public: |
| base_checker() : const_tree_(tree_) {} |
| base_checker(const base_checker& other) |
| : tree_(other.tree_), const_tree_(tree_), checker_(other.checker_) {} |
| template <typename InputIterator> |
| base_checker(InputIterator b, InputIterator e) |
| : tree_(b, e), const_tree_(tree_), checker_(b, e) {} |
| |
| iterator begin() { return tree_.begin(); } |
| const_iterator begin() const { return tree_.begin(); } |
| iterator end() { return tree_.end(); } |
| const_iterator end() const { return tree_.end(); } |
| reverse_iterator rbegin() { return tree_.rbegin(); } |
| const_reverse_iterator rbegin() const { return tree_.rbegin(); } |
| reverse_iterator rend() { return tree_.rend(); } |
| const_reverse_iterator rend() const { return tree_.rend(); } |
| |
| template <typename IterType, typename CheckerIterType> |
| IterType iter_check(IterType tree_iter, CheckerIterType checker_iter) const { |
| if (tree_iter == tree_.end()) { |
| ABSL_INTERNAL_CHECK(checker_iter == checker_.end(), |
| "Checker iterator not at end."); |
| } else { |
| CheckPairEquals(*tree_iter, *checker_iter); |
| } |
| return tree_iter; |
| } |
| template <typename IterType, typename CheckerIterType> |
| IterType riter_check(IterType tree_iter, CheckerIterType checker_iter) const { |
| if (tree_iter == tree_.rend()) { |
| ABSL_INTERNAL_CHECK(checker_iter == checker_.rend(), |
| "Checker iterator not at rend."); |
| } else { |
| CheckPairEquals(*tree_iter, *checker_iter); |
| } |
| return tree_iter; |
| } |
| void value_check(const value_type& v) { |
| typename KeyOfValue<typename TreeType::key_type, |
| typename TreeType::value_type>::type key_of_value; |
| const key_type& key = key_of_value(v); |
| CheckPairEquals(*find(key), v); |
| lower_bound(key); |
| upper_bound(key); |
| equal_range(key); |
| contains(key); |
| count(key); |
| } |
| void erase_check(const key_type& key) { |
| EXPECT_FALSE(tree_.contains(key)); |
| EXPECT_EQ(tree_.find(key), const_tree_.end()); |
| EXPECT_FALSE(const_tree_.contains(key)); |
| EXPECT_EQ(const_tree_.find(key), tree_.end()); |
| EXPECT_EQ(tree_.equal_range(key).first, |
| const_tree_.equal_range(key).second); |
| } |
| |
| iterator lower_bound(const key_type& key) { |
| return iter_check(tree_.lower_bound(key), checker_.lower_bound(key)); |
| } |
| const_iterator lower_bound(const key_type& key) const { |
| return iter_check(tree_.lower_bound(key), checker_.lower_bound(key)); |
| } |
| iterator upper_bound(const key_type& key) { |
| return iter_check(tree_.upper_bound(key), checker_.upper_bound(key)); |
| } |
| const_iterator upper_bound(const key_type& key) const { |
| return iter_check(tree_.upper_bound(key), checker_.upper_bound(key)); |
| } |
| std::pair<iterator, iterator> equal_range(const key_type& key) { |
| std::pair<typename CheckerType::iterator, typename CheckerType::iterator> |
| checker_res = checker_.equal_range(key); |
| std::pair<iterator, iterator> tree_res = tree_.equal_range(key); |
| iter_check(tree_res.first, checker_res.first); |
| iter_check(tree_res.second, checker_res.second); |
| return tree_res; |
| } |
| std::pair<const_iterator, const_iterator> equal_range( |
| const key_type& key) const { |
| std::pair<typename CheckerType::const_iterator, |
| typename CheckerType::const_iterator> |
| checker_res = checker_.equal_range(key); |
| std::pair<const_iterator, const_iterator> tree_res = tree_.equal_range(key); |
| iter_check(tree_res.first, checker_res.first); |
| iter_check(tree_res.second, checker_res.second); |
| return tree_res; |
| } |
| iterator find(const key_type& key) { |
| return iter_check(tree_.find(key), checker_.find(key)); |
| } |
| const_iterator find(const key_type& key) const { |
| return iter_check(tree_.find(key), checker_.find(key)); |
| } |
| bool contains(const key_type& key) const { return find(key) != end(); } |
| size_type count(const key_type& key) const { |
| size_type res = checker_.count(key); |
| EXPECT_EQ(res, tree_.count(key)); |
| return res; |
| } |
| |
| base_checker& operator=(const base_checker& other) { |
| tree_ = other.tree_; |
| checker_ = other.checker_; |
| return *this; |
| } |
| |
| int erase(const key_type& key) { |
| int size = tree_.size(); |
| int res = checker_.erase(key); |
| EXPECT_EQ(res, tree_.count(key)); |
| EXPECT_EQ(res, tree_.erase(key)); |
| EXPECT_EQ(tree_.count(key), 0); |
| EXPECT_EQ(tree_.size(), size - res); |
| erase_check(key); |
| return res; |
| } |
| iterator erase(iterator iter) { |
| key_type key = iter.key(); |
| int size = tree_.size(); |
| int count = tree_.count(key); |
| auto checker_iter = checker_.lower_bound(key); |
| for (iterator tmp(tree_.lower_bound(key)); tmp != iter; ++tmp) { |
| ++checker_iter; |
| } |
| auto checker_next = checker_iter; |
| ++checker_next; |
| checker_.erase(checker_iter); |
| iter = tree_.erase(iter); |
| EXPECT_EQ(tree_.size(), checker_.size()); |
| EXPECT_EQ(tree_.size(), size - 1); |
| EXPECT_EQ(tree_.count(key), count - 1); |
| if (count == 1) { |
| erase_check(key); |
| } |
| return iter_check(iter, checker_next); |
| } |
| |
| void erase(iterator begin, iterator end) { |
| int size = tree_.size(); |
| int count = std::distance(begin, end); |
| auto checker_begin = checker_.lower_bound(begin.key()); |
| for (iterator tmp(tree_.lower_bound(begin.key())); tmp != begin; ++tmp) { |
| ++checker_begin; |
| } |
| auto checker_end = |
| end == tree_.end() ? checker_.end() : checker_.lower_bound(end.key()); |
| if (end != tree_.end()) { |
| for (iterator tmp(tree_.lower_bound(end.key())); tmp != end; ++tmp) { |
| ++checker_end; |
| } |
| } |
| const auto checker_ret = checker_.erase(checker_begin, checker_end); |
| const auto tree_ret = tree_.erase(begin, end); |
| EXPECT_EQ(std::distance(checker_.begin(), checker_ret), |
| std::distance(tree_.begin(), tree_ret)); |
| EXPECT_EQ(tree_.size(), checker_.size()); |
| EXPECT_EQ(tree_.size(), size - count); |
| } |
| |
| void clear() { |
| tree_.clear(); |
| checker_.clear(); |
| } |
| void swap(base_checker& other) { |
| tree_.swap(other.tree_); |
| checker_.swap(other.checker_); |
| } |
| |
| void verify() const { |
| tree_.verify(); |
| EXPECT_EQ(tree_.size(), checker_.size()); |
| |
| // Move through the forward iterators using increment. |
| auto checker_iter = checker_.begin(); |
| const_iterator tree_iter(tree_.begin()); |
| for (; tree_iter != tree_.end(); ++tree_iter, ++checker_iter) { |
| CheckPairEquals(*tree_iter, *checker_iter); |
| } |
| |
| // Move through the forward iterators using decrement. |
| for (int n = tree_.size() - 1; n >= 0; --n) { |
| iter_check(tree_iter, checker_iter); |
| --tree_iter; |
| --checker_iter; |
| } |
| EXPECT_EQ(tree_iter, tree_.begin()); |
| EXPECT_EQ(checker_iter, checker_.begin()); |
| |
| // Move through the reverse iterators using increment. |
| auto checker_riter = checker_.rbegin(); |
| const_reverse_iterator tree_riter(tree_.rbegin()); |
| for (; tree_riter != tree_.rend(); ++tree_riter, ++checker_riter) { |
| CheckPairEquals(*tree_riter, *checker_riter); |
| } |
| |
| // Move through the reverse iterators using decrement. |
| for (int n = tree_.size() - 1; n >= 0; --n) { |
| riter_check(tree_riter, checker_riter); |
| --tree_riter; |
| --checker_riter; |
| } |
| EXPECT_EQ(tree_riter, tree_.rbegin()); |
| EXPECT_EQ(checker_riter, checker_.rbegin()); |
| } |
| |
| const TreeType& tree() const { return tree_; } |
| |
| size_type size() const { |
| EXPECT_EQ(tree_.size(), checker_.size()); |
| return tree_.size(); |
| } |
| size_type max_size() const { return tree_.max_size(); } |
| bool empty() const { |
| EXPECT_EQ(tree_.empty(), checker_.empty()); |
| return tree_.empty(); |
| } |
| |
| protected: |
| TreeType tree_; |
| const TreeType& const_tree_; |
| CheckerType checker_; |
| }; |
| |
| namespace { |
| |
| // A checker for unique sorted associative containers. TreeType is expected to |
| // be btree_{set,map} and CheckerType is expected to be {set,map}. |
| template <typename TreeType, typename CheckerType> |
| class unique_checker : public base_checker<TreeType, CheckerType> { |
| using super_type = base_checker<TreeType, CheckerType>; |
| |
| public: |
| using iterator = typename super_type::iterator; |
| using value_type = typename super_type::value_type; |
| |
| public: |
| unique_checker() : super_type() {} |
| unique_checker(const unique_checker& other) : super_type(other) {} |
| template <class InputIterator> |
| unique_checker(InputIterator b, InputIterator e) : super_type(b, e) {} |
| unique_checker& operator=(const unique_checker&) = default; |
| |
| // Insertion routines. |
| std::pair<iterator, bool> insert(const value_type& v) { |
| int size = this->tree_.size(); |
| std::pair<typename CheckerType::iterator, bool> checker_res = |
| this->checker_.insert(v); |
| std::pair<iterator, bool> tree_res = this->tree_.insert(v); |
| CheckPairEquals(*tree_res.first, *checker_res.first); |
| EXPECT_EQ(tree_res.second, checker_res.second); |
| EXPECT_EQ(this->tree_.size(), this->checker_.size()); |
| EXPECT_EQ(this->tree_.size(), size + tree_res.second); |
| return tree_res; |
| } |
| iterator insert(iterator position, const value_type& v) { |
| int size = this->tree_.size(); |
| std::pair<typename CheckerType::iterator, bool> checker_res = |
| this->checker_.insert(v); |
| iterator tree_res = this->tree_.insert(position, v); |
| CheckPairEquals(*tree_res, *checker_res.first); |
| EXPECT_EQ(this->tree_.size(), this->checker_.size()); |
| EXPECT_EQ(this->tree_.size(), size + checker_res.second); |
| return tree_res; |
| } |
| template <typename InputIterator> |
| void insert(InputIterator b, InputIterator e) { |
| for (; b != e; ++b) { |
| insert(*b); |
| } |
| } |
| }; |
| |
| // A checker for multiple sorted associative containers. TreeType is expected |
| // to be btree_{multiset,multimap} and CheckerType is expected to be |
| // {multiset,multimap}. |
| template <typename TreeType, typename CheckerType> |
| class multi_checker : public base_checker<TreeType, CheckerType> { |
| using super_type = base_checker<TreeType, CheckerType>; |
| |
| public: |
| using iterator = typename super_type::iterator; |
| using value_type = typename super_type::value_type; |
| |
| public: |
| multi_checker() : super_type() {} |
| multi_checker(const multi_checker& other) : super_type(other) {} |
| template <class InputIterator> |
| multi_checker(InputIterator b, InputIterator e) : super_type(b, e) {} |
| multi_checker& operator=(const multi_checker&) = default; |
| |
| // Insertion routines. |
| iterator insert(const value_type& v) { |
| int size = this->tree_.size(); |
| auto checker_res = this->checker_.insert(v); |
| iterator tree_res = this->tree_.insert(v); |
| CheckPairEquals(*tree_res, *checker_res); |
| EXPECT_EQ(this->tree_.size(), this->checker_.size()); |
| EXPECT_EQ(this->tree_.size(), size + 1); |
| return tree_res; |
| } |
| iterator insert(iterator position, const value_type& v) { |
| int size = this->tree_.size(); |
| auto checker_res = this->checker_.insert(v); |
| iterator tree_res = this->tree_.insert(position, v); |
| CheckPairEquals(*tree_res, *checker_res); |
| EXPECT_EQ(this->tree_.size(), this->checker_.size()); |
| EXPECT_EQ(this->tree_.size(), size + 1); |
| return tree_res; |
| } |
| template <typename InputIterator> |
| void insert(InputIterator b, InputIterator e) { |
| for (; b != e; ++b) { |
| insert(*b); |
| } |
| } |
| }; |
| |
| template <typename T, typename V> |
| void DoTest(const char* name, T* b, const std::vector<V>& values) { |
| typename KeyOfValue<typename T::key_type, V>::type key_of_value; |
| |
| T& mutable_b = *b; |
| const T& const_b = *b; |
| |
| // Test insert. |
| for (int i = 0; i < values.size(); ++i) { |
| mutable_b.insert(values[i]); |
| mutable_b.value_check(values[i]); |
| } |
| ASSERT_EQ(mutable_b.size(), values.size()); |
| |
| const_b.verify(); |
| |
| // Test copy constructor. |
| T b_copy(const_b); |
| EXPECT_EQ(b_copy.size(), const_b.size()); |
| for (int i = 0; i < values.size(); ++i) { |
| CheckPairEquals(*b_copy.find(key_of_value(values[i])), values[i]); |
| } |
| |
| // Test range constructor. |
| T b_range(const_b.begin(), const_b.end()); |
| EXPECT_EQ(b_range.size(), const_b.size()); |
| for (int i = 0; i < values.size(); ++i) { |
| CheckPairEquals(*b_range.find(key_of_value(values[i])), values[i]); |
| } |
| |
| // Test range insertion for values that already exist. |
| b_range.insert(b_copy.begin(), b_copy.end()); |
| b_range.verify(); |
| |
| // Test range insertion for new values. |
| b_range.clear(); |
| b_range.insert(b_copy.begin(), b_copy.end()); |
| EXPECT_EQ(b_range.size(), b_copy.size()); |
| for (int i = 0; i < values.size(); ++i) { |
| CheckPairEquals(*b_range.find(key_of_value(values[i])), values[i]); |
| } |
| |
| // Test assignment to self. Nothing should change. |
| b_range.operator=(b_range); |
| EXPECT_EQ(b_range.size(), b_copy.size()); |
| |
| // Test assignment of new values. |
| b_range.clear(); |
| b_range = b_copy; |
| EXPECT_EQ(b_range.size(), b_copy.size()); |
| |
| // Test swap. |
| b_range.clear(); |
| b_range.swap(b_copy); |
| EXPECT_EQ(b_copy.size(), 0); |
| EXPECT_EQ(b_range.size(), const_b.size()); |
| for (int i = 0; i < values.size(); ++i) { |
| CheckPairEquals(*b_range.find(key_of_value(values[i])), values[i]); |
| } |
| b_range.swap(b_copy); |
| |
| // Test non-member function swap. |
| swap(b_range, b_copy); |
| EXPECT_EQ(b_copy.size(), 0); |
| EXPECT_EQ(b_range.size(), const_b.size()); |
| for (int i = 0; i < values.size(); ++i) { |
| CheckPairEquals(*b_range.find(key_of_value(values[i])), values[i]); |
| } |
| swap(b_range, b_copy); |
| |
| // Test erase via values. |
| for (int i = 0; i < values.size(); ++i) { |
| mutable_b.erase(key_of_value(values[i])); |
| // Erasing a non-existent key should have no effect. |
| ASSERT_EQ(mutable_b.erase(key_of_value(values[i])), 0); |
| } |
| |
| const_b.verify(); |
| EXPECT_EQ(const_b.size(), 0); |
| |
| // Test erase via iterators. |
| mutable_b = b_copy; |
| for (int i = 0; i < values.size(); ++i) { |
| mutable_b.erase(mutable_b.find(key_of_value(values[i]))); |
| } |
| |
| const_b.verify(); |
| EXPECT_EQ(const_b.size(), 0); |
| |
| // Test insert with hint. |
| for (int i = 0; i < values.size(); i++) { |
| mutable_b.insert(mutable_b.upper_bound(key_of_value(values[i])), values[i]); |
| } |
| |
| const_b.verify(); |
| |
| // Test range erase. |
| mutable_b.erase(mutable_b.begin(), mutable_b.end()); |
| EXPECT_EQ(mutable_b.size(), 0); |
| const_b.verify(); |
| |
| // First half. |
| mutable_b = b_copy; |
| typename T::iterator mutable_iter_end = mutable_b.begin(); |
| for (int i = 0; i < values.size() / 2; ++i) ++mutable_iter_end; |
| mutable_b.erase(mutable_b.begin(), mutable_iter_end); |
| EXPECT_EQ(mutable_b.size(), values.size() - values.size() / 2); |
| const_b.verify(); |
| |
| // Second half. |
| mutable_b = b_copy; |
| typename T::iterator mutable_iter_begin = mutable_b.begin(); |
| for (int i = 0; i < values.size() / 2; ++i) ++mutable_iter_begin; |
| mutable_b.erase(mutable_iter_begin, mutable_b.end()); |
| EXPECT_EQ(mutable_b.size(), values.size() / 2); |
| const_b.verify(); |
| |
| // Second quarter. |
| mutable_b = b_copy; |
| mutable_iter_begin = mutable_b.begin(); |
| for (int i = 0; i < values.size() / 4; ++i) ++mutable_iter_begin; |
| mutable_iter_end = mutable_iter_begin; |
| for (int i = 0; i < values.size() / 4; ++i) ++mutable_iter_end; |
| mutable_b.erase(mutable_iter_begin, mutable_iter_end); |
| EXPECT_EQ(mutable_b.size(), values.size() - values.size() / 4); |
| const_b.verify(); |
| |
| mutable_b.clear(); |
| } |
| |
| template <typename T> |
| void ConstTest() { |
| using value_type = typename T::value_type; |
| typename KeyOfValue<typename T::key_type, value_type>::type key_of_value; |
| |
| T mutable_b; |
| const T& const_b = mutable_b; |
| |
| // Insert a single value into the container and test looking it up. |
| value_type value = Generator<value_type>(2)(2); |
| mutable_b.insert(value); |
| EXPECT_TRUE(mutable_b.contains(key_of_value(value))); |
| EXPECT_NE(mutable_b.find(key_of_value(value)), const_b.end()); |
| EXPECT_TRUE(const_b.contains(key_of_value(value))); |
| EXPECT_NE(const_b.find(key_of_value(value)), mutable_b.end()); |
| EXPECT_EQ(*const_b.lower_bound(key_of_value(value)), value); |
| EXPECT_EQ(const_b.upper_bound(key_of_value(value)), const_b.end()); |
| EXPECT_EQ(*const_b.equal_range(key_of_value(value)).first, value); |
| |
| // We can only create a non-const iterator from a non-const container. |
| typename T::iterator mutable_iter(mutable_b.begin()); |
| EXPECT_EQ(mutable_iter, const_b.begin()); |
| EXPECT_NE(mutable_iter, const_b.end()); |
| EXPECT_EQ(const_b.begin(), mutable_iter); |
| EXPECT_NE(const_b.end(), mutable_iter); |
| typename T::reverse_iterator mutable_riter(mutable_b.rbegin()); |
| EXPECT_EQ(mutable_riter, const_b.rbegin()); |
| EXPECT_NE(mutable_riter, const_b.rend()); |
| EXPECT_EQ(const_b.rbegin(), mutable_riter); |
| EXPECT_NE(const_b.rend(), mutable_riter); |
| |
| // We can create a const iterator from a non-const iterator. |
| typename T::const_iterator const_iter(mutable_iter); |
| EXPECT_EQ(const_iter, mutable_b.begin()); |
| EXPECT_NE(const_iter, mutable_b.end()); |
| EXPECT_EQ(mutable_b.begin(), const_iter); |
| EXPECT_NE(mutable_b.end(), const_iter); |
| typename T::const_reverse_iterator const_riter(mutable_riter); |
| EXPECT_EQ(const_riter, mutable_b.rbegin()); |
| EXPECT_NE(const_riter, mutable_b.rend()); |
| EXPECT_EQ(mutable_b.rbegin(), const_riter); |
| EXPECT_NE(mutable_b.rend(), const_riter); |
| |
| // Make sure various methods can be invoked on a const container. |
| const_b.verify(); |
| ASSERT_TRUE(!const_b.empty()); |
| EXPECT_EQ(const_b.size(), 1); |
| EXPECT_GT(const_b.max_size(), 0); |
| EXPECT_TRUE(const_b.contains(key_of_value(value))); |
| EXPECT_EQ(const_b.count(key_of_value(value)), 1); |
| } |
| |
| template <typename T, typename C> |
| void BtreeTest() { |
| ConstTest<T>(); |
| |
| using V = typename remove_pair_const<typename T::value_type>::type; |
| const std::vector<V> random_values = GenerateValuesWithSeed<V>( |
| absl::GetFlag(FLAGS_test_values), 4 * absl::GetFlag(FLAGS_test_values), |
| GTEST_FLAG_GET(random_seed)); |
| |
| unique_checker<T, C> container; |
| |
| // Test key insertion/deletion in sorted order. |
| std::vector<V> sorted_values(random_values); |
| std::sort(sorted_values.begin(), sorted_values.end()); |
| DoTest("sorted: ", &container, sorted_values); |
| |
| // Test key insertion/deletion in reverse sorted order. |
| std::reverse(sorted_values.begin(), sorted_values.end()); |
| DoTest("rsorted: ", &container, sorted_values); |
| |
| // Test key insertion/deletion in random order. |
| DoTest("random: ", &container, random_values); |
| } |
| |
| template <typename T, typename C> |
| void BtreeMultiTest() { |
| ConstTest<T>(); |
| |
| using V = typename remove_pair_const<typename T::value_type>::type; |
| const std::vector<V> random_values = GenerateValuesWithSeed<V>( |
| absl::GetFlag(FLAGS_test_values), 4 * absl::GetFlag(FLAGS_test_values), |
| GTEST_FLAG_GET(random_seed)); |
| |
| multi_checker<T, C> container; |
| |
| // Test keys in sorted order. |
| std::vector<V> sorted_values(random_values); |
| std::sort(sorted_values.begin(), sorted_values.end()); |
| DoTest("sorted: ", &container, sorted_values); |
| |
| // Test keys in reverse sorted order. |
| std::reverse(sorted_values.begin(), sorted_values.end()); |
| DoTest("rsorted: ", &container, sorted_values); |
| |
| // Test keys in random order. |
| DoTest("random: ", &container, random_values); |
| |
| // Test keys in random order w/ duplicates. |
| std::vector<V> duplicate_values(random_values); |
| duplicate_values.insert(duplicate_values.end(), random_values.begin(), |
| random_values.end()); |
| DoTest("duplicates:", &container, duplicate_values); |
| |
| // Test all identical keys. |
| std::vector<V> identical_values(100); |
| std::fill(identical_values.begin(), identical_values.end(), |
| Generator<V>(2)(2)); |
| DoTest("identical: ", &container, identical_values); |
| } |
| |
| template <typename T> |
| void BtreeMapTest() { |
| using value_type = typename T::value_type; |
| using mapped_type = typename T::mapped_type; |
| |
| mapped_type m = Generator<mapped_type>(0)(0); |
| (void)m; |
| |
| T b; |
| |
| // Verify we can insert using operator[]. |
| for (int i = 0; i < 1000; i++) { |
| value_type v = Generator<value_type>(1000)(i); |
| b[v.first] = v.second; |
| } |
| EXPECT_EQ(b.size(), 1000); |
| |
| // Test whether we can use the "->" operator on iterators and |
| // reverse_iterators. This stresses the btree_map_params::pair_pointer |
| // mechanism. |
| EXPECT_EQ(b.begin()->first, Generator<value_type>(1000)(0).first); |
| EXPECT_EQ(b.begin()->second, Generator<value_type>(1000)(0).second); |
| EXPECT_EQ(b.rbegin()->first, Generator<value_type>(1000)(999).first); |
| EXPECT_EQ(b.rbegin()->second, Generator<value_type>(1000)(999).second); |
| } |
| |
| template <typename T> |
| void BtreeMultiMapTest() { |
| using mapped_type = typename T::mapped_type; |
| mapped_type m = Generator<mapped_type>(0)(0); |
| (void)m; |
| } |
| |
| template <typename K, int N = 256> |
| void SetTest() { |
| EXPECT_EQ(sizeof(absl::btree_set<K>), |
| 2 * sizeof(void*) + sizeof(typename absl::btree_set<K>::size_type)); |
| using BtreeSet = absl::btree_set<K>; |
| BtreeTest<BtreeSet, std::set<K>>(); |
| } |
| |
| template <typename K, int N = 256> |
| void MapTest() { |
| EXPECT_EQ( |
| sizeof(absl::btree_map<K, K>), |
| 2 * sizeof(void*) + sizeof(typename absl::btree_map<K, K>::size_type)); |
| using BtreeMap = absl::btree_map<K, K>; |
| BtreeTest<BtreeMap, std::map<K, K>>(); |
| BtreeMapTest<BtreeMap>(); |
| } |
| |
| TEST(Btree, set_int32) { SetTest<int32_t>(); } |
| TEST(Btree, set_string) { SetTest<std::string>(); } |
| TEST(Btree, set_cord) { SetTest<absl::Cord>(); } |
| TEST(Btree, map_int32) { MapTest<int32_t>(); } |
| TEST(Btree, map_string) { MapTest<std::string>(); } |
| TEST(Btree, map_cord) { MapTest<absl::Cord>(); } |
| |
| template <typename K, int N = 256> |
| void MultiSetTest() { |
| EXPECT_EQ( |
| sizeof(absl::btree_multiset<K>), |
| 2 * sizeof(void*) + sizeof(typename absl::btree_multiset<K>::size_type)); |
| using BtreeMSet = absl::btree_multiset<K>; |
| BtreeMultiTest<BtreeMSet, std::multiset<K>>(); |
| } |
| |
| template <typename K, int N = 256> |
| void MultiMapTest() { |
| EXPECT_EQ(sizeof(absl::btree_multimap<K, K>), |
| 2 * sizeof(void*) + |
| sizeof(typename absl::btree_multimap<K, K>::size_type)); |
| using BtreeMMap = absl::btree_multimap<K, K>; |
| BtreeMultiTest<BtreeMMap, std::multimap<K, K>>(); |
| BtreeMultiMapTest<BtreeMMap>(); |
| } |
| |
| TEST(Btree, multiset_int32) { MultiSetTest<int32_t>(); } |
| TEST(Btree, multiset_string) { MultiSetTest<std::string>(); } |
| TEST(Btree, multiset_cord) { MultiSetTest<absl::Cord>(); } |
| TEST(Btree, multimap_int32) { MultiMapTest<int32_t>(); } |
| TEST(Btree, multimap_string) { MultiMapTest<std::string>(); } |
| TEST(Btree, multimap_cord) { MultiMapTest<absl::Cord>(); } |
| |
| } // namespace |
| } // namespace container_internal |
| ABSL_NAMESPACE_END |
| } // namespace absl |