libcxx/test/std/algorithms/alg.modifying.operations/alg.fill/pstl.exception_handling.pass.cpp

   1 //===----------------------------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 // UNSUPPORTED: c++03, c++11, c++14
  10 // UNSUPPORTED: no-exceptions
  11 // REQUIRES: has-unix-headers
  12
  13 // UNSUPPORTED: libcpp-has-no-incomplete-pstl
  14
  15 // check that std::fill(ExecutionPolicy) and std::fill_n(ExecutionPolicy) terminate on user-thrown exceptions
  16
  17 #include <algorithm>
  18
  19 #include "check_assertion.h"
  20 #include "test_execution_policies.h"
  21 #include "test_iterators.h"
  22
  23 #ifndef TEST_HAS_NO_EXCEPTIONS
  24 struct ThrowOnCopy {
  25   ThrowOnCopy& operator=(const ThrowOnCopy&) { throw int{}; }
  26 };
  27 #endif
  28
  29 int main(int, char**) {
  30   ThrowOnCopy a[2]{};
  31   int b[2]{};
  32
  33   test_execution_policies([&](auto&& policy) {
  34     // std::fill
  35     EXPECT_STD_TERMINATE([&] { (void)std::fill(policy, std::begin(a), std::end(a), ThrowOnCopy{}); });
  36     EXPECT_STD_TERMINATE([&] {
  37       try {
  38         (void)std::fill(
  39             policy, util::throw_on_move_iterator(std::begin(b), 1), util::throw_on_move_iterator(std::end(b), 1), 0);
  40       } catch (const util::iterator_error&) {
  41         assert(false);
  42       }
  43       std::terminate(); // make the test pass in case the algorithm didn't move the iterator
  44     });
  45
  46     // std::fill_n
  47     EXPECT_STD_TERMINATE([&] { (void)std::fill_n(policy, std::begin(a), std::size(a), ThrowOnCopy{}); });
  48     EXPECT_STD_TERMINATE([&] {
  49       try {
  50         (void)std::fill_n(policy, util::throw_on_move_iterator(std::begin(b), 1), std::size(b), 0);
  51       } catch (const util::iterator_error&) {
  52         assert(false);
  53       }
  54       std::terminate(); // make the test pass in case the algorithm didn't move the iterator
  55     });
  56   });
  57 }