1#include <string>
2#if !defined (STLPORT) || !defined (_STLP_USE_NO_IOSTREAMS)
3#include <sstream>
4#include <vector>
5#include <iterator>
6
7#include "cppunit/cppunit_proxy.h"
8
9#if !defined (STLPORT) || defined(_STLP_USE_NAMESPACES)
10using namespace std;
11#endif
12
13class IoiterTest : public CPPUNIT_NS::TestCase
14{
15  CPPUNIT_TEST_SUITE(IoiterTest);
16  CPPUNIT_TEST(ioiter_test);
17  CPPUNIT_TEST(assign_test);
18  CPPUNIT_TEST(assign2_test);
19  CPPUNIT_TEST_SUITE_END();
20
21protected:
22  void ioiter_test();
23  void assign_test();
24  void assign2_test();
25};
26
27CPPUNIT_TEST_SUITE_REGISTRATION(IoiterTest);
28
29void IoiterTest::ioiter_test()
30{
31
32  char c;
33  const char *pc;
34  const char *strorg = "abcd";
35  string tmp;
36
37  string objStr(strorg);
38
39  istringstream objIStrStrm1(objStr);
40  istringstream objIStrStrm2(objStr);
41  istringstream objIStrStrm3(objStr);
42
43  pc = strorg;
44  string::size_type sz = strlen(strorg);
45  string::size_type i;
46  for ( i = 0; i < sz; ++i ) {
47    c = *pc++;
48    tmp += c;
49  }
50  CPPUNIT_ASSERT( tmp == "abcd" );
51
52  istreambuf_iterator<char, char_traits<char> > objIStrmbIt1( objIStrStrm1.rdbuf() );
53  istreambuf_iterator<char, char_traits<char> > end;
54
55  tmp.clear();
56
57  for ( i = 0; i < sz /* objIStrmbIt1 != end */; ++i ) {
58    c = *objIStrmbIt1++;
59    tmp += c;
60  }
61  CPPUNIT_ASSERT( tmp == "abcd" );
62
63  tmp.clear();
64
65  istreambuf_iterator<char, char_traits<char> > objIStrmbIt2( objIStrStrm2.rdbuf() );
66  for ( i = 0; i < sz; ++i ) {
67    c = *objIStrmbIt2;
68    tmp += c;
69    objIStrmbIt2++;
70  }
71  CPPUNIT_ASSERT( tmp == "abcd" );
72
73  tmp.clear();
74
75  istreambuf_iterator<char, char_traits<char> > objIStrmbIt3( objIStrStrm3.rdbuf() );
76
77  while ( objIStrmbIt3 != end ) {
78    c = *objIStrmbIt3++;
79    tmp += c;
80  }
81  CPPUNIT_ASSERT( tmp == "abcd" );
82}
83
84void IoiterTest::assign_test()
85{
86  stringstream s( "1234567890" );
87  vector<char> v;
88
89  v.assign( istreambuf_iterator<char>(s), istreambuf_iterator<char>() );
90  CPPUNIT_CHECK( v.size() == 10 );
91  if ( v.size() == 10 ) {
92    CPPUNIT_CHECK( v[0] == '1' );
93    CPPUNIT_CHECK( v[9] == '0' );
94  }
95}
96
97void IoiterTest::assign2_test()
98{
99  stringstream s( "1234567890" );
100  vector<char> v;
101
102  v.assign( istreambuf_iterator<char>(s.rdbuf()), istreambuf_iterator<char>() );
103  CPPUNIT_CHECK( v.size() == 10 );
104  if ( v.size() == 10 ) {
105    CPPUNIT_CHECK( v[0] == '1' );
106    CPPUNIT_CHECK( v[9] == '0' );
107  }
108}
109
110#endif
111