77 lines · cpp
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-exception6//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 22template <class CharT>23class testbuf24 : public std::basic_streambuf<CharT>25{26 typedef std::basic_streambuf<CharT> base;27 std::basic_string<CharT> str_;28public:29 testbuf()30 {31 }32 33 std::basic_string<CharT> str() const34 {return std::basic_string<CharT>(base::pbase(), base::pptr());}35 36protected:37 38 virtual typename base::int_type39 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 54int main(int, char**)55{56 {57 testbuf<char> sb;58 std::ostream os(&sb);59 std::ends(os);60 assert(sb.str().size() == 1);61 assert(sb.str().back() == 0);62 assert(os.good());63 }64#ifndef TEST_HAS_NO_WIDE_CHARACTERS65 {66 testbuf<wchar_t> sb;67 std::wostream os(&sb);68 std::ends(os);69 assert(sb.str().size() == 1);70 assert(sb.str().back() == 0);71 assert(os.good());72 }73#endif74 75 return 0;76}77