forked from OSchip/llvm-project
				
			
		
			
				
	
	
		
			93 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			93 lines
		
	
	
		
			3.5 KiB
		
	
	
	
		
			C++
		
	
	
	
//===----------------------------------------------------------------------===//
 | 
						|
//
 | 
						|
//                     The LLVM Compiler Infrastructure
 | 
						|
//
 | 
						|
// This file is dual licensed under the MIT and the University of Illinois Open
 | 
						|
// Source Licenses. See LICENSE.TXT for details.
 | 
						|
//
 | 
						|
//===----------------------------------------------------------------------===//
 | 
						|
 | 
						|
// <sstream>
 | 
						|
 | 
						|
// template <class charT, class traits = char_traits<charT>, class Allocator = allocator<charT> >
 | 
						|
// class basic_stringbuf
 | 
						|
 | 
						|
// int_type pbackfail(int_type c = traits::eof());
 | 
						|
 | 
						|
#include <sstream>
 | 
						|
#include <cassert>
 | 
						|
 | 
						|
template <class CharT>
 | 
						|
struct testbuf
 | 
						|
    : public std::basic_stringbuf<CharT>
 | 
						|
{
 | 
						|
    typedef std::basic_stringbuf<CharT> base;
 | 
						|
    explicit testbuf(const std::basic_string<CharT>& str,
 | 
						|
                     std::ios_base::openmode which = std::ios_base::in | std::ios_base::out)
 | 
						|
        : base(str, which) {}
 | 
						|
 | 
						|
    typename base::int_type
 | 
						|
        pbackfail(typename base::int_type c = base::type_traits::eof())
 | 
						|
        {return base::pbackfail(c);}
 | 
						|
 | 
						|
    void pbump(int n) {base::pbump(n);}
 | 
						|
};
 | 
						|
 | 
						|
int main()
 | 
						|
{
 | 
						|
    {
 | 
						|
        testbuf<char> sb("123", std::ios_base::in);
 | 
						|
        assert(sb.sgetc() == '1');
 | 
						|
        assert(sb.snextc() == '2');
 | 
						|
        assert(sb.snextc() == '3');
 | 
						|
        assert(sb.sgetc() == '3');
 | 
						|
        assert(sb.snextc() == std::char_traits<char>::eof());
 | 
						|
        assert(sb.pbackfail('3') == '3');
 | 
						|
        assert(sb.pbackfail('3') == std::char_traits<char>::eof());
 | 
						|
        assert(sb.pbackfail('2') == '2');
 | 
						|
        assert(sb.pbackfail(std::char_traits<char>::eof()) != std::char_traits<char>::eof());
 | 
						|
        assert(sb.pbackfail(std::char_traits<char>::eof()) == std::char_traits<char>::eof());
 | 
						|
        assert(sb.str() == "123");
 | 
						|
    }
 | 
						|
    {
 | 
						|
        testbuf<char> sb("123");
 | 
						|
        assert(sb.sgetc() == '1');
 | 
						|
        assert(sb.snextc() == '2');
 | 
						|
        assert(sb.snextc() == '3');
 | 
						|
        assert(sb.sgetc() == '3');
 | 
						|
        assert(sb.snextc() == std::char_traits<char>::eof());
 | 
						|
        assert(sb.pbackfail('3') == '3');
 | 
						|
        assert(sb.pbackfail('3') == '3');
 | 
						|
        assert(sb.pbackfail(std::char_traits<char>::eof()) != std::char_traits<char>::eof());
 | 
						|
        assert(sb.pbackfail(std::char_traits<char>::eof()) == std::char_traits<char>::eof());
 | 
						|
        assert(sb.str() == "133");
 | 
						|
    }
 | 
						|
    {
 | 
						|
        testbuf<wchar_t> sb(L"123", std::ios_base::in);
 | 
						|
        assert(sb.sgetc() == L'1');
 | 
						|
        assert(sb.snextc() == L'2');
 | 
						|
        assert(sb.snextc() == L'3');
 | 
						|
        assert(sb.sgetc() == L'3');
 | 
						|
        assert(sb.snextc() == std::char_traits<wchar_t>::eof());
 | 
						|
        assert(sb.pbackfail(L'3') == L'3');
 | 
						|
        assert(sb.pbackfail(L'3') == std::char_traits<wchar_t>::eof());
 | 
						|
        assert(sb.pbackfail(L'2') == L'2');
 | 
						|
        assert(sb.pbackfail(std::char_traits<wchar_t>::eof()) != std::char_traits<wchar_t>::eof());
 | 
						|
        assert(sb.pbackfail(std::char_traits<wchar_t>::eof()) == std::char_traits<wchar_t>::eof());
 | 
						|
        assert(sb.str() == L"123");
 | 
						|
    }
 | 
						|
    {
 | 
						|
        testbuf<wchar_t> sb(L"123");
 | 
						|
        assert(sb.sgetc() == L'1');
 | 
						|
        assert(sb.snextc() == L'2');
 | 
						|
        assert(sb.snextc() == L'3');
 | 
						|
        assert(sb.sgetc() == L'3');
 | 
						|
        assert(sb.snextc() == std::char_traits<wchar_t>::eof());
 | 
						|
        assert(sb.pbackfail(L'3') == L'3');
 | 
						|
        assert(sb.pbackfail(L'3') == L'3');
 | 
						|
        assert(sb.pbackfail(std::char_traits<wchar_t>::eof()) != std::char_traits<wchar_t>::eof());
 | 
						|
        assert(sb.pbackfail(std::char_traits<wchar_t>::eof()) == std::char_traits<wchar_t>::eof());
 | 
						|
        assert(sb.str() == L"133");
 | 
						|
    }
 | 
						|
}
 |