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 22 template <class CharT> 23 class testbuf 24 : public std::basic_streambuf<CharT> 25 { 26 typedef std::basic_streambuf<CharT> base; 27 std::basic_string<CharT> str_; 28 public: testbuf()29 testbuf() 30 { 31 } 32 str() const33 std::basic_string<CharT> str() const 34 {return std::basic_string<CharT>(base::pbase(), base::pptr());} 35 36 protected: 37 38 virtual typename base::int_type overflow(typename base::int_type ch=base::traits_type::eof ())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 main(int,char **)54int main(int, char**) 55 { 56 { 57 testbuf<char> sb; 58 std::ostream os(&sb); 59 ends(os); 60 assert(sb.str().size() == 1); 61 assert(sb.str().back() == 0); 62 assert(os.good()); 63 } 64 { 65 testbuf<wchar_t> sb; 66 std::wostream os(&sb); 67 ends(os); 68 assert(sb.str().size() == 1); 69 assert(sb.str().back() == 0); 70 assert(os.good()); 71 } 72 73 return 0; 74 } 75