torture_odb.cxx

Go to the documentation of this file.

//
// test_odb.cxx --- test ODB corner cases
//
// K.Olchanski
//
 
#include <stdio.h>
#include <sys/time.h>
#include <iostream>
#include <assert.h>
#include <signal.h>
#include <string.h>
#include <stdlib.h> // exit()
 
#include "tmfe.h"
#include "midasio.h"
#include "mvodb.h"
#include "midas.h"
 
std::string toString(int i)
{
   char buf[256];
   sprintf(buf, "%d", i);
   return buf;
}
 
void print_ia(const std::vector<int> &ia)
{
   int size = ia.size();
   printf("int[%d] has [", size);
   for (int i=0; i<size; i++) {
      if (i>0)
         printf(", ");
      printf("%d", ia[i]);
   }
   printf("]");
}
 
void print_da(const std::vector<double> &da)
{
   int size = da.size();
   printf("int[%d] has [", size);
   for (int i=0; i<size; i++) {
      if (i>0)
         printf(", ");
      printf("%f", da[i]);
   }
   printf("]");
}
 
static int gCountFail = 0;
 
void report_fail(const char* text)
{
   printf("FAIL: %s\n", text);
   gCountFail++;
}
 
static void test_depth_bomb(MVOdb* odb, bool cleanup)
{
   printf("Depth bomb: creating test_odb/depth_bomb/down/down/...\n");
 
   MVOdb *s = odb->Chdir("test_odb", true);
   s = s->Chdir("depth_bomb", true);
 
   assert(s);
 
   int level = 1;
 
   MVOdb *p = s;
 
   while (1) {
      p = p->Chdir("down", true);
      if (!p)
         break;
      if (p->IsReadOnly()) // detect NullOdb created when Chdir() failed to mkdir()
         break;
      level ++;
      //printf("%d.", level);
   }
 
   printf("Created tree %d deep\n", level);
 
   if (cleanup)
      odb->Delete("test_odb");
}
 
static void test_width_bomb_key(MVOdb* odb, bool cleanup)
{
   printf("Width bomb: creating test_odb/width_bomb_key/dir1,dir2,dir3,...\n");
 
   MVOdb *s = odb->Chdir("test_odb", true);
   s = s->Chdir("width_bomb_key", true);
 
   assert(s);
 
   int count = 1;
 
   MVOdb *p = s;
 
   while (1) {
      std::string name = msprintf("width_bomb_%06d", count);
 
      p = s->Chdir(name.c_str(), true);
      if (!p)
         break;
      if (p->IsReadOnly()) // detect NullOdb created when Chdir() failed to mkdir()
         break;
      count ++;
      //printf("%d.", count);
      //if (count > 10)
      //   break;
   }
 
   printf("Created %d subdirectories\n", count);
 
   if (cleanup)
      odb->Delete("test_odb");
}
 
static void test_width_bomb_data(MVOdb* odb, bool cleanup)
{
   printf("Width bomb: creating test_odb/width_bomb_data/dir1,dir2,dir3,...\n");
 
   MVOdb *s = odb->Chdir("test_odb", true);
   s = s->Chdir("width_bomb_data", true);
 
   assert(s);
 
   int count = 1;
 
   std::string data;
 
   for (int i=0; i<20; i++) {
      data += (char)('a' + i);
   }
 
   data += data;
   data += data;
 
   while (1) {
      std::string name = msprintf("width_data_%06d", count);
 
      s->WS(name.c_str(), data.c_str());
      count ++;
 
      std::string r;
      s->RS(name.c_str(), &r);
 
      if (r.empty())
         break;
 
      //printf("%d.", count);
      //if (count > 10)
      //   break;
   }
 
   printf("Created %d subdirectories\n", count);
 
   if (cleanup)
      odb->Delete("test_odb");
}
 
void all(MVOdb* odb)
{
   printf("\n");
   printf("Starting all tests:\n");
   printf("\n");
 
   test_depth_bomb(odb, true);
   test_width_bomb_key(odb, true);
   test_width_bomb_data(odb, true);
}
 
void help()
{
 
 
 
}
 
// Main function call
 
int main(int argc, char *argv[])
{
   setbuf(stdout,NULL);
   setbuf(stderr,NULL);
 
   signal(SIGILL,  SIG_DFL);
   signal(SIGBUS,  SIG_DFL);
   signal(SIGSEGV, SIG_DFL);
   signal(SIGPIPE, SIG_DFL);
 
   const char* hostname = NULL;
   const char* exptname = NULL;
 
   TMFE* mfe = NULL;
   
   MVOdb* odb = NULL;
   MVOdbError odberror;
 
   printf("Using TMFE ODB\n");
   mfe = TMFE::Instance();
   
   TMFeResult r;
   
   r = mfe->Connect("test_odb", hostname, exptname);
   if (r.error_flag) {
      fprintf(stderr,"Cannot connect to MIDAS, error %s\n", r.error_message.c_str());
      exit(1);
   }
 
   odb = MakeMidasOdb(mfe->fDB, &odberror);
 
   if (odberror.fError) {
      fprintf(stderr, "Cannot make MVOdb object, error: %s\n", odberror.fErrorString.c_str());
      exit(1);
   }
 
   for (int i=0; i<argc; i++) {
      std::string arg = argv[i];
 
      if (arg == "--all")
         all(odb);
      else if (arg == "--depth-bomb")
         test_depth_bomb(odb, false);
      else if (arg == "--width-bomb-key")
         test_width_bomb_key(odb, false);
      else if (arg == "--width-bomb-data")
         test_width_bomb_data(odb, false);
      else
         help();
   }
 
   if (mfe)
      mfe->Disconnect();
 
   return 0;
}
 
//end
/* emacs
 * Local Variables:
 * tab-width: 8
 * c-basic-offset: 3
 * indent-tabs-mode: nil
 * End:
 */