testTypeLabeledContainer.cc

Go to the documentation of this file.

#include <iostream>

#include <utl/TypeLabeledContainer.h>

#include <tst/Verify.h>
#include <cppunit/extensions/HelperMacros.h>

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


#define ASSERT_EQUAL(x, y) CPPUNIT_ASSERT(Verify<Equal>(x, y))


struct Base {
  Base() { fLabel = 0; }
  virtual ~Base() { }
  int fLabel;
};

struct Foo : public Base {
  Foo() { fLabel = 10; }
};

struct Bar : public Base {
  Bar() { fLabel = 20; }
};

struct Baz : public Bar {
  Baz() { fLabel = 320; }
};


class TestTypeLabeledContainer : public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(TestTypeLabeledContainer);
  CPPUNIT_TEST(TestInsert);
  CPPUNIT_TEST_SUITE_END();

public:
  void setUp() { cerr << "\nTestTypeLabeledContainer: Start\n"; }

  void tearDown() { cerr << "TestTypeLabeledContainer: Stop" << endl; }

  template<class Cast, class T>
  void
  TestIterator(T& c, const int label, const int count)
  {
    int i = 0;
    for (typename T::template Iterator<Cast> it = c.template Begin<Cast>();
         it != c.template End<Cast>(); ++it) {
      ASSERT_EQUAL(it->fLabel, label);
      ++i;
    }
    ASSERT_EQUAL(count, i);
  }

  template<class Cast, class T>
  void
  TestConstIterator(T& c, const int label, const int count)
  {
    int i = 0;
    for (typename T::template ConstIterator<Cast> it = c.template Begin<Cast>();
         it != c.template End<Cast>(); ++it) {
      ASSERT_EQUAL(it->fLabel, label);
      ++i;
    }
    ASSERT_EQUAL(count, i);
  }

  void
  TestInsert()
  {
    Base* const b = new Base;
    Foo* const fb= new Foo;
    Bar* const bb[] = { new Bar, new Bar };
    Baz* const bbb[] = { new Baz, new Baz, new Baz };
    typedef TypeLabeledContainer<Base> MyContainer;
    MyContainer c;
    c.Insert(b);
    c.Insert(fb);
    c.Insert(bb[0]);
    c.Insert(bb[1]);
    c.Insert(bbb[0]);
    c.Insert(bbb[1]);
    c.Insert(bbb[2]);
    const MyContainer& d = c;

    TestIterator<Base>(c, 0, 1);
    TestConstIterator<Base>(d, 0, 1);

    TestIterator<Foo>(c, 10, 1);
    TestConstIterator<Foo>(d, 10, 1);

    TestIterator<Bar>(c, 20, 2);
    TestConstIterator<Bar>(d, 20, 2);

    TestIterator<Baz>(c, 320, 3);
    TestConstIterator<Baz>(d, 320, 3);
  }

};


CPPUNIT_TEST_SUITE_REGISTRATION(TestTypeLabeledContainer);


// Configure (x)emacs for this file ...
// Local Variables:
// mode: c++
// End: