/home/palakars/workspace-inprogress/code-in-progress/aitools3/aitools3-aq-invertedindex-cpp/test/reader/TestPostlistReader.cpp

Go to the documentation of this file.

#include <boost/test/unit_test.hpp>
#include "PostlistBuilder.hpp"
#include "PostlistWriter.hpp"
#include "PostlistReader.hpp"

using namespace aitools::invertedindex;

BOOST_AUTO_TEST_SUITE(postlist_reader);

void
write_postlist(PostlistWriter& writer, size_t num, size_t length, size_t esize)
{
        PostlistBuilder builder;
        ByteBuffer buffer(esize);
        for (unsigned i(0); i != length; ++i)
        {
                if (esize == 0) buffer.resize(i);
                builder.append(buffer);
        }
        Iterator::SharedPointer iterator(builder.build());
        for (unsigned i(0); i != num; ++i)
        {
                writer.write(iterator);
        }
}

void
read_postlist(PostlistReader& reader, size_t num, size_t length, size_t esize)
{
        BOOST_CHECK_EQUAL(reader.tell(), 0);

        std::string buffer;
        Iterator::SharedPointer iterator;
        for (unsigned i(0); i != num; ++i)
        {
                BOOST_CHECK(iterator = reader.read(length));
                BOOST_CHECK_EQUAL(iterator->length(), length);
                if (iterator->payload() > Iterator::max_chunk_size)
                {
                        BOOST_CHECK(iterator->is_paged());
                }
                for (unsigned j(0); j != length; ++j)
                {
                        BOOST_CHECK_EQUAL(iterator->size(), esize == 0 ? j : esize);
                        iterator->advance();
                }
        }
}

//BOOST_TEST_CASE(explicit_ctor)
//{
//      bfs::path path("/home/smartie/does/not/exist");
//      BOOST_CHECK_THROW(PostlistReader(path), std::invalid_argument);
//}

BOOST_AUTO_TEST_CASE(read_empty_postlist)
{
        size_t num(10);
        size_t length(0);
        size_t value_size(0);

        // write postlist
        bfs::path path("test.postlist.reader");
        PostlistWriter writer(path);
        write_postlist(writer, num, length, value_size);
        writer.close();
        // read postlist
        PostlistReader reader(path);
        read_postlist(reader, num, length, value_size);
        reader.close();
        bfs::remove(path);
}

BOOST_AUTO_TEST_CASE(read_non_paged_postlist_with_constant_value_size)
{
        size_t num(10);
        size_t length(100);
        size_t value_size(1);

        // write postlist
        bfs::path path("test.postlist.reader");
        PostlistWriter writer(path);
        write_postlist(writer, num, length, value_size);
        writer.close();
        // read postlist
        PostlistReader reader(path);
        read_postlist(reader, num, length, value_size);
        // read partial postlist
        reader.rewind();
        read_postlist(reader, num, length / 2, value_size);
        reader.close();
        bfs::remove(path);
}

BOOST_AUTO_TEST_CASE(read_non_paged_postlist_with_variable_value_size)
{
        size_t num(10);
        size_t length(100);
        size_t value_size(0);

        // write postlist
        bfs::path path("test.postlist.reader");
        PostlistWriter writer(path);
        write_postlist(writer, num, length, value_size);
        writer.close();
        // read postlist
        PostlistReader reader(path);
        read_postlist(reader, num, length, value_size);
        // read partial postlist
        reader.rewind();
        read_postlist(reader, num, length / 2, value_size);
        reader.close();
        bfs::remove(path);
        reader.close();
        bfs::remove(path);
}

BOOST_AUTO_TEST_CASE(read_paged_postlist_with_constant_value_size)
{
        size_t num(3);
        size_t length(100000);
        size_t value_size(110);

        // write postlist
        bfs::path path("test.postlist.reader");
        PostlistWriter writer(path);
        write_postlist(writer, num, length, value_size);
        writer.close();
        // read postlist
        PostlistReader reader(path);
        read_postlist(reader, num, length, value_size);
        // read partial postlist
        reader.rewind();
        read_postlist(reader, num, length / 2, value_size);
        reader.close();
        bfs::remove(path);
        reader.close();
        bfs::remove(path);
}

BOOST_AUTO_TEST_CASE(read_paged_postlist_with_variable_value_size)
{
        size_t num(3);
        size_t length(5000);
        size_t value_size(0);

        // write postlist
        bfs::path path("test.postlist.reader");
        PostlistWriter writer(path);
        write_postlist(writer, num, length, value_size);
        writer.close();
        // read postlist
        PostlistReader reader(path);
        read_postlist(reader, num, length, value_size);
        // read partial postlist
        reader.rewind();
        read_postlist(reader, num, length / 2, value_size);
        reader.close();
        bfs::remove(path);
        reader.close();
        bfs::remove(path);
}

BOOST_AUTO_TEST_SUITE_END();

---