aria2/test/SequenceTest.cc

#include "Sequence.h"
#include <cppunit/extensions/HelperMacros.h>

using namespace std;

class SequenceTest:public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(SequenceTest);
  CPPUNIT_TEST(testParseAndNext);
  CPPUNIT_TEST(testParseAndNext2);
  CPPUNIT_TEST(testFlush);
  CPPUNIT_TEST_SUITE_END();
public:
  void testParseAndNext();
  void testParseAndNext2();
  void testFlush();
};


CPPUNIT_TEST_SUITE_REGISTRATION(SequenceTest);

typedef Sequence<int32_t> IntSequence;

void SequenceTest::testParseAndNext()
{
  IntSequence::Value params[] = {
    IntSequence::Value(1, 2),
    IntSequence::Value(3, 9),
    IntSequence::Value(10, 11),
  };
  IntSequence seq = IntSequence(IntSequence::Values(&params[0], &params[3]));
  CPPUNIT_ASSERT(seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)1, seq.next());
  CPPUNIT_ASSERT(seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)3, seq.next());
  CPPUNIT_ASSERT(seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)4, seq.next());
  CPPUNIT_ASSERT(seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)5, seq.next());
  CPPUNIT_ASSERT(seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)6, seq.next());
  CPPUNIT_ASSERT(seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)7, seq.next());
  CPPUNIT_ASSERT(seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)8, seq.next());
  CPPUNIT_ASSERT(seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)10, seq.next());
  CPPUNIT_ASSERT(!seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)0, seq.next());
 
}

void SequenceTest::testParseAndNext2()
{
  IntSequence::Value params[] = {
    IntSequence::Value(1, 2),
  };
  IntSequence seq = IntSequence(IntSequence::Values(&params[0], &params[1]));
  CPPUNIT_ASSERT(seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)1, seq.next());
  CPPUNIT_ASSERT(!seq.hasNext());
  CPPUNIT_ASSERT_EQUAL((int32_t)0, seq.next());
 
}

void SequenceTest::testFlush()
{
  IntSequence::Value params[] = {
    IntSequence::Value(1, 2),
    IntSequence::Value(3, 9),
    IntSequence::Value(10, 11),
  };
  IntSequence seq = IntSequence(IntSequence::Values(&params[0], &params[3]));
  deque<int32_t> r = seq.flush();

  int32_t answers[] = { 1, 3, 4, 5, 6, 7, 8, 10 };

  CPPUNIT_ASSERT(equal(r.begin(), r.end(), &answers[0])); 
}