C++程序  |  160行  |  5.04 KB

#include <algorithm>
#if !defined (STLPORT) || !defined (_STLP_USE_NO_IOSTREAMS)
#  include <sstream>
#  include <functional>
#  include <iterator>
#  include <vector>
#  include <string>
#endif

#include "cppunit/cppunit_proxy.h"

#if !defined (STLPORT) || defined(_STLP_USE_NAMESPACES)
using namespace std;
#endif

//
// TestCase class
//
class IStreamIteratorTest : public CPPUNIT_NS::TestCase
{
  CPPUNIT_TEST_SUITE(IStreamIteratorTest);
#if defined (STLPORT) && defined (_STLP_USE_NO_IOSTREAMS)
  CPPUNIT_IGNORE;
#endif
  CPPUNIT_TEST(istmit1);
#if !defined (STLPORT) || defined (_STLP_NO_EXTENSIONS)
  CPPUNIT_IGNORE;
#endif
  CPPUNIT_TEST(copy_n_test);
  CPPUNIT_TEST_SUITE_END();

protected:
  void istmit1();
  void copy_n_test();
};

CPPUNIT_TEST_SUITE_REGISTRATION(IStreamIteratorTest);

#if !defined (STLPORT) || !defined (_STLP_USE_NO_IOSTREAMS)
#  if !defined (STLPORT) || !defined (_STLP_LIMITED_DEFAULT_TEMPLATES)
typedef istream_iterator<char> istream_char_ite;
typedef istream_iterator<int> istream_int_ite;
typedef istream_iterator<string> istream_string_ite;
#  else
typedef istream_iterator<char, ptrdiff_t> istream_char_ite;
typedef istream_iterator<int, ptrdiff_t> istream_int_ite;
typedef istream_iterator<string, ptrdiff_t> istream_string_ite;
#  endif
#endif

//
// tests implementation
//
void IStreamIteratorTest::istmit1()
{
#if !defined (STLPORT) || !defined (_STLP_USE_NO_IOSTREAMS)
  const char* buff = "MyString";
  istringstream istr(buff);

  char buffer[100];
  size_t i = 0;
  istr.unsetf(ios::skipws); // Disable white-space skipping.
  istream_char_ite s(istr), meos;
  while (!(s == meos)  &&
  //*TY 01/10/1999 - added end of stream check
  // NOTE operator!= should not be used here ifndef _STLP_FUNCTION_TMPL_PARTIAL_ORDER
         (*s != '\n') &&
         (i < sizeof(buffer) / sizeof(buffer[0]))) {  //*TY 07/28/98 - added index check
    buffer[i++] = *s++;
  }
  buffer[i] = '\0'; // Null terminate buffer.

  CPPUNIT_ASSERT(!strcmp(buffer, buff));

  {
    istringstream empty_istr;
    CPPUNIT_ASSERT( istream_char_ite(empty_istr) == istream_char_ite() );
  }
#endif
}

void IStreamIteratorTest::copy_n_test()
{
#if defined (STLPORT) && !defined (_STLP_NO_EXTENSIONS) && !defined (_STLP_USE_NO_IOSTREAMS)
  //This test check that no character is lost while reading the istream
  //through a istream_iterator.
  {
    istringstream istr("aabbcd");
    string chars;
    istream_char_ite ite = copy_n(copy_n(istream_char_ite(istr),
                                         2, back_inserter(chars)).first,
                                  2, back_inserter(chars)).first;
    CPPUNIT_ASSERT( chars == "aabb" );
    copy_n(ite, 2, back_inserter(chars));
    CPPUNIT_ASSERT( chars == "aabbcd" );
  }

  {
    istringstream istr("11 22 AA BB 33 44 CC DD");
    vector<int> ints;
    vector<string> strings;

    copy_n(istream_int_ite(istr), 2, back_inserter(ints));
    CPPUNIT_ASSERT( ints.size() == 2 );
    CPPUNIT_ASSERT( ints[0] == 11 );
    CPPUNIT_ASSERT( ints[1] == 22 );
    ints.clear();
    istr.clear();
    copy_n(istream_string_ite(istr), 2, back_inserter(strings));
    CPPUNIT_ASSERT( strings.size() == 2 );
    CPPUNIT_ASSERT( strings[0] == "AA" );
    CPPUNIT_ASSERT( strings[1] == "BB" );
    strings.clear();
    istr.clear();
    /* The following code cannot work, '33' is extracted as a string
     * in the previous copy_n call, this value is returned in the pair
     * returned by copy_n but is lost as this istream_iterator is not used.
     * copy_n and istream_iterator can only be combined safely if:
     * - you always extract the same type of istream_iterator and you always reuse
     * the istream_iterator returned by copy_n (see previous test with "aabbcd")
     * - you extract different type of object and no object is convertible to an other
     * as in this current test when you extract int and string (when you extract ints
     * again it fails as int can be converted to strings.
     *
    copy_n(istream_int_ite(istr), 2, back_inserter(ints));
    CPPUNIT_ASSERT( ints.size() == 2 );
    CPPUNIT_ASSERT( ints[0] == 33 );
    CPPUNIT_ASSERT( ints[1] == 44 );
    istr.clear();
    copy_n(istream_string_ite(istr), 2, back_inserter(strings));
    CPPUNIT_ASSERT( strings.size() == 2 );
    CPPUNIT_ASSERT( strings[0] == "CC" );
    CPPUNIT_ASSERT( strings[1] == "DD" );
    */
  }

  {
    istringstream is("1 2 3 4 5 6 7 8 9 10");
    vector<int> ints;
    istream_iterator<int> itr(is);
    itr = copy_n(itr, 0, back_inserter(ints)).first;
    CPPUNIT_ASSERT( ints.empty() );
    itr = copy_n(itr, -1, back_inserter(ints)).first;
    CPPUNIT_ASSERT( ints.empty() );
    itr = copy_n(itr, 2, back_inserter(ints)).first;
    CPPUNIT_ASSERT( ints.size() == 2 );
    CPPUNIT_ASSERT( ints[0] == 1 );
    CPPUNIT_ASSERT( ints[1] == 2 );
    itr = copy_n(itr, 2, back_inserter(ints)).first;
    CPPUNIT_ASSERT( ints.size() == 4 );
    CPPUNIT_ASSERT( ints[2] == 3 );
    CPPUNIT_ASSERT( ints[3] == 4 );
    itr = copy_n(itr, 2, back_inserter(ints)).first;
    CPPUNIT_ASSERT( ints.size() == 6 );
    CPPUNIT_ASSERT( ints[4] == 5 );
    CPPUNIT_ASSERT( ints[5] == 6 );
  }
#endif
}