151 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			151 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			C++
		
	
	
	
| //===----------------------------------------------------------------------===//
 | |
| //
 | |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 | |
| // See https://llvm.org/LICENSE.txt for license information.
 | |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| 
 | |
| // UNSUPPORTED: c++03, c++11, c++14
 | |
| 
 | |
| // <map>
 | |
| 
 | |
| // class map
 | |
| 
 | |
| // template <class C2>
 | |
| //   void merge(map<key_type, value_type, C2, allocator_type>& source);
 | |
| // template <class C2>
 | |
| //   void merge(map<key_type, value_type, C2, allocator_type>&& source);
 | |
| // template <class C2>
 | |
| //   void merge(multimap<key_type, value_type, C2, allocator_type>& source);
 | |
| // template <class C2>
 | |
| //   void merge(multimap<key_type, value_type, C2, allocator_type>&& source);
 | |
| 
 | |
| #include <map>
 | |
| #include <cassert>
 | |
| #include "test_macros.h"
 | |
| #include "Counter.h"
 | |
| 
 | |
| template <class Map>
 | |
| bool map_equal(const Map& map, Map other)
 | |
| {
 | |
|     return map == other;
 | |
| }
 | |
| 
 | |
| #ifndef TEST_HAS_NO_EXCEPTIONS
 | |
| struct throw_comparator
 | |
| {
 | |
|     bool& should_throw_;
 | |
| 
 | |
|     throw_comparator(bool& should_throw) : should_throw_(should_throw) {}
 | |
| 
 | |
|     template <class T>
 | |
|     bool operator()(const T& lhs, const T& rhs) const
 | |
|     {
 | |
|         if (should_throw_)
 | |
|             throw 0;
 | |
|         return lhs < rhs;
 | |
|     }
 | |
| };
 | |
| #endif
 | |
| 
 | |
| int main(int, char**)
 | |
| {
 | |
|     {
 | |
|         std::map<int, int> src{{1, 0}, {3, 0}, {5, 0}};
 | |
|         std::map<int, int> dst{{2, 0}, {4, 0}, {5, 0}};
 | |
|         dst.merge(src);
 | |
|         assert(map_equal(src, {{5,0}}));
 | |
|         assert(map_equal(dst, {{1, 0}, {2, 0}, {3, 0}, {4, 0}, {5, 0}}));
 | |
|     }
 | |
| 
 | |
| #ifndef TEST_HAS_NO_EXCEPTIONS
 | |
|     {
 | |
|         bool do_throw = false;
 | |
|         typedef std::map<Counter<int>, int, throw_comparator> map_type;
 | |
|         map_type src({{1, 0}, {3, 0}, {5, 0}}, throw_comparator(do_throw));
 | |
|         map_type dst({{2, 0}, {4, 0}, {5, 0}}, throw_comparator(do_throw));
 | |
| 
 | |
|         assert(Counter_base::gConstructed == 6);
 | |
| 
 | |
|         do_throw = true;
 | |
|         try
 | |
|         {
 | |
|             dst.merge(src);
 | |
|         }
 | |
|         catch (int)
 | |
|         {
 | |
|             do_throw = false;
 | |
|         }
 | |
|         assert(!do_throw);
 | |
|         assert(map_equal(src, map_type({{1, 0}, {3, 0}, {5, 0}}, throw_comparator(do_throw))));
 | |
|         assert(map_equal(dst, map_type({{2, 0}, {4, 0}, {5, 0}}, throw_comparator(do_throw))));
 | |
|     }
 | |
| #endif
 | |
|     assert(Counter_base::gConstructed == 0);
 | |
|     struct comparator
 | |
|     {
 | |
|         comparator() = default;
 | |
| 
 | |
|         bool operator()(const Counter<int>& lhs, const Counter<int>& rhs) const
 | |
|         {
 | |
|             return lhs < rhs;
 | |
|         }
 | |
|     };
 | |
|     {
 | |
|         typedef std::map<Counter<int>, int, std::less<Counter<int>>> first_map_type;
 | |
|         typedef std::map<Counter<int>, int, comparator> second_map_type;
 | |
|         typedef std::multimap<Counter<int>, int, comparator> third_map_type;
 | |
| 
 | |
|         {
 | |
|             first_map_type first{{1, 0}, {2, 0}, {3, 0}};
 | |
|             second_map_type second{{2, 0}, {3, 0}, {4, 0}};
 | |
|             third_map_type third{{1, 0}, {3, 0}};
 | |
| 
 | |
|             assert(Counter_base::gConstructed == 8);
 | |
| 
 | |
|             first.merge(second);
 | |
|             first.merge(third);
 | |
| 
 | |
|             assert(map_equal(first, {{1, 0}, {2, 0}, {3, 0}, {4, 0}}));
 | |
|             assert(map_equal(second, {{2, 0}, {3, 0}}));
 | |
|             assert(map_equal(third, {{1, 0}, {3, 0}}));
 | |
| 
 | |
|             assert(Counter_base::gConstructed == 8);
 | |
|         }
 | |
|         assert(Counter_base::gConstructed == 0);
 | |
|         {
 | |
|             first_map_type first{{1, 0}, {2, 0}, {3, 0}};
 | |
|             second_map_type second{{2, 0}, {3, 0}, {4, 0}};
 | |
|             third_map_type third{{1, 0}, {3, 0}};
 | |
| 
 | |
|             assert(Counter_base::gConstructed == 8);
 | |
| 
 | |
|             first.merge(std::move(second));
 | |
|             first.merge(std::move(third));
 | |
| 
 | |
|             assert(map_equal(first, {{1, 0}, {2, 0}, {3, 0}, {4, 0}}));
 | |
|             assert(map_equal(second, {{2, 0}, {3, 0}}));
 | |
|             assert(map_equal(third, {{1, 0}, {3, 0}}));
 | |
| 
 | |
|             assert(Counter_base::gConstructed == 8);
 | |
|         }
 | |
|         assert(Counter_base::gConstructed == 0);
 | |
|     }
 | |
|     assert(Counter_base::gConstructed == 0);
 | |
|     {
 | |
|         std::map<int, int> first;
 | |
|         {
 | |
|             std::map<int, int> second;
 | |
|             first.merge(second);
 | |
|             first.merge(std::move(second));
 | |
|         }
 | |
|         {
 | |
|             std::multimap<int, int> second;
 | |
|             first.merge(second);
 | |
|             first.merge(std::move(second));
 | |
|         }
 | |
|     }
 | |
|     return 0;
 | |
| }
 |