sbuffer-test.cpp

Go to the documentation of this file.

/*
 * sbuffer-test.cpp
 *
 * Copyright (C) 2010  Thomas A. Vaughan
 * All rights reserved.
 *
 * Quick test of string buffer object
 */

// includes --------------------------------------------------------------------
#include <iostream>
#include <sstream>

#include "perf/perf.h"
#include "util/string-buffer.h"


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

static void
doTest
(
IN int nStrings
)
{
        ASSERT(nStrings > 0, "Bad string count: %d", nStrings);

        smart_ptr<StringBuffer> sbuf = StringBuffer::create();
        ASSERT(sbuf, "failed to create string buffer?");

        const int minLen = 0;
        const int maxLen = 100;
        const int modulo = maxLen - minLen + 1;
        char buffer[maxLen + 1];
        for (int i = 0; i < nStrings; ++i) {
                int len = minLen + (rand() % modulo);
                for (int j = 0; j < len; ++j) {
                        buffer[j] = 'a' + (rand() % 26);
                }
                buffer[len] = 0;        // null-terminated

//              DPRINTF("String[%3d] = %s", i + 1, buffer);
                int offset = sbuf->appendString(buffer);
//              DPRINTF("  offset = %d", offset);

                ASSERT_THROW(!strcmp(buffer, sbuf->getString(offset)),
                    "Strings should match!");
        }

        if (nStrings > 20) {
                DPRINTF("More than 20 strings, not writing them all.");
        } else {
                DPRINTF("all strings:");
                sbuf->dumpStrings(std::cout);
        }
}



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

int
main
(
IN int argc,
IN const char * argv[]
)
{
        int retval = 0;
        DPRINTF("Usage: util-sbuffer-test [nStrings]");
        int nStrings = 5000;
        if (argc > 1) {
                nStrings = atoi(argv[1]);
                DPRINTF("Using %d strings", nStrings);
        } else {
                DPRINTF("Using default of %d strings", nStrings);
        }

        try {
                perf::Timer timer("overall timer");

                doTest(nStrings);

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

        perf::dumpTimingSummary(std::cerr);

        return retval;
}