mariadb-columnstore-engine/oamapps/sessionWalker/tdriver.cpp

/* 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. */

/***************************************************************************
 *   jrodriguez@calpont.com   *
 *                                                                         *
 ***************************************************************************/

using namespace std;
#include <iostream>
#include <cstdlib>
#include <sys/types.h>
#include <unistd.h>
#include "sessionmonitor.h"
#include "sessionmanager.h"

#include <cppunit/extensions/HelperMacros.h>

using namespace execplan;

int maxNewTxns = 1000;
int maxTxns = 1000;

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

  CPPUNIT_TEST(MonitorTestPlan_1);
  CPPUNIT_TEST_SUITE_END();

 private:
 public:
  void setUp()
  {
  }

  void tearDown()
  {
  }

  int verifyLen;
  SessionManager* manager;
  SessionManager::TxnID managerTxns[1000];

  int createTxns(const int& start, const int& end)
  {
    int first = start;
    int last = end;
    int newTxns = 0;

    verifyLen = manager->verifySize();

    for (int idx = first; idx < last && verifyLen < maxNewTxns; idx++)
    {
      managerTxns[idx] = manager->newTxnID((uint32_t)idx + 1000);
      CPPUNIT_ASSERT(managerTxns[idx].id > 0);
      CPPUNIT_ASSERT(managerTxns[idx].valid == true);
      verifyLen = manager->verifySize();
      CPPUNIT_ASSERT(verifyLen > 0);
      newTxns++;
    }

    CPPUNIT_ASSERT(newTxns == last - first);
    return newTxns;
  }

  int closeTxns(const int& start, const int& end)
  {
    int first = start;
    int last = end;
    int totalClosed = 0;

    for (int idx = first; idx < last; idx++)
    {
      try
      {
        SessionManager::TxnID tmp = manager->getTxnID(idx + 1000);

        if (tmp.valid == true)
        {
          manager->committed(tmp);
          CPPUNIT_ASSERT(tmp.valid == false);
          totalClosed++;
        }
      }
      catch (exception& e)
      {
        cerr << e.what() << endl;
        continue;
      }
    }

    return totalClosed;

  }  // closeTxns

  void MonitorTestPlan_1()
  {
    int currStartTxn = 0;
    int currEndTxn = 5;
    int txnCntIncr = 5;
    const int sleepTime = 1;
    const int iterMax = 1;
    vector<SessionMonitor::MonSIDTIDEntry*> toTxns;

    manager = new SessionManager();
    // CPPUNIT_ASSERT(manager->verifySize()==0);

    SessionMonitor* monitor = NULL;

    for (int jdx = 0; jdx < iterMax; jdx++)
    {
      // store the current state of the SessionManager
      monitor = new SessionMonitor();
      monitor->AgeLimit(sleepTime);
      delete monitor;
      int idx = 0;
      int grpStart = currStartTxn;

      for (idx = 0; idx < 3; idx++)
      {
        createTxns(currStartTxn, currEndTxn);
        // CPPUNIT_ASSERT(manager->verifySize()==(idx+1)*txnCntIncr);

        currStartTxn += txnCntIncr;
        currEndTxn += txnCntIncr;
        sleep(sleepTime + 1);  // make sessions time out

        monitor = new SessionMonitor();  // read Monitor data
        monitor->AgeLimit(sleepTime);
        toTxns.clear();
        toTxns = monitor->timedOutTxns();  // get timed out txns
        CPPUNIT_ASSERT(toTxns.size() == (uint32_t)txnCntIncr * idx);

        delete monitor;
      }

      int grpEnd = currEndTxn;
      monitor = new SessionMonitor();
      monitor->AgeLimit(sleepTime);
      closeTxns(grpStart, grpEnd);  // close this iteration of txns
      // CPPUNIT_ASSERT(manager->verifySize()==0);
      toTxns = monitor->timedOutTxns();  // get timed out txns
      CPPUNIT_ASSERT(toTxns.size() == 0);

      delete monitor;
    }

    monitor = new SessionMonitor();  // readload Monitor data
    monitor->AgeLimit(sleepTime - 1);

    toTxns.clear();
    toTxns = monitor->timedOutTxns();  // get timed out txns
    CPPUNIT_ASSERT(toTxns.size() == 0);
    delete monitor;

    // CPPUNIT_ASSERT(manager->verifySize()==0);
    delete manager;
  }

};  // test suite

CPPUNIT_TEST_SUITE_REGISTRATION(ExecPlanTest);

#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);
}