libstdc++-v3/testsuite/23_containers/unordered_set/allocator/swap.cc

   1 // Copyright (C) 2013-2017 Free Software Foundation, Inc.
   2 //
   3 // This file is part of the GNU ISO C++ Library.  This library is free
   4 // software; you can redistribute it and/or modify it under the
   5 // terms of the GNU General Public License as published by the
   6 // Free Software Foundation; either version 3, or (at your option)
   7 // any later version.
   8
   9 // This library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13
  14 // You should have received a copy of the GNU General Public License along
  15 // with this library; see the file COPYING3.  If not see
  16 // <http://www.gnu.org/licenses/>.
  17
  18 // { dg-do run { target c++11 } }
  19
  20 #include <unordered_set>
  21 #include <testsuite_hooks.h>
  22 #include <testsuite_allocator.h>
  23
  24 struct T { int i; };
  25
  26 struct hash
  27 {
  28   std::size_t operator()(const T t) const noexcept
  29   { return t.i; }
  30 };
  31
  32 struct equal_to
  33 {
  34   bool operator()(const T& lhs, const T& rhs) const noexcept
  35   { return lhs.i == rhs.i; }
  36 };
  37
  38 using __gnu_test::propagating_allocator;
  39
  40 // It is undefined behaviour to swap() containers wth unequal allocators
  41 // if the allocator doesn't propagate, so ensure the allocators compare
  42 // equal, while still being able to test propagation via get_personality().
  43 bool
  44 operator==(const propagating_allocator<T, false>&,
  45            const propagating_allocator<T, false>&)
  46 {
  47   return true;
  48 }
  49
  50 bool
  51 operator!=(const propagating_allocator<T, false>&,
  52            const propagating_allocator<T, false>&)
  53 {
  54   return false;
  55 }
  56
  57 void test01()
  58 {
  59   typedef propagating_allocator<T, false> alloc_type;
  60   typedef std::unordered_set<T, hash, equal_to, alloc_type> test_type;
  61   test_type v1(alloc_type(1));
  62   v1.insert(T());
  63   test_type v2(alloc_type(2));
  64   v2.insert(T());
  65   std::swap(v1, v2);
  66   VERIFY(1 == v1.get_allocator().get_personality());
  67   VERIFY(2 == v2.get_allocator().get_personality());
  68   // swap back so assertions in uneq_allocator::deallocate don't fail
  69   std::swap(v1, v2);
  70 }
  71
  72 void test02()
  73 {
  74   typedef propagating_allocator<T, true> alloc_type;
  75   typedef std::unordered_set<T, hash, equal_to, alloc_type> test_type;
  76   test_type v1(alloc_type(1));
  77   v1.insert(T());
  78   test_type v2(alloc_type(2));
  79   v2.insert(T());
  80   std::swap(v1, v2);
  81   VERIFY(2 == v1.get_allocator().get_personality());
  82   VERIFY(1 == v2.get_allocator().get_personality());
  83 }
  84
  85 int main()
  86 {
  87   test01();
  88   test02();
  89   return 0;
  90 }