libstdc++-v3/testsuite/20_util/polymorphic_allocator/resource.cc

   1 // Copyright (C) 2018-2022 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++17 } }
  19 // { dg-skip-if "" { *-*-* } { -fno-aligned-new } }
  20
  21 #include <memory_resource>
  22 #include <testsuite_allocator.h>
  23
  24 struct X { int i = 0; };
  25
  26 using test_type = std::pmr::polymorphic_allocator<X>;
  27
  28 void
  29 test01()
  30 {
  31   __gnu_test::memory_resource r;
  32   test_type a(&r), b(&r);
  33   VERIFY( a == a );
  34   VERIFY( ! (a != a) );
  35   VERIFY( a == b );
  36   VERIFY( ! (a != b) );
  37   VERIFY( a.resource() == &r );
  38   VERIFY( a.resource() == b.resource() );
  39
  40   __gnu_test::memory_resource r2(r);
  41   test_type c(&r2);
  42   VERIFY( c.resource() == &r2 );
  43   VERIFY( c.resource() != a.resource() );
  44   VERIFY( c == a );
  45 }
  46
  47 void
  48 test02()
  49 {
  50   __gnu_test::memory_resource r1, r2;
  51   test_type a(&r1), b(&r2);
  52   VERIFY( a == a );
  53   VERIFY( b == b );
  54   VERIFY( ! (a == b) );
  55   VERIFY( ! (b == a) );
  56   VERIFY( a != b );
  57   VERIFY( b != a );
  58   VERIFY( a.resource() == &r1 );
  59   VERIFY( a.resource() != b.resource() );
  60
  61   test_type c;
  62   VERIFY( c == c );
  63   VERIFY( ! (a == c) );
  64   VERIFY( ! (c == a) );
  65   VERIFY( ! (b == c) );
  66   VERIFY( ! (c == b) );
  67   VERIFY( a.resource() != c.resource() );
  68   VERIFY( c.resource() == std::pmr::get_default_resource() );
  69
  70   std::pmr::set_default_resource(&r1);
  71   VERIFY( c.resource() != &r1 );
  72
  73   test_type d;
  74   VERIFY( d.resource() == &r1 );
  75   VERIFY( d != c );
  76   VERIFY( d == a );
  77
  78   std::pmr::set_default_resource(nullptr);
  79 }
  80
  81 int
  82 main()
  83 {
  84   test01();
  85   test02();
  86 }