STXXL (GIT-copy)

/***************************************************************************
 *            test.cpp
 *
 *  Copyright 2007  Roman Dementiev
 *  dementiev@ira.uka.de
 ****************************************************************************/

#include <stxxl.h>
#include <cppunit/extensions/TestFactoryRegistry.h>
#include <cppunit/extensions/HelperMacros.h>
#include <cppunit/ui/text/TestRunner.h>

using namespace stxxl;

struct my_handler
{
    void operator ()  (request * ptr)
    {
        STXXL_MSG("Request completed: " << ptr);
    }
};


class IOLayerTest : public CppUnit::TestCase
{
    CPPUNIT_TEST_SUITE( IOLayerTest );
    CPPUNIT_TEST( testIO );
    CPPUNIT_TEST_EXCEPTION( testIOException, stxxl::io_error );
    CPPUNIT_TEST_SUITE_END();

public:
    IOLayerTest( std::string name ) : CppUnit::TestCase( name ) { }
    IOLayerTest() { }

    void testIO()
    {
        const int size = 1024 * 384;
        char * buffer = new char[size];
#ifdef BOOST_MSVC
        const char * paths[2] = { "data1", "data2" };
#else
        const char * paths[2] = { "/var/tmp/data1", "/var/tmp/data2" };
        mmap_file file1(paths[0], file::CREAT | file::RDWR, 0);
        file1.set_size(size * 1024);
#endif

        syscall_file file2(paths[1], file::CREAT | file::RDWR, 1);

        request_ptr req[16];
        unsigned i = 0;
        for ( ; i < 16; i++)
            req[i] = file2.awrite(buffer, i * size, size, my_handler());


        wait_all(req, 16);

        delete [] buffer;
    }

    void testIOException()
    {
        unlink("TestFile");
#ifdef BOOST_MSVC
        mmap_file file1("TestFile", file::RDWR, 0);
#else
        mmap_file file1("TestFile", file::RDWR, 0);
#endif
    }
};


CPPUNIT_TEST_SUITE_REGISTRATION( IOLayerTest );


int main()
{
    CppUnit::TextUi::TestRunner runner;
    CppUnit::TestFactoryRegistry &registry = CppUnit::TestFactoryRegistry::getRegistry();
    runner.addTest( registry.makeTest() );
    bool wasSuccessful = runner.run( "", false );
    return wasSuccessful;
}