lib/nstream/test/test.cpp

Go to the documentation of this file.

/*
 * test.cpp
 *
 * Copyright (C) 2009  Thomas A. Vaughan
 * All rights reserved.
 *
 * Test program for the nstream library.  See nstream.h
 */

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

#include <iostream>

#include "common/wave_ex.h"
#include "perf/perf.h"



static void * const s_magic             = (void * const) 0x5488943a;


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

static nstream::eIterationFlag
displayEntry
(
IN nstream::Entry * entry,
IN void * context
)
{
        ASSERT(entry, "null");
        ASSERT(s_magic == context, "context mismatch?");

        DPRINTF("  Entry: %s", entry->getName());

        return nstream::eIterate_Continue;
}



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

        smart_ptr<nstream::Manager> mgr = nstream::getFilesystemManager(dir);
        THROW(mgr, "failed to get manager for directory: " << dir);

        smart_ptr<nstream::Folder> root = mgr->getRoot();
        THROW(root, "failed to get root folder");

        DPRINTF("Iterating root for directory: %s", dir);
        smart_ptr<nstream::Entry> entry;
        while (entry = root->getNextChild()) {
                DPRINTF("  Entry: %s", entry->getName());
                switch (entry->getType()) {
                case nstream::Entry::eType_Folder:
                        DPRINTF("    Folder!");
                        break;

                case nstream::Entry::eType_File:
                        DPRINTF("    File!");
                        break;

                default:
                        {
                                WAVE_EX(wex);
                                wex << "Unknown type!  path: ";
                                wex << entry->getName();
                        }
                }
        }

        // recursively walk all entries
        DPRINTF("Recursive walk of '%s':", dir);
        nstream::walkChildFolders(root, displayEntry, s_magic);
}



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

int
main
(
IN int argc,
IN const char * argv[]
)
{
        ASSERT(2 == argc, "Usage: nstream-test <directory>");
        const char * dir = argv[1];

        try {
                perf::Timer timer("overall timer");
                doTest(dir);
        } catch (std::exception& e) {
                DPRINTF("Exception!  %s", e.what());
                return 1;
        }

        // all done
        perf::dumpTimingSummary(std::cout);
        return 0;
}