boost_1_45_0/libs/iostreams/test/pipeline_test.cpp

// (C) Copyright 2008 CodeRage, LLC (turkanis at coderage dot com)
// (C) Copyright 2004-2007 Jonathan Turkanis
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt.)

// See http://www.boost.org/libs/iostreams for documentation.

#include <fstream>
#include <boost/iostreams/device/file.hpp>
#include <boost/iostreams/filtering_stream.hpp>
#include <boost/test/test_tools.hpp>
#include <boost/test/unit_test.hpp>  
#include "detail/filters.hpp"
#include "detail/sequence.hpp"
#include "detail/temp_file.hpp"
#include "detail/verification.hpp"

using boost::unit_test::test_suite;    

void pipeline_test()
{
    using namespace std;
    using namespace boost;
    using namespace boost::iostreams;
    using namespace boost::iostreams::test;

    {
        test_file src;
        filtering_istream 
            in1( toupper_filter() | 
                 file_source(src.name()) );
        filtering_istream 
            in2( toupper_filter() | 
                 toupper_filter() | 
                 file_source(src.name()) );
        filtering_istream 
            in3( toupper_filter() | 
                 toupper_filter() | 
                 toupper_filter() | 
                 file_source(src.name()) );
        filtering_istream 
            in4( toupper_filter() | 
                 toupper_filter() | 
                 toupper_filter() | 
                 toupper_filter() | 
                 file_source(src.name()) );
        BOOST_CHECK(in1.size() == 2);
        BOOST_CHECK(in2.size() == 3);
        BOOST_CHECK(in3.size() == 4);
        BOOST_CHECK(in4.size() == 5);
    }

    {
        test_file       src;
        uppercase_file  upper;
        filtering_istream 
            first( toupper_filter() | 
                   toupper_multichar_filter() | 
                   file_source(src.name(), in_mode) );
        ifstream second(upper.name().c_str(), in_mode);
        BOOST_CHECK_MESSAGE(
            compare_streams_in_chunks(first, second),
            "failed reading from a filtering_istream in chunks with a "
            "multichar input filter"
        );
    }    
    
    {
        temp_file       dest;
        lowercase_file  lower;
        filtering_ostream  
            out( tolower_filter() |
                 tolower_multichar_filter() |
                 file_sink(dest.name(), out_mode) );
        write_data_in_chunks(out);
        out.reset();
        BOOST_CHECK_MESSAGE(
            compare_files(dest.name(), lower.name()),
            "failed writing to a filtering_ostream in chunks with a "
            "multichar output filter with no buffer"
        );
    }
}

test_suite* init_unit_test_suite(int, char* []) 
{
    test_suite* test = BOOST_TEST_SUITE("pipeline test");
    test->add(BOOST_TEST_CASE(&pipeline_test));
    return test;
}