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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
// (C) COPYRIGHT 2017 ARM Limited
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt.)
#include <boost/iostreams/categories.hpp> // tags.
#include <boost/iostreams/detail/adapter/non_blocking_adapter.hpp>
#include <boost/iostreams/filter/gzip.hpp>
#include <boost/iostreams/filtering_stream.hpp>
#include <boost/iostreams/write.hpp>
#include <boost/test/unit_test.hpp>
using namespace boost;
using namespace boost::iostreams;
using boost::unit_test::test_suite;
struct limit_device {
typedef char char_type;
typedef sink_tag category;
int written, overflow_count, limit;
explicit limit_device(int limit = 20) : written(0), overflow_count(0), limit(limit) {}
std::streamsize write(const char_type *, std::streamsize n)
{
if (written > limit) {
// first return partial writes, then an error
++overflow_count;
if (overflow_count > 2 || n < 2)
{
return -1;
}
n /= 2;
}
written += n;
return n;
}
};
static void disk_full_test()
{
// non_blocking_adapter used to handle write returning
// -1 correctly, usually hanging (see ticket 2557).
// As non_blocking_adapter is used for ofstream, this
// would happen for ordinary files when reaching quota,
// disk full or rlimits.
// Could be tested without gzip_compressor,
// but this tests a bit better that the whole path can handle it.
// TODO: should there be some check on error bit being set
// or should an exception be triggered?
limit_device outdev;
non_blocking_adapter<limit_device> nonblock_outdev(outdev);
filtering_ostream out;
out.push(gzip_compressor());
out.push(nonblock_outdev);
write(out, "teststring0123456789", 20);
out.flush();
write(out, "secondwrite123456789", 20);
close(out);
}
test_suite* init_unit_test_suite(int, char* [])
{
test_suite* test = BOOST_TEST_SUITE("disk full test");
test->add(BOOST_TEST_CASE(&disk_full_test));
return test;
}
|
