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

   1 // Copyright (C) 2013-2016 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-options "-std=gnu++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   bool test __attribute__((unused)) = true;
  60   typedef propagating_allocator<T, false> alloc_type;
  61   typedef std::unordered_set<T, hash, equal_to, alloc_type> test_type;
  62   test_type v1(alloc_type(1));
  63   v1.insert(T());
  64   test_type v2(alloc_type(2));
  65   v2.insert(T());
  66   std::swap(v1, v2);
  67   VERIFY(1 == v1.get_allocator().get_personality());
  68   VERIFY(2 == v2.get_allocator().get_personality());
  69   // swap back so assertions in uneq_allocator::deallocate don't fail
  70   std::swap(v1, v2);
  71 }
  72
  73 void test02()
  74 {
  75   bool test __attribute__((unused)) = true;
  76   typedef propagating_allocator<T, true> alloc_type;
  77   typedef std::unordered_set<T, hash, equal_to, alloc_type> test_type;
  78   test_type v1(alloc_type(1));
  79   v1.insert(T());
  80   test_type v2(alloc_type(2));
  81   v2.insert(T());
  82   std::swap(v1, v2);
  83   VERIFY(2 == v1.get_allocator().get_personality());
  84   VERIFY(1 == v2.get_allocator().get_personality());
  85 }
  86
  87 int main()
  88 {
  89   test01();
  90   test02();
  91   return 0;
  92 }