1 //===----------------------------------------------------------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is dual licensed under the MIT and the University of Illinois Open 6 // Source Licenses. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 // <ostream> 11 12 // template <class charT, class traits = char_traits<charT> > 13 // class basic_ostream; 14 15 // template <class charT, class traits> 16 // basic_ostream<charT,traits>& ends(basic_ostream<charT,traits>& os); 17 18 #include <ostream> 19 #include <cassert> 20 21 template <class CharT> 22 class testbuf 23 : public std::basic_streambuf<CharT> 24 { 25 typedef std::basic_streambuf<CharT> base; 26 std::basic_string<CharT> str_; 27 public: 28 testbuf() 29 { 30 } 31 32 std::basic_string<CharT> str() const 33 {return std::basic_string<CharT>(base::pbase(), base::pptr());} 34 35 protected: 36 37 virtual typename base::int_type 38 overflow(typename base::int_type ch = base::traits_type::eof()) 39 { 40 if (ch != base::traits_type::eof()) 41 { 42 int n = static_cast<int>(str_.size()); 43 str_.push_back(static_cast<CharT>(ch)); 44 str_.resize(str_.capacity()); 45 base::setp(const_cast<CharT*>(str_.data()), 46 const_cast<CharT*>(str_.data() + str_.size())); 47 base::pbump(n+1); 48 } 49 return ch; 50 } 51 }; 52 53 int main() 54 { 55 { 56 testbuf<char> sb; 57 std::ostream os(&sb); 58 ends(os); 59 assert(sb.str().size() == 1); 60 assert(sb.str().back() == 0); 61 assert(os.good()); 62 } 63 { 64 testbuf<wchar_t> sb; 65 std::wostream os(&sb); 66 ends(os); 67 assert(sb.str().size() == 1); 68 assert(sb.str().back() == 0); 69 assert(os.good()); 70 } 71 } 72
