hash_db/test/test1.cpp

Go to the documentation of this file.

/*
 * test_hash_db.cpp
 *
 * Copyright (C) 2006,2010  Thomas A. Vaughan
 * All rights reserved.
 *
 * Tests the hash_db::Database objects
 */

// includes --------------------------------------------------------------------
#include "hash_db/hash_db.h"

#include <iostream>

#include "perf/perf.h"


static const char * s_testDbFilename            = "test-db-AUTOGEN.txt";


////////////////////////////////////////////////////////////////////////////////
//
//      static helper methods
//
////////////////////////////////////////////////////////////////////////////////

static smart_ptr<Datahash>
getBogusHash
(
IN const char * text
)
{
        ASSERT(text, "null");

        smart_ptr<Datahash> h = Datahash::create();
        ASSERT(h, "Failed to create empty bogus hash");

        h->insert("foo", "bar");
        h->insert("foo", "bar2");
        h->insert("x", "y");
        h->insert("alpha", "beta");
        h->insert("text", text);

        return h;
}



static void
doTest
(
IN const char * filename
)
{
        ASSERT(filename, "null");

        smart_ptr<hash_db::Database> db = hash_db::Database::create(filename);
        ASSERT(db, "null hash");

        smart_ptr<Datahash> h1 = getBogusHash("first bogus hash");
        ASSERT(db->addObject("x", h1),
            "Should have succeeded");

        smart_ptr<Datahash> h2 = getBogusHash("second bogus hash");
        ASSERT(db->addObject("y", h2),
            "Should have succeeded");

        ASSERT(!db->addObject("y", h2),
            "Should have failed");

        SetString ids;
        db->getIds(ids);
        for (SetString::const_iterator i = ids.begin(); i != ids.end(); ++i) {
                DPRINTF("Hash database contains object with id='%s'",
                    i->c_str());
        }

        ASSERT(db->deleteObject("x"),
            "Should have succeeded");

        ASSERT(!db->deleteObject("x"),
            "Should have failed");

        db->commit();

        // TODO: need to automate multi-process access!
//      DPRINTF("Sleeping for 15 seconds so you can test multi-process access.");
//      sleep(15);
}



////////////////////////////////////////////////////////////////////////////////
//
//      entry point
//
////////////////////////////////////////////////////////////////////////////////

int
main
(
IN int argc,
IN const char * argv[]
)
{
        const char * filename = s_testDbFilename;

#ifdef WIN32
        DPRINTF("Quitting test early: hash_db not supported in Win32");
        return 0;
#endif  // WIN32

        int retval = 0;
        try {
                doTest(filename);

                // clean up the file
                std::string lockfile = filename;
                lockfile += ".lock";

                remove(filename);
                remove(lockfile.c_str());

        } catch (std::exception& e) {
                DPRINTF("Exception: %s", e.what());
                retval = 1;
        }

        perf::dumpTimingSummary(std::cerr);

        // all done
        return retval;
}