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, class T>
16 //   basic_ostream<charT, traits>&
17 //   operator<<(basic_ostream<charT, traits>&& os, const T& x);
18 
19 #include <ostream>
20 #include <cassert>
21 
22 #ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
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:
31     testbuf()
32     {
33     }
34 
35     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
41         overflow(typename base::int_type __c = base::traits_type::eof())
42         {
43             if (__c != base::traits_type::eof())
44             {
45                 int n = str_.size();
46                 str_.push_back(__c);
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 __c;
53         }
54 };
55 
56 #endif  // _LIBCPP_HAS_NO_RVALUE_REFERENCES
57 
58 int main()
59 {
60 #ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
61     {
62         testbuf<char> sb;
63         std::ostream(&sb) << "testing...";
64         assert(sb.str() == "testing...");
65     }
66     {
67         testbuf<wchar_t> sb;
68         std::wostream(&sb) << L"123";
69         assert(sb.str() == L"123");
70     }
71 #endif  // _LIBCPP_HAS_NO_RVALUE_REFERENCES
72 }
73