1 // Copyright 2011 The Kyua Authors.
2 // All rights reserved.
3 //
4 // Redistribution and use in source and binary forms, with or without
5 // modification, are permitted provided that the following conditions are
6 // met:
7 //
8 // * Redistributions of source code must retain the above copyright
9 // notice, this list of conditions and the following disclaimer.
10 // * Redistributions in binary form must reproduce the above copyright
11 // notice, this list of conditions and the following disclaimer in the
12 // documentation and/or other materials provided with the distribution.
13 // * Neither the name of Google Inc. nor the names of its contributors
14 // may be used to endorse or promote products derived from this software
15 // without specific prior written permission.
16 //
17 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
19 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
20 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
21 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
24 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
25 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
26 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
27 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
28
29 #include "utils/stream.hpp"
30
31 #include <cstdlib>
32 #include <sstream>
33
34 #include <atf-c++.hpp>
35
36 #include "utils/fs/path.hpp"
37
38 namespace fs = utils::fs;
39
40
41 ATF_TEST_CASE_WITHOUT_HEAD(open_ostream__stdout);
ATF_TEST_CASE_BODY(open_ostream__stdout)42 ATF_TEST_CASE_BODY(open_ostream__stdout)
43 {
44 const pid_t pid = atf::utils::fork();
45 if (pid == 0) {
46 std::auto_ptr< std::ostream > output = utils::open_ostream(
47 fs::path("/dev/stdout"));
48 (*output) << "Message to stdout\n";
49 output.reset();
50 std::exit(EXIT_SUCCESS);
51 }
52 atf::utils::wait(pid, EXIT_SUCCESS, "Message to stdout\n", "");
53 }
54
55
56 ATF_TEST_CASE_WITHOUT_HEAD(open_ostream__stderr);
ATF_TEST_CASE_BODY(open_ostream__stderr)57 ATF_TEST_CASE_BODY(open_ostream__stderr)
58 {
59 const pid_t pid = atf::utils::fork();
60 if (pid == 0) {
61 std::auto_ptr< std::ostream > output = utils::open_ostream(
62 fs::path("/dev/stderr"));
63 (*output) << "Message to stderr\n";
64 output.reset();
65 std::exit(EXIT_SUCCESS);
66 }
67 atf::utils::wait(pid, EXIT_SUCCESS, "", "Message to stderr\n");
68 }
69
70
71 ATF_TEST_CASE_WITHOUT_HEAD(open_ostream__other);
ATF_TEST_CASE_BODY(open_ostream__other)72 ATF_TEST_CASE_BODY(open_ostream__other)
73 {
74 const pid_t pid = atf::utils::fork();
75 if (pid == 0) {
76 std::auto_ptr< std::ostream > output = utils::open_ostream(
77 fs::path("some-file.txt"));
78 (*output) << "Message to other file\n";
79 output.reset();
80 std::exit(EXIT_SUCCESS);
81 }
82 atf::utils::wait(pid, EXIT_SUCCESS, "", "");
83 atf::utils::compare_file("some-file.txt", "Message to other file\n");
84 }
85
86
87 ATF_TEST_CASE_WITHOUT_HEAD(stream_length__empty);
ATF_TEST_CASE_BODY(stream_length__empty)88 ATF_TEST_CASE_BODY(stream_length__empty)
89 {
90 std::istringstream input("");
91 ATF_REQUIRE_EQ(0, utils::stream_length(input));
92 }
93
94
95 ATF_TEST_CASE_WITHOUT_HEAD(stream_length__some);
ATF_TEST_CASE_BODY(stream_length__some)96 ATF_TEST_CASE_BODY(stream_length__some)
97 {
98 const std::string contents(8192, 'x');
99 std::istringstream input(contents);
100 ATF_REQUIRE_EQ(
101 contents.length(),
102 static_cast< std::string::size_type >(utils::stream_length(input)));
103 }
104
105
106 ATF_TEST_CASE_WITHOUT_HEAD(read_file__ok);
ATF_TEST_CASE_BODY(read_file__ok)107 ATF_TEST_CASE_BODY(read_file__ok)
108 {
109 const char* contents = "These are\nsome file contents";
110 atf::utils::create_file("input.txt", contents);
111 ATF_REQUIRE_EQ(contents, utils::read_file(fs::path("input.txt")));
112 }
113
114
115 ATF_TEST_CASE_WITHOUT_HEAD(read_file__missing_file);
ATF_TEST_CASE_BODY(read_file__missing_file)116 ATF_TEST_CASE_BODY(read_file__missing_file)
117 {
118 ATF_REQUIRE_THROW_RE(std::runtime_error,
119 "Failed to open 'foo.txt' for read",
120 utils::read_file(fs::path("foo.txt")));
121 }
122
123
124 ATF_TEST_CASE_WITHOUT_HEAD(read_stream__empty);
ATF_TEST_CASE_BODY(read_stream__empty)125 ATF_TEST_CASE_BODY(read_stream__empty)
126 {
127 std::istringstream input("");
128 ATF_REQUIRE_EQ("", utils::read_stream(input));
129 }
130
131
132 ATF_TEST_CASE_WITHOUT_HEAD(read_stream__some);
ATF_TEST_CASE_BODY(read_stream__some)133 ATF_TEST_CASE_BODY(read_stream__some)
134 {
135 std::string contents;
136 for (int i = 0; i < 1000; i++)
137 contents += "abcdef";
138 std::istringstream input(contents);
139 ATF_REQUIRE_EQ(contents, utils::read_stream(input));
140 }
141
142
ATF_INIT_TEST_CASES(tcs)143 ATF_INIT_TEST_CASES(tcs)
144 {
145 ATF_ADD_TEST_CASE(tcs, open_ostream__stdout);
146 ATF_ADD_TEST_CASE(tcs, open_ostream__stderr);
147 ATF_ADD_TEST_CASE(tcs, open_ostream__other);
148
149 ATF_ADD_TEST_CASE(tcs, stream_length__empty);
150 ATF_ADD_TEST_CASE(tcs, stream_length__some);
151
152 ATF_ADD_TEST_CASE(tcs, read_file__ok);
153 ATF_ADD_TEST_CASE(tcs, read_file__missing_file);
154
155 ATF_ADD_TEST_CASE(tcs, read_stream__empty);
156 ATF_ADD_TEST_CASE(tcs, read_stream__some);
157 }
158