blob: 82108335ace5db14310785fae63438715934e400 (
plain)
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
37
38
|
#include "util/stream/io.hh"
#include "util/stream/chain.hh"
#include "util/file.hh"
#define BOOST_TEST_MODULE IOTest
#include <boost/test/unit_test.hpp>
#include <unistd.h>
namespace util { namespace stream { namespace {
BOOST_AUTO_TEST_CASE(CopyFile) {
std::string temps("io_test_temp");
scoped_fd in(MakeTemp(temps));
for (uint64_t i = 0; i < 100000; ++i) {
WriteOrThrow(in.get(), &i, sizeof(uint64_t));
}
SeekOrThrow(in.get(), 0);
scoped_fd out(MakeTemp(temps));
ChainConfig config;
config.entry_size = 8;
config.total_memory = 1024;
config.block_count = 10;
Chain(config) >> PRead(in.get()) >> Write(out.get());
SeekOrThrow(out.get(), 0);
for (uint64_t i = 0; i < 100000; ++i) {
uint64_t got;
ReadOrThrow(out.get(), &got, sizeof(uint64_t));
BOOST_CHECK_EQUAL(i, got);
}
}
}}} // namespaces
|
