testReadStream.cc

Go to the documentation of this file.

#include <utl/Test.h>
#include <utl/Math.h>
#include <tst/Verify.h>
#include <utl/ReadStream.h>
#include <cppunit/extensions/HelperMacros.h>
#include <cstdio>

using namespace tst;
using namespace utl;
using namespace std;


struct Data {
  int fTime;
  double fPressure;
  bool fFlag;
};


inline
istream&
operator>>(istream& is, Data& d)
{
  return is >> d.fTime >> ws >> d.fPressure >> ws >> d.fFlag;
}


const string file =
  "1   13.9 0\n"
  "  2   18   1  \t\n"
  "#2  19   1\n"
  "3   123/* foo */ 0\n"
  "4   3    1  //comment \n"
  "   \t   \t   \n"
  "5   22   0 # another\n"
  "6   11 \r  1 # /* bar \n"
  "7   12   0 // what # fool\n"
  "\n"
  "8   14   1 # baz // some \n"
  "/* cc */ 9   15   1      /**//**/         \n"
  "/* crap *//* more crap */ \n"
  "///* an abundance of crap \n"
  "\t 10 /*foo*/16 /* *//**/  0               \n";


const Data result[] = {
  { 1,  13.9, false },
  { 2,  18,   true  },
  { 3,  123,  false },
  { 4,  3,    true  },
  { 5,  22,   false },
  { 6,  11,   true  },
  { 7,  12,   false },
  { 8,  14,   true  },
  { 9,  15,   true  },
  { 10, 16,   false }
};


const string stringResult[] = {
  "1 13.9 0",
  "2 18 1",
  "3 123 0",
  "4 3 1",
  "5 22 0",
  "6 11 1",
  "7 12 0",
  "8 14 1",
  "9 15 1",
  "10 16 0"
};


const string stringResultNoStripping[] = {
  "1   13.9 0",
  "  2   18   1  \t",
  "#2  19   1",
  "3   123/* foo */ 0",
  "4   3    1  //comment ",
  "   \t   \t   ",
  "5   22   0 # another",
  "6   11 \r  1 # /* bar ",
  "7   12   0 // what # fool",
  "8   14   1 # baz // some ",
  "/* cc */ 9   15   1      /**//**/         ",
  "/* crap *//* more crap */ ",
  "///* an abundance of crap ",
  "\t 10 /*foo*/16 /* *//**/  0               "
};


class testReadStream : public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(testReadStream);
  CPPUNIT_TEST(TestVectorData);
  CPPUNIT_TEST(TestVectorString);
  CPPUNIT_TEST(TestVectorStringNoStripping);
  CPPUNIT_TEST_SUITE_END();

public:
  void setUp() { }

  void tearDown() { }

  void
  TestVectorData()
  {
    istringstream is(file);
    vector<Data> v;
    ReadStream(is).GetLines(v);
    CPPUNIT_ASSERT(v.size() == Length(result));
    for (unsigned int i = 0, n = v.size(); i < n; ++i) {
      CPPUNIT_ASSERT(Verify<Equal>(v[i].fTime, result[i].fTime));
      CPPUNIT_ASSERT(Verify<Equal>(v[i].fPressure, result[i].fPressure));
      CPPUNIT_ASSERT(Verify<Equal>(v[i].fFlag, result[i].fFlag));
    }
  }

  void
  TestVectorString()
  {
    istringstream is(file);
    vector<string> v;
    ReadStream(is).GetLines(v);
    CPPUNIT_ASSERT(v.size() == Length(stringResult));
    for (unsigned int i = 0, n = v.size(); i < n; ++i)
      CPPUNIT_ASSERT(Verify<Equal>(v[i], stringResult[i]));
  }

  void
  TestVectorStringNoStripping()
  {
    istringstream is(file);
    vector<string> v;
    ReadStream(is).GetLines(v, false);
    CPPUNIT_ASSERT(v.size() == Length(stringResultNoStripping));
    for (unsigned int i = 0, n = v.size(); i < n; ++i)
      CPPUNIT_ASSERT(Verify<Equal>(v[i], stringResultNoStripping[i]));
  }

};


CPPUNIT_TEST_SUITE_REGISTRATION(testReadStream);