1// (C) Copyright 2008 CodeRage, LLC (turkanis at coderage dot com)
2// (C) Copyright 2004-2007 Jonathan Turkanis
3// Distributed under the Boost Software License, Version 1.0. (See accompanying
4// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt.)
5
6// See http://www.boost.org/libs/iostreams for documentation.
7
8#include <cctype>
9#include <boost/iostreams/copy.hpp>
10#include <boost/iostreams/device/file.hpp>
11#include <boost/iostreams/device/null.hpp>
12#include <boost/iostreams/filter/counter.hpp>
13#include <boost/iostreams/filtering_stream.hpp>
14#include <boost/test/test_tools.hpp>
15#include <boost/test/unit_test.hpp>
16#include "detail/constants.hpp"
17#include "detail/filters.hpp"
18#include "detail/temp_file.hpp"
19#include "detail/verification.hpp"
20
21using namespace std;
22using namespace boost;
23using namespace boost::iostreams;
24using namespace boost::iostreams::test;
25using boost::unit_test::test_suite;
26
27void read_counter()
28{
29 test_file src;
30 filtering_istream in;
31 in.push(t: counter());
32 in.push(t: padding_filter('a'), buffer_size: 0);
33 in.push(t: counter());
34 in.push(t: file_source(src.name(), in_mode));
35
36 counter* first_counter = BOOST_IOSTREAMS_COMPONENT(in, 0, counter);
37 counter* second_counter = BOOST_IOSTREAMS_COMPONENT(in, 2, counter);
38 int first_count = 0;
39 int second_count = 0;
40 int line_count = 0;
41 int reps = data_reps < 50 ? data_reps : 25; // Keep test short.
42 for (int w = 0; w < reps; ++w) {
43 int len = data_length();
44 for (int z = 0; z < len; ++z) {
45 in.get();
46 ++first_count;
47 ++second_count;
48 BOOST_CHECK(first_counter->characters() == first_count);
49 BOOST_CHECK(second_counter->characters() == second_count);
50 in.get();
51 ++first_count;
52 BOOST_CHECK(first_counter->characters() == first_count);
53 BOOST_CHECK(second_counter->characters() == second_count);
54 }
55 ++line_count;
56 BOOST_CHECK(first_counter->lines() == line_count);
57 BOOST_CHECK(first_counter->lines() == line_count);
58 }
59}
60
61void write_counter()
62{
63 filtering_ostream out;
64 out.push(t: counter());
65 out.push(t: padding_filter('a'), buffer_size: 0);
66 out.push(t: counter());
67 out.push(t: null_sink());
68
69 counter* first_counter = BOOST_IOSTREAMS_COMPONENT(out, 0, counter);
70 counter* second_counter = BOOST_IOSTREAMS_COMPONENT(out, 2, counter);
71 int first_count = 0;
72 int second_count = 0;
73 int line_count = 0;
74 int reps = data_reps < 50 ? data_reps : 25; // Keep test short.
75 const char* data = narrow_data();
76 for (int w = 0; w < reps; ++w) {
77 int len = data_length();
78 for (int z = 0; z < len; ++z) {
79 out.put(c: data[z]);
80 out.flush();
81 ++first_count;
82 second_count += 2;
83 BOOST_CHECK(first_counter->characters() == first_count);
84 BOOST_CHECK(second_counter->characters() == second_count);
85 }
86 ++line_count;
87 BOOST_CHECK(first_counter->lines() == line_count);
88 BOOST_CHECK(first_counter->lines() == line_count);
89 }
90}
91
92test_suite* init_unit_test_suite(int, char* [])
93{
94 test_suite* test = BOOST_TEST_SUITE("counter test");
95 test->add(BOOST_TEST_CASE(&read_counter));
96 test->add(BOOST_TEST_CASE(&write_counter));
97 return test;
98}
99

source code of boost/libs/iostreams/test/counter_test.cpp