| #include "util/stream/io.hh" |
| |
| #include "util/stream/stream.hh" |
| #include "util/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 |