#include <vector>

#include <numeric>

#include <algorithm>

#include "iota.h"

#include "cppunit/cppunit_proxy.h"

#if !defined (STLPORT) || defined(_STLP_USE_NAMESPACES)

using namespace std;

#endif

//

// TestCase class

//

class SearchTest : public CPPUNIT_NS::TestCase

{

  CPPUNIT_TEST_SUITE(SearchTest);

  CPPUNIT_TEST(search0);

  CPPUNIT_TEST(search1);

  CPPUNIT_TEST(search2);

  CPPUNIT_TEST_SUITE_END();

protected:

  void search0();

  void search1();

  void search2();

  static bool str_equal(const char* a_, const char* b_)

  {

    return strcmp(a_, b_) == 0 ? 1 : 0;

  }

};

CPPUNIT_TEST_SUITE_REGISTRATION(SearchTest);

//

// tests implementation

//

void SearchTest::search0()

{

  int v1[6] = { 1, 1, 2, 3, 5, 8 };

  int v2[6] = { 0, 1, 2, 3, 4, 5 };

  int v3[2] = { 3, 4 };

  int* location;

  location = search((int*)v1, (int*)v1 + 6, (int*)v3, (int*)v3 + 2);

  CPPUNIT_ASSERT(location == v1 + 6);

  location = search((int*)v2, (int*)v2 + 6, (int*)v3, (int*)v3 + 2);

  CPPUNIT_ASSERT(location != v2 + 6);

  CPPUNIT_ASSERT(location - v2 == 3);

}

void SearchTest::search1()

{

  typedef vector <int> IntVec;

  IntVec v1(10);

  __iota(v1.begin(), v1.end(), 0);

  IntVec v2(3);

  __iota(v2.begin(), v2.end(), 50);

  IntVec::iterator location;

  location = search(v1.begin(), v1.end(), v2.begin(), v2.end());

  CPPUNIT_ASSERT(location == v1.end());

  __iota(v2.begin(), v2.end(), 4);

  location = search(v1.begin(), v1.end(), v2.begin(), v2.end());

  CPPUNIT_ASSERT(location != v1.end());

  CPPUNIT_ASSERT(location - v1.begin() == 4);

}

void SearchTest::search2()

{

  char const* grades[] = { "A", "B", "C", "D", "F" };

  char const* letters[] = { "Q", "E", "D" };

  const unsigned gradeCount = sizeof(grades) / sizeof(grades[0]);

  const unsigned letterCount = sizeof(letters) / sizeof(letters[0]);

  char const** location = search((char const**)grades, (char const**)grades + gradeCount, (char const**)letters, (char const**)letters + letterCount, str_equal);

  CPPUNIT_ASSERT(location == grades + gradeCount);

  copy((char const**)grades + 1, (char const**)grades + 1 + letterCount, (char const**)letters);

  location = search((char const**)grades, (char const**)grades + gradeCount, (char const**)letters, (char const**)letters + letterCount, str_equal);

  CPPUNIT_ASSERT(location != grades + gradeCount);

  CPPUNIT_ASSERT(location - grades == 1);

}