xref: /llvm-project/libcxx/test/std/strings/basic.string/string.modifiers/string_append/push_back.pass.cpp (revision aeecef08c385b1e4955155dd649a2d3724463849)
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 // <string>
10 
11 // void push_back(charT c) // constexpr since C++20
12 
13 #include <string>
14 #include <algorithm>
15 #include <cassert>
16 
17 #include "test_macros.h"
18 #include "min_allocator.h"
19 
20 struct VeryLarge {
21   long long a;
22   char b;
23 };
24 
25 namespace std {
26   template <>
27   struct char_traits<VeryLarge> {
28     using char_type           = VeryLarge;
29     using int_type            = int;
30     using off_type            = streamoff;
31     using pos_type            = streampos;
32     using state_type          = mbstate_t;
33 
34     static TEST_CONSTEXPR_CXX20 void assign(char_type& c1, const char_type& c2) { c1 = c2; }
35     static bool eq(char_type c1, char_type c2);
36     static bool lt(char_type c1, char_type c2);
37 
38     static int compare(const char_type* s1, const char_type* s2, size_t n);
39     static size_t length(const char_type* s);
40     static const char_type* find(const char_type* s, size_t n, const char_type& a);
41     static char_type* move(char_type* s1, const char_type* s2, size_t n);
42     static TEST_CONSTEXPR_CXX20 char_type* copy(char_type* s1, const char_type* s2, size_t n) {
43       std::copy_n(s2, n, s1);
44       return s1;
45     }
46     static TEST_CONSTEXPR_CXX20 char_type* assign(char_type* s, size_t n, char_type a) {
47       std::fill_n(s, n, a);
48       return s;
49     }
50 
51     static int_type not_eof(int_type c);
52     static char_type to_char_type(int_type c);
53     static int_type to_int_type(char_type c);
54     static bool eq_int_type(int_type c1, int_type c2);
55     static int_type eof();
56   };
57 } // end namespace std
58 
59 template <class S>
60 TEST_CONSTEXPR_CXX20 void
61 test(S s, typename S::value_type c, S expected)
62 {
63     s.push_back(c);
64     LIBCPP_ASSERT(s.__invariants());
65     assert(s == expected);
66 }
67 
68 TEST_CONSTEXPR_CXX20 bool test() {
69   {
70     typedef std::string S;
71     test(S(), 'a', S(1, 'a'));
72     test(S("12345"), 'a', S("12345a"));
73     test(S("12345678901234567890"), 'a', S("12345678901234567890a"));
74   }
75 #if TEST_STD_VER >= 11
76   {
77     typedef std::basic_string<char, std::char_traits<char>, min_allocator<char>> S;
78     test(S(), 'a', S(1, 'a'));
79     test(S("12345"), 'a', S("12345a"));
80     test(S("12345678901234567890"), 'a', S("12345678901234567890a"));
81   }
82 #endif
83 
84   {
85     // https://llvm.org/PR31454
86     std::basic_string<VeryLarge> s;
87     VeryLarge vl = {};
88     s.push_back(vl);
89     s.push_back(vl);
90     s.push_back(vl);
91   }
92 
93   return true;
94 }
95 
96 int main(int, char**)
97 {
98   test();
99 #if TEST_STD_VER > 17
100   static_assert(test());
101 #endif
102 
103   return 0;
104 }
105