File size: 759 Bytes
1ce325b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
#include "io.hh"
#include "stream.hh"
#include "../file.hh"
#define BOOST_TEST_MODULE StreamTest
#include <boost/test/unit_test.hpp>
#include <unistd.h>
namespace util { namespace stream { namespace {
BOOST_AUTO_TEST_CASE(StreamTest) {
scoped_fd in(MakeTemp("io_test_temp"));
for (uint64_t i = 0; i < 100000; ++i) {
WriteOrThrow(in.get(), &i, sizeof(uint64_t));
}
SeekOrThrow(in.get(), 0);
ChainConfig config;
config.entry_size = 8;
config.total_memory = 100;
config.block_count = 12;
Stream s;
Chain chain(config);
chain >> Read(in.get()) >> s >> kRecycle;
uint64_t i = 0;
for (; s; ++s, ++i) {
BOOST_CHECK_EQUAL(i, *static_cast<const uint64_t*>(s.Get()));
}
BOOST_CHECK_EQUAL(100000ULL, i);
}
}}} // namespaces
|