libstdc++-v3/testsuite/30_threads/stop_token/stop_callback/invoke.cc

   1 // Copyright (C) 2020 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++2a" }
  19 // { dg-add-options libatomic }
  20 // { dg-do run { target c++2a } }
  21
  22 #include <stop_token>
  23 #include <testsuite_hooks.h>
  24
  25 int lval[5];
  26 int rval[5];
  27
  28 void
  29 test01()
  30 {
  31   std::stop_source ssrc;
  32   std::stop_token stok = ssrc.get_token();
  33   struct F
  34   {
  35     void operator()() const & { ++lval[i]; }
  36     void operator()() && { ++rval[i]; }
  37
  38     int i;
  39   };
  40   std::stop_callback<F> cb0(stok, F{0});
  41   std::stop_callback<F> cb1(stok, F{1});
  42   std::stop_callback<F> cb2(stok, F{2});
  43   F f3{3};
  44   std::stop_callback<F&> cb3(stok, f3);
  45   std::stop_callback<const F> cb4(stok, F{4});
  46
  47   // PR libstdc++/92895
  48   // Callback should be invoked with correct value category.
  49   ssrc.request_stop();
  50
  51   VERIFY( lval[0] == 0 && lval[1] == 0 && lval[2] == 0 );
  52   VERIFY( lval[3] == 1 );
  53   VERIFY( lval[4] == 1 );
  54   VERIFY( rval[0] == 1 );
  55   VERIFY( rval[1] == 1 );
  56   VERIFY( rval[2] == 1 );
  57   VERIFY( rval[3] == 0 && rval[4] == 0 );
  58 }
  59
  60 int main()
  61 {
  62   test01();
  63 }