Ticket #955: testIostreams.cpp

#include <iostream>
#include <sstream>

#include <boost/iostreams/filtering_stream.hpp>
#include <boost/iostreams/copy.hpp>

#ifdef USE_BZIP2
#    include <boost/iostreams/filter/bzip2.hpp>
#else
#    include <boost/iostreams/filter/zlib.hpp>
#endif

int main (int argc, char** argv)
   {
   int loops(NUM_LOOPS);
   if (argc > 1)
      {
      loops = atoi(argv[1]);
      }

    std::string testData =
    "\n\
    <node name='car'>\n\
        <interface name='CarInterface'>\n\
            <method name='accelerate'/>\n\
            <method name='decelerate'/>\n\
            <method name='turnLeft'/>\n\
            <method name='turnRight'/>\n\
            <signal name='crashed'/>\n\
        </interface>\n\
    </node>\n\
    ";

    std::cerr << "Test data : " << testData << std::endl;

    int i(0);
    try
      {
      for (i = 0; i < loops; ++i)
         {
         if (!(i % 1000))
            {
            std::cerr << i << ' ';
            }

         std::istringstream ifs(testData);
         std::ostringstream toFS;

         boost::iostreams::filtering_ostream ofs;

#ifdef USE_BZIP2
         // pushing bzip2 has no problems
         ofs.push(boost::iostreams::bzip2_compressor());
#else
         // pushing zlib causes the bad_alloc exception after many loops
         // consuming more memory each loop
         ofs.push(boost::iostreams::zlib_compressor());
#endif

         ofs.push(toFS);

         boost::iostreams::copy(ifs, ofs);
         ofs.reset();
         }
      }
   catch (const std::exception& e)
      {
      std::cerr << std::endl << "Caught std::exception on iteration " << i << ": " << e.what() << std::endl;
      }

   if (i == loops)
      {
      std::cerr << std::endl << "TEST OK on " << i << " iterations" << std::endl;
      }
   else
      {
      std::cerr << std::endl << "TEST FAILED on iteration " << i << std::endl;
      }
   return 0;
   }