t_huge_file.cpp

Go to the documentation of this file.

/*****************************************************************************\
*                                                                             *
*  Name   : test_huge_file                                                    *
*  Author : Chris Koeritz                                                     *
*                                                                             *
*******************************************************************************
* Copyright (c) 1991-$now By Author.  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; either version 2 of   *
* the License or (at your option) any later version.  This is online at:      *
*     http://www.fsf.org/copyleft/gpl.html                                    *
* Please send any updates to: fred@gruntose.com                               *
\*****************************************************************************/

#include <basis/byte_array.h>
#include <basis/chaos.h>
#include <basis/function.h>
#include <basis/guards.h>
#include <basis/istring.h>
#include <basis/portable.h>
#include <basis/string_array.h>
#include <opsystem/application_shell.h>
#include <loggers/console_logger.h>
#include <opsystem/directory.h>
#include <loggers/file_logger.h>
#include <opsystem/filename.h>
#include <opsystem/huge_file.h>
#include <data_struct/static_memory_gremlin.h>

#define LOG(s) CLASS_EMERGENCY_LOG(program_wide_logger(), s)

class test_huge_file : public application_shell
{
public:
  test_huge_file() : application_shell(class_name()) {}
  IMPLEMENT_CLASS_NAME("test_huge_file");
  int run_file_scan();
  virtual int execute();
};

int test_huge_file::run_file_scan()
{
  FUNCDEF("run_file_scan");
  chaos randomizer;

  string_array files(__argc, (const char **)__argv);
  files.zap(0, 0);  // toss the first element since that's our app filename.

  if (!files.length()) {
    // pretend they gave us the list of files in the TMP directory.  some of
    // these might fail if they're locked up.
//    istring tmpdir = portable::env_string("TMP");
    istring tmpdir = portable::current_directory();
    directory dir(tmpdir);
    for (int i = 0; i < dir.files().length(); i++) {
      // skip text files since we use those right here.
      if (dir.files()[i].ends(".txt"))
        continue;
      istring chewed_string = tmpdir + "/" + dir.files()[i];
      files += chewed_string;
    }
LOG(istring("added files since no cmd args: ") + files.text_form());
  }

  byte_array data_found;
  for (int i = 0; i < files.length(); i++) {
    istring curr = files[i];
    LOG(isprintf("file %d: ", i) + curr);
    huge_file test(curr, "rb");
    if (!test.good())
      LOG(istring("good check: ") + curr + " cannot be opened.");
    double len = test.length();
log(isprintf("file len is %.0f", len));
    double posn = 0;
    while ( (posn < len) && !test.eof() ) {
      int readlen = randomizer.inclusive(1, 256 * KILOBYTE);
log(isprintf("read %.0f bytes, posn now %.0f bytes", double(readlen), posn));
      int bytes_read = 0;
      outcome ret = test.read(data_found, readlen, bytes_read);
      if (ret != huge_file::OKAY) {
        deadly_error(class_name(), "reading", curr + " failed to be read.");
      } else {
        posn += bytes_read;
      }
    }
    if (!test.eof())
      deadly_error(class_name(), "eof check", curr + " not at eof.");
    if (posn != len)
      deadly_error(class_name(), "eof check", curr + " is at wrong position: "
          + isprintf("want %.0f, got %.0f", double(len), double(posn)));
    log(istring("successfully read ") + curr);
  }

  return 0;
}

int test_huge_file::execute()
{
  FUNCDEF("execute");
  SET_DEFAULT_COMBO_LOGGER;
  int ret = run_file_scan();
  if (ret) return ret;  // failed here.

  guards::alert_message("huge_file:: works for those functions tested.\n");
  return 0;
}

HOOPLE_MAIN(test_huge_file, )

---