testBank.cpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:4k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2003 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. #include <NDBT.hpp>
  14. #include <NDBT_Test.hpp>
  15. #include <HugoTransactions.hpp>
  16. #include <UtilTransactions.hpp>
  17. #include <NdbRestarter.hpp>
  18. #include <NdbBackup.hpp>
  19. #define CHECK(b) if (!(b)) { 
  20.   g_err << "ERR: "<< step->getName() 
  21.          << " failed on line " << __LINE__ << endl; 
  22.   result = NDBT_FAILED; 
  23.   continue; } 
  24. #include "Bank.hpp"
  25. int runCreateBank(NDBT_Context* ctx, NDBT_Step* step){
  26.   Bank bank;
  27.   int overWriteExisting = true;
  28.   if (bank.createAndLoadBank(overWriteExisting) != NDBT_OK)
  29.     return NDBT_FAILED;
  30.   return NDBT_OK;
  31. }
  32. int runBankTimer(NDBT_Context* ctx, NDBT_Step* step){
  33.   Bank bank;
  34.   int wait = 30; // Max seconds between each "day"
  35.   int yield = 1; // Loops before bank returns 
  36.   while (ctx->isTestStopped() == false) {
  37.     bank.performIncreaseTime(wait, yield);
  38.   }
  39.   return NDBT_OK;
  40. }
  41. int runBankTransactions(NDBT_Context* ctx, NDBT_Step* step){
  42.   Bank bank;
  43.   int wait = 10; // Max ms between each transaction
  44.   int yield = 100; // Loops before bank returns 
  45.   while (ctx->isTestStopped() == false) {
  46.     bank.performTransactions(wait, yield);
  47.   }
  48.   return NDBT_OK;
  49. }
  50. int runBankGL(NDBT_Context* ctx, NDBT_Step* step){
  51.   Bank bank;
  52.   int yield = 20; // Loops before bank returns 
  53.   int result = NDBT_OK;
  54.   while (ctx->isTestStopped() == false) {
  55.     if (bank.performMakeGLs(yield) != NDBT_OK){
  56.       ndbout << "bank.performMakeGLs FAILED" << endl;
  57.       result = NDBT_FAILED;
  58.     }
  59.   }
  60.   return NDBT_OK;
  61. }
  62. int runBankSum(NDBT_Context* ctx, NDBT_Step* step){
  63.   Bank bank;
  64.   int wait = 2000; // Max ms between each sum of accounts
  65.   int yield = 1; // Loops before bank returns 
  66.   int result = NDBT_OK;
  67.   while (ctx->isTestStopped() == false) {
  68.     if (bank.performSumAccounts(wait, yield) != NDBT_OK){
  69.       ndbout << "bank.performSumAccounts FAILED" << endl;
  70.       result = NDBT_FAILED;
  71.     }
  72.   }
  73.   return result ;
  74. }
  75. int runDropBank(NDBT_Context* ctx, NDBT_Step* step){
  76.   Bank bank;
  77.   if (bank.dropBank() != NDBT_OK)
  78.     return NDBT_FAILED;
  79.   return NDBT_OK;
  80. }
  81. int runBankController(NDBT_Context* ctx, NDBT_Step* step){
  82.   Ndb* pNdb = GETNDB(step);
  83.   int loops = ctx->getNumLoops();
  84.   int records = ctx->getNumRecords();
  85.   int l = 0;
  86.   int result = NDBT_OK;
  87.   while (l < loops && result != NDBT_FAILED){
  88.     if (pNdb->waitUntilReady() != 0){
  89.       result = NDBT_FAILED;
  90.       continue;
  91.     }
  92.     // Sleep for a while
  93.     NdbSleep_SecSleep(records);
  94.     
  95.     l++;
  96.   }
  97.   if (pNdb->waitUntilReady() != 0){
  98.     result = NDBT_FAILED;
  99.   }
  100.   ctx->stopTest();
  101.   return result;
  102. }
  103. NDBT_TESTSUITE(testBank);
  104. TESTCASE("Bank", 
  105.  "Run the bankn"){
  106.   INITIALIZER(runCreateBank);
  107.   STEP(runBankTimer);
  108.   STEP(runBankTransactions);
  109.   STEP(runBankGL);
  110.   // TODO  STEP(runBankSum);
  111.   STEP(runBankController);
  112.   FINALIZER(runDropBank);
  113. }
  114. NDBT_TESTSUITE_END(testBank);
  115. int main(int argc, const char** argv){
  116.   ndb_init();
  117.   // Tables should not be auto created
  118.   testBank.setCreateTable(false);
  119.   return testBank.execute(argc, argv);
  120. }