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