libcxx/test/std/input.output/iostream.format/output.streams/ostream.manip/endl.pass.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-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:
  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 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
  56         sync()
  57         {
  58             ++sync_called;
  59             return 0;
  60         }
  61 };
  62
  63 int 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 }