81 lines · 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-exception6//7//===----------------------------------------------------------------------===//8 9// <string>10 11// basic_string<charT,traits,Allocator>& assign(const charT* s); // constexpr since C++2012 13#include <string>14#include <stdexcept>15#include <cassert>16 17#include "test_macros.h"18#include "min_allocator.h"19#include "asan_testing.h"20 21template <class S>22TEST_CONSTEXPR_CXX20 void test(S s, const typename S::value_type* str, S expected) {23 s.assign(str);24 LIBCPP_ASSERT(s.__invariants());25 assert(s == expected);26 LIBCPP_ASSERT(is_string_asan_correct(s));27}28 29template <class S>30TEST_CONSTEXPR_CXX20 void test_string() {31 test(S(), "", S());32 test(S(), "12345", S("12345"));33 test(S(), "12345678901234567890", S("12345678901234567890"));34 35 test(S("12345"), "", S());36 test(S("12345"), "12345", S("12345"));37 test(S("12345"), "1234567890", S("1234567890"));38 39 test(S("12345678901234567890"), "", S());40 test(S("12345678901234567890"), "12345", S("12345"));41 test(S("12345678901234567890"), "12345678901234567890", S("12345678901234567890"));42 43 // Starting from long string (no SSO)44 test(S("1234512345678901234567890"), "", S());45 test(S("1234512345678901234567890"), "12345", S("12345"));46 test(S("1234512345678901234567890"), "12345678901234567890", S("12345678901234567890"));47}48 49TEST_CONSTEXPR_CXX20 bool test() {50 test_string<std::string>();51#if TEST_STD_VER >= 1152 test_string<std::basic_string<char, std::char_traits<char>, min_allocator<char>>>();53 test_string<std::basic_string<char, std::char_traits<char>, safe_allocator<char>>>();54#endif55 56 { // test assignment to self57 typedef std::string S;58 S s_short = "123/";59 S s_long = "Lorem ipsum dolor sit amet, consectetur/";60 61 s_short.assign(s_short.c_str());62 assert(s_short == "123/");63 s_short.assign(s_short.c_str() + 2);64 assert(s_short == "3/");65 66 s_long.assign(s_long.c_str() + 30);67 assert(s_long == "nsectetur/");68 }69 70 return true;71}72 73int main(int, char**) {74 test();75#if TEST_STD_VER > 1776 static_assert(test());77#endif78 79 return 0;80}81