libcxx/test/std/strings/basic.string/string.modifiers/string_assign/string_view.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 // <string>
  10
  11 // basic_string<charT,traits,Allocator>&
  12 //   assign(basic_string_view<charT,traits> sv);
  13
  14 #include <string>
  15 #include <string_view>
  16 #include <cassert>
  17
  18 #include "test_macros.h"
  19 #include "min_allocator.h"
  20 #include "test_allocator.h"
  21
  22 template <class S, class SV>
  23 void
  24 test(S s, SV sv, S expected)
  25 {
  26     s.assign(sv);
  27     LIBCPP_ASSERT(s.__invariants());
  28     assert(s == expected);
  29 }
  30
  31 template <class S, class SV>
  32 void
  33 testAlloc(S s, SV sv, const typename S::allocator_type& a)
  34 {
  35     s.assign(sv);
  36     LIBCPP_ASSERT(s.__invariants());
  37     assert(s == sv);
  38     assert(s.get_allocator() == a);
  39 }
  40
  41 int main(int, char**)
  42 {
  43     {
  44     typedef std::string S;
  45     typedef std::string_view SV;
  46     test(S(), SV(), S());
  47     test(S(), SV("12345"), S("12345"));
  48     test(S(), SV("1234567890"), S("1234567890"));
  49     test(S(), SV("12345678901234567890"), S("12345678901234567890"));
  50
  51     test(S("12345"), SV(), S());
  52     test(S("12345"), SV("12345"), S("12345"));
  53     test(S("12345"), SV("1234567890"), S("1234567890"));
  54     test(S("12345"), SV("12345678901234567890"), S("12345678901234567890"));
  55
  56     test(S("1234567890"), SV(), S());
  57     test(S("1234567890"), SV("12345"), S("12345"));
  58     test(S("1234567890"), SV("1234567890"), S("1234567890"));
  59     test(S("1234567890"), SV("12345678901234567890"), S("12345678901234567890"));
  60
  61     test(S("12345678901234567890"), SV(), S());
  62     test(S("12345678901234567890"), SV("12345"), S("12345"));
  63     test(S("12345678901234567890"), SV("1234567890"), S("1234567890"));
  64     test(S("12345678901234567890"), SV("12345678901234567890"),
  65          S("12345678901234567890"));
  66
  67     testAlloc(S(), SV(), std::allocator<char>());
  68     testAlloc(S(), SV("12345"), std::allocator<char>());
  69     testAlloc(S(), SV("1234567890"), std::allocator<char>());
  70     testAlloc(S(), SV("12345678901234567890"), std::allocator<char>());
  71     }
  72
  73 #if TEST_STD_VER >= 11
  74     {
  75     typedef std::basic_string     <char, std::char_traits<char>, min_allocator<char>> S;
  76     typedef std::basic_string_view<char, std::char_traits<char> > SV;
  77     test(S(), SV(), S());
  78     test(S(), SV("12345"), S("12345"));
  79     test(S(), SV("1234567890"), S("1234567890"));
  80     test(S(), SV("12345678901234567890"), S("12345678901234567890"));
  81
  82     test(S("12345"), SV(), S());
  83     test(S("12345"), SV("12345"), S("12345"));
  84     test(S("12345"), SV("1234567890"), S("1234567890"));
  85     test(S("12345"), SV("12345678901234567890"), S("12345678901234567890"));
  86
  87     test(S("1234567890"), SV(), S());
  88     test(S("1234567890"), SV("12345"), S("12345"));
  89     test(S("1234567890"), SV("1234567890"), S("1234567890"));
  90     test(S("1234567890"), SV("12345678901234567890"), S("12345678901234567890"));
  91
  92     test(S("12345678901234567890"), SV(), S());
  93     test(S("12345678901234567890"), SV("12345"), S("12345"));
  94     test(S("12345678901234567890"), SV("1234567890"), S("1234567890"));
  95     test(S("12345678901234567890"), SV("12345678901234567890"),
  96          S("12345678901234567890"));
  97
  98     testAlloc(S(), SV(), min_allocator<char>());
  99     testAlloc(S(), SV("12345"), min_allocator<char>());
 100     testAlloc(S(), SV("1234567890"), min_allocator<char>());
 101     testAlloc(S(), SV("12345678901234567890"), min_allocator<char>());
 102     }
 103 #endif
 104
 105   return 0;
 106 }