/* Copyright (C) 2014 InfiniDB, Inc.

   This program is free software; you can redistribute it and/or
   modify it under the terms of the GNU General Public License
   as published by the Free Software Foundation; version 2 of
   the License.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
   MA 02110-1301, USA. */

/**
 * $Id$
 */

#include <iostream>
#include <string>
using namespace std;

#include <cppunit/extensions/HelperMacros.h>

#include "dataconvert.h"
#include "funchelpers.h"
using namespace funcexp;
using namespace dataconvert;
#define MAX_DAY_NUMBER 3652424L

#include "functioncolumn.h"
#include "intervalcolumn.h"
using namespace execplan;

#include "timeextract.h"

class FuncExpTest : public CppUnit::TestFixture
{
  CPPUNIT_TEST_SUITE(FuncExpTest);

  CPPUNIT_TEST(fe_getnumbertest);
  CPPUNIT_TEST(fe_strtodatetest);
  CPPUNIT_TEST(fe_dateaddtest);
  CPPUNIT_TEST(fe_daynametest);
  CPPUNIT_TEST(fe_fromunixtimetest);
  CPPUNIT_TEST_SUITE_END();

 private:
 public:
  void setUp()
  {
  }

  void tearDown()
  {
  }

  void show_datetime_debugs(const string& nm, const DateTime& d1)
  {
    cout << nm << ".day     = " << setw(10) << d1.day << endl;
    cout << nm << ".month   = " << setw(10) << d1.month << endl;
    cout << nm << ".year    = " << setw(10) << d1.year << endl;
    cout << nm << ".hour    = " << setw(10) << d1.hour << endl;
    cout << nm << ".minute  = " << setw(10) << d1.minute << endl;
    cout << nm << ".second  = " << setw(10) << d1.second << endl;
    cout << nm << ".msecond = " << setw(10) << d1.msecond << endl;
  }

  void fe_strtodatetest()
  {
    struct DateCheck
    {
      const char* inputstr;
      const char* formatstr;
      DateTime date;
    };

    DateCheck date_tests[] = {
        {"2009", "%Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
        {"     2009", "%Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
        {"     2009", "  %Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
        {"jan", "%b", DateTime(0, 1, 0, 0, 0, 0, 0)},
        {"2009feb", "%Y%b", DateTime(2009, 2, 0, 0, 0, 0, 0)},
        {"  2009    ApR", " %Y %b", DateTime(2009, 4, 0, 0, 0, 0, 0)},
        {"200910", "%Y%m", DateTime(2009, 10, 0, 0, 0, 0, 0)},
        {"  2009    10", " %Y %m", DateTime(2009, 10, 0, 0, 0, 0, 0)},
        {"200910", "%Y%c", DateTime(2009, 10, 0, 0, 0, 0, 0)},
        {"  2009    10", " %Y %c", DateTime(2009, 10, 0, 0, 0, 0, 0)},
        {"01,5,2013", "%d,%m,%Y", DateTime(2013, 5, 1, 0, 0, 0, 0)},
        {"01,5,2013", "%e,%m,%Y", DateTime(2013, 5, 1, 0, 0, 0, 0)},
        {"7,5th,2012", "%m,%D,%Y", DateTime(2012, 7, 5, 0, 0, 0, 0)},
        {"a09:30:17", "a%h:%i", DateTime(0, 0, 0, 9, 30, 0, 0)},
        {"a09:30:17", "%h:%i", DateTime(0, 0, 0, 0, 0, 0, 0)},
        {"9:30.170", "%h:%i.%f", DateTime(0, 0, 0, 9, 30, 0, 170000)},
        {"178546 17", "%f  %H", DateTime(0, 0, 0, 17, 0, 0, 178546)},
        {"2009march415", "%Y%b", DateTime(2009, 3, 0, 0, 0, 0, 0)},
        {"  2009    July", " %Y %b", DateTime(2009, 7, 0, 0, 0, 0, 0)},
        {"09  pm", "%h %p", DateTime(0, 0, 0, 21, 0, 0, 0)},
        {"09:13:14pm", "%r", DateTime(0, 0, 0, 9, 13, 14, 0)},
        {"12:13:14am", "%r", DateTime(0, 0, 0, 12, 13, 14, 0)},
        {"92", "    %Y", DateTime(1992, 0, 0, 0, 0, 0, 0)},
        {"9", "    %Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
        {"2013 31", "%Y %j", DateTime(2013, 1, 31, 0, 0, 0, 0)},
        {"2013 1 3", "%Y %U %w", DateTime(2013, 1, 9, 0, 0, 0, 0)},
        {"2013 1 3", "%Y %u %w", DateTime(2013, 1, 9, 0, 0, 0, 0)},
        {"2007 1 3", "%Y %U %w", DateTime(2007, 1, 10, 0, 0, 0, 0)},
        {"2007 1 3", "%Y %u %w", DateTime(2007, 1, 3, 0, 0, 0, 0)},
        {"2013 1 3", "%X %V %w", DateTime(2013, 1, 9, 0, 0, 0, 0)},
        {"2013 1 3", "%X %v %w", DateTime(2013, 1, 9, 0, 0, 0, 0)},
        {"2007 1 3", "%X %V %w", DateTime(2007, 1, 10, 0, 0, 0, 0)},
        {"2007 1 3", "%X %v %w", DateTime(2007, 1, 3, 0, 0, 0, 0)},
        {"33:30.170", "%h:%i.%f", DateTime(0, 0, 0, 0, 0, 0, 0)},
        {"01,25,2013", "%d,%m,%Y", DateTime(0, 0, 0, 0, 0, 0, 0)},
        {"%09:30:17", "%%%h:%i:%s", DateTime(0, 0, 0, 9, 30, 17, 0)},
        {"2009 foobar", "%Y", DateTime(2009, 0, 0, 0, 0, 0, 0)},
        {"2009-12-31", "%h:%i:%s %p", DateTime(0, 0, 0, 0, 0, 0, 0)},
        {"12:00:01 AM", "%h:%i:%s %p", DateTime(0, 0, 0, 0, 0, 1, 0)},
        {"12:00:01 PM", "%h:%i:%s %p", DateTime(0, 0, 0, 12, 0, 1, 0)},
        {"11:00:01 PM", "%h:%i:%s %p", DateTime(0, 0, 0, 23, 0, 1, 0)},
        {"11:00:01 AM", "%h:%i:%s %p", DateTime(0, 0, 0, 11, 0, 1, 0)},
        {"10000-01-02", "%Y:%m:%d", DateTime(0, 0, 0, 0, 0, 0, 0)},
    };

    for (unsigned i = 0; i < sizeof(date_tests) / sizeof(DateCheck); i++)
    {
      DateTime dt;
      dt.year = 0;
      dt.month = 0;
      dt.day = 0;
      dt.hour = 0;
      dt.minute = 0;
      dt.second = 0;
      dt.msecond = 0;

      TimeExtractor extractor;
      int ret = extractor.extractTime(date_tests[i].inputstr, date_tests[i].formatstr, dt);

      if (ret != 0)
        cout << "Extractor reported error for " << date_tests[i].inputstr << "," << date_tests[i].formatstr
             << endl;

      bool check =
          ((*(reinterpret_cast<uint64_t*>(&date_tests[i].date))) == ((*(reinterpret_cast<uint64_t*>(&dt)))));

      if (!check)
      {
        printf("For input \"%s\", format \"%s\", check 0x%016lx vs 0x%016lx\n", date_tests[i].inputstr,
               date_tests[i].formatstr, *(reinterpret_cast<uint64_t*>(&dt)),
               *(reinterpret_cast<uint64_t*>(&date_tests[i].date)));
        show_datetime_debugs("check", dt);
        show_datetime_debugs("ref", date_tests[i].date);
      }

      CPPUNIT_ASSERT(check);
    }
  }

  void fe_getnumbertest()
  {
    const int MAX_VALS = 5;
    struct Check
    {
      const char* str;
      int numvals;
      int vals[MAX_VALS];
    };

    Check tests[] = {
        {" ", 0, {}},
        {"0", 1, {0}},
        {"0.0", 2, {0, 0}},
        {"0:0.0", 3, {0, 0, 0}},
        {"-0", 1, {0}},
        {"0000-000", 1, {0}},
        {"-2", 1, {-2}},
        {"2223", 1, {2223}},
        {"37", 1, {37}},
        {"393", 1, {393}},
        {"39-3", 1, {39}},
        {"17-", 1, {17}},
        {"10.2303", 2, {10, 2303}},
        {"20:10.2303", 3, {20, 10, 2303}},
        {"17:37", 2, {17, 37}},
    };

    int a1[MAX_VALS] = {0xdeadbeef};

    for (unsigned i = 0; i < sizeof(tests) / sizeof(Check); i++)
    {
      int rc1 = helpers::getNumbers(tests[i].str, a1, execplan::OP_ADD);

      cout << "For input \"" << tests[i].str << "\", numbers = " << rc1 << ",[";

      bool check = true;

      if (rc1 != tests[i].numvals)
        check = false;

      for (int j = 0; j < rc1; ++j)
      {
        if (j > 0)
          cout << ',';

        cout << a1[j];

        if (a1[j] != tests[i].vals[j])
        {
          check = false;
        }
      }

      cout << ']' << endl;

      CPPUNIT_ASSERT(check);
    }
  }

  void fe_dateaddtest()
  {
    struct DateCheck
    {
      const char* intervalstr;
      IntervalColumn::interval_type unit;
      OpType funcType;
      DateTime date;
      DateTime ref;
    };

    DateCheck date_tests[] = {
        {"111111:22", IntervalColumn::INTERVAL_YEAR_MONTH, execplan::OP_ADD,
         DateTime(1997, 01, 01, 0, 0, 0, 0), DateTime(0, 0, 0, 0, 0, 0, 0)},
        {"-30", IntervalColumn::INTERVAL_DAY, execplan::OP_ADD, DateTime(2013, 3, 1, 0, 0, 0, 0),
         DateTime(2013, 1, 30, 0, 0, 0, 0)},
    };

    for (unsigned i = 0; i < sizeof(date_tests) / sizeof(DateCheck); i++)
    {
      DateTime dt;

      uint64_t timeval = *(reinterpret_cast<uint64_t*>(&date_tests[i].date));
      uint64_t value = helpers::dateAdd(timeval, date_tests[i].intervalstr, date_tests[i].unit, false,
                                        date_tests[i].funcType);

      *(reinterpret_cast<uint64_t*>(&dt)) = value;

      bool check = (value == *(reinterpret_cast<uint64_t*>(&date_tests[i].ref)));

      if (!check)
      {
        printf("For interval \"%s\", check 0x%016lx vs 0x%016lx\n", date_tests[i].intervalstr,
               *(reinterpret_cast<uint64_t*>(&dt)), *(reinterpret_cast<uint64_t*>(&date_tests[i].ref)));
        show_datetime_debugs("check", dt);
        show_datetime_debugs("ref", date_tests[i].ref);
      }

      CPPUNIT_ASSERT(check);
    }
  }

  void fe_daynametest()
  {
    struct DateCheck
    {
      DateTime date;
      const char* dayname;
    };

    DateCheck date_tests[] = {
        {DateTime(1997, 01, 01, 0, 0, 0, 0), "Wednesday"},
    };

    for (unsigned i = 0; i < sizeof(date_tests) / sizeof(DateCheck); i++)
    {
      boost::gregorian::date d(date_tests[i].date.year, date_tests[i].date.month, date_tests[i].date.day);
      uint32_t dayofweek = helpers::calc_mysql_weekday(date_tests[i].date.year, date_tests[i].date.month,
                                                       date_tests[i].date.day, false);

      bool check = (strcmp(helpers::weekdayFullNames[dayofweek].c_str(), date_tests[i].dayname) == 0);

      if (!check)
      {
        printf("For date %s, check %s vs %s\n", to_simple_string(d).c_str(),
               helpers::weekdayFullNames[dayofweek].c_str(), date_tests[i].dayname);
      }

      CPPUNIT_ASSERT(check);
    }
  }

  DateTime getDateTime(int64_t val)
  {
    if (val < 0 || val > helpers::TIMESTAMP_MAX_VALUE)
      return 0;

    DateTime dt;

    struct tm tmp_tm;
    time_t tmp_t = (time_t)val;
    localtime_r(&tmp_t, &tmp_tm);

    // to->neg=0;
    dt.year = (int64_t)((tmp_tm.tm_year + 1900) % 10000);
    dt.month = (int64_t)tmp_tm.tm_mon + 1;
    dt.day = (int64_t)tmp_tm.tm_mday;
    dt.hour = (int64_t)tmp_tm.tm_hour;
    dt.minute = (int64_t)tmp_tm.tm_min;
    dt.second = (int64_t)tmp_tm.tm_sec;
    dt.msecond = 0;
    return dt;
  }

  void fe_fromunixtimetest()
  {
    struct DateCheck
    {
      int64_t ts;
      const char* formatstr;
      const char* refstr;
    };

    DateCheck date_tests[] = {
        {-26, "%H:%i:%s", "NULL"},
        {26, "%H:%i:%s", "18:00:26"},
        {0, "%H:%i:%s", "18:00:00"},
    };

    for (unsigned i = 0; i < sizeof(date_tests) / sizeof(DateCheck); i++)
    {
      bool check = false;
      string ret;
      DateTime dt = getDateTime(date_tests[i].ts);

      if (*reinterpret_cast<int64_t*>(&dt) == 0)
      {
        check = (strcmp(date_tests[i].refstr, "NULL") == 0);
      }
      else
      {
        ret = helpers::IDB_date_format(dt, date_tests[i].formatstr);
        check = (strcmp(date_tests[i].refstr, ret.c_str()) == 0);
      }

      if (!check)
      {
        printf("For timestamp %ld, check %s vs %s\n", date_tests[i].ts, ret.c_str(), date_tests[i].refstr);
      }

      CPPUNIT_ASSERT(check);
    }
  }
};

CPPUNIT_TEST_SUITE_REGISTRATION(FuncExpTest);

#include <cppunit/extensions/TestFactoryRegistry.h>
#include <cppunit/ui/text/TestRunner.h>

int main(int argc, char** argv)
{
  CppUnit::TextUi::TestRunner runner;
  CppUnit::TestFactoryRegistry& registry = CppUnit::TestFactoryRegistry::getRegistry();
  runner.addTest(registry.makeTest());
  bool wasSuccessful = runner.run("", false);
  return (wasSuccessful ? 0 : 1);
}