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// <ostream>
10
11// template <class charT, class traits = char_traits<charT> >
12// class basic_ostream;
13
14// template <class charT, class traits>
15// basic_ostream<charT,traits>& ends(basic_ostream<charT,traits>& os);
16
17#include <ostream>
18#include <cassert>
19
20#include "test_macros.h"
21
22template <class CharT>
23class testbuf
24 : public std::basic_streambuf<CharT>
25{
26 typedef std::basic_streambuf<CharT> base;
27 std::basic_string<CharT> str_;
28public:
29 testbuf()
30 {
31 }
32
33 std::basic_string<CharT> str() const
34 {return std::basic_string<CharT>(base::pbase(), base::pptr());}
35
36protected:
37
38 virtual typename base::int_type
39 overflow(typename base::int_type ch = base::traits_type::eof())
40 {
41 if (ch != base::traits_type::eof())
42 {
43 int n = static_cast<int>(str_.size());
44 str_.push_back(static_cast<CharT>(ch));
45 str_.resize(str_.capacity());
46 base::setp(const_cast<CharT*>(str_.data()),
47 const_cast<CharT*>(str_.data() + str_.size()));
48 base::pbump(n+1);
49 }
50 return ch;
51 }
52};
53
54int main(int, char**)
55{
56 {
57 testbuf<char> sb;
58 std::ostream os(&sb);
59 std::ends(os);
60 assert(sb.str().size() == 1);
61 assert(sb.str().back() == 0);
62 assert(os.good());
63 }
64#ifndef TEST_HAS_NO_WIDE_CHARACTERS
65 {
66 testbuf<wchar_t> sb;
67 std::wostream os(&sb);
68 std::ends(os);
69 assert(sb.str().size() == 1);
70 assert(sb.str().back() == 0);
71 assert(os.good());
72 }
73#endif
74
75 return 0;
76}
77

source code of libcxx/test/std/input.output/iostream.format/output.streams/ostream.manip/ends.pass.cpp