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 **)63int main(int, char**) 64 { 65 { 66 testbuf<char> sb; 67 std::ostream os(&sb); 68 std::endl(os); 69 assert(sb.str() == "\n"); 70 assert(sync_called == 1); 71 assert(os.good()); 72 } 73 #ifndef TEST_HAS_NO_WIDE_CHARACTERS 74 { 75 testbuf<wchar_t> sb; 76 std::wostream os(&sb); 77 std::endl(os); 78 assert(sb.str() == L"\n"); 79 assert(sync_called == 2); 80 assert(os.good()); 81 } 82 #endif 83 84 return 0; 85 } 86