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 | |
11 | // <memory> |
12 | |
13 | // template <class ForwardIt> |
14 | // constexpr void destroy(ForwardIt, ForwardIt); |
15 | |
16 | #include <memory> |
17 | #include <cassert> |
18 | #include <type_traits> |
19 | |
20 | #include "test_macros.h" |
21 | #include "test_iterators.h" |
22 | |
23 | struct Counted { |
24 | int* counter_; |
25 | TEST_CONSTEXPR Counted(int* counter) : counter_(counter) { ++*counter_; } |
26 | TEST_CONSTEXPR Counted(Counted const& other) : counter_(other.counter_) { ++*counter_; } |
27 | TEST_CONSTEXPR_CXX20 ~Counted() { --*counter_; } |
28 | friend void operator&(Counted) = delete; |
29 | }; |
30 | |
31 | #if TEST_STD_VER > 17 |
32 | constexpr bool test_arrays() { |
33 | { |
34 | using Array = Counted[3]; |
35 | using Alloc = std::allocator<Array>; |
36 | int counter = 0; |
37 | Alloc alloc; |
38 | Array* pool = std::allocator_traits<Alloc>::allocate(alloc, 5); |
39 | |
40 | for (Array* p = pool; p != pool + 5; ++p) { |
41 | Array& arr = *p; |
42 | for (int i = 0; i != 3; ++i) { |
43 | std::allocator_traits<Alloc>::construct(alloc, std::addressof(arr[i]), &counter); |
44 | } |
45 | } |
46 | assert(counter == 5 * 3); |
47 | |
48 | std::destroy(pool, pool + 5); |
49 | ASSERT_SAME_TYPE(decltype(std::destroy(pool, pool + 5)), void); |
50 | assert(counter == 0); |
51 | |
52 | std::allocator_traits<Alloc>::deallocate(alloc, pool, 5); |
53 | } |
54 | { |
55 | using Array = Counted[3][2]; |
56 | using Alloc = std::allocator<Array>; |
57 | int counter = 0; |
58 | Alloc alloc; |
59 | Array* pool = std::allocator_traits<Alloc>::allocate(alloc, 5); |
60 | |
61 | for (Array* p = pool; p != pool + 5; ++p) { |
62 | Array& arr = *p; |
63 | for (int i = 0; i != 3; ++i) { |
64 | for (int j = 0; j != 2; ++j) { |
65 | std::allocator_traits<Alloc>::construct(alloc, std::addressof(arr[i][j]), &counter); |
66 | } |
67 | } |
68 | } |
69 | assert(counter == 5 * 3 * 2); |
70 | |
71 | std::destroy(pool, pool + 5); |
72 | ASSERT_SAME_TYPE(decltype(std::destroy(pool, pool + 5)), void); |
73 | assert(counter == 0); |
74 | |
75 | std::allocator_traits<Alloc>::deallocate(alloc, pool, 5); |
76 | } |
77 | |
78 | return true; |
79 | } |
80 | #endif |
81 | |
82 | template <class It> |
83 | TEST_CONSTEXPR_CXX20 void test() { |
84 | using Alloc = std::allocator<Counted>; |
85 | int counter = 0; |
86 | Alloc alloc; |
87 | Counted* pool = std::allocator_traits<Alloc>::allocate(a&: alloc, n: 5); |
88 | |
89 | for (Counted* p = pool; p != pool + 5; ++p) |
90 | std::allocator_traits<Alloc>::construct(a&: alloc, p: p, args: &counter); |
91 | assert(counter == 5); |
92 | |
93 | std::destroy(It(pool), It(pool + 5)); |
94 | ASSERT_SAME_TYPE(decltype(std::destroy(It(pool), It(pool + 5))), void); |
95 | assert(counter == 0); |
96 | |
97 | std::allocator_traits<Alloc>::deallocate(a&: alloc, p: pool, n: 5); |
98 | } |
99 | |
100 | TEST_CONSTEXPR_CXX20 bool tests() { |
101 | test<Counted*>(); |
102 | test<forward_iterator<Counted*>>(); |
103 | return true; |
104 | } |
105 | |
106 | int main(int, char**) { |
107 | tests(); |
108 | #if TEST_STD_VER > 17 |
109 | test_arrays(); |
110 | static_assert(tests()); |
111 | // TODO: Until std::construct_at has support for arrays, it's impossible to test this |
112 | // in a constexpr context (see https://reviews.llvm.org/D114903). |
113 | // static_assert(test_arrays()); |
114 | #endif |
115 | return 0; |
116 | } |
117 | |