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
38namespace fs = utils::fs;
39
40
41ATF_TEST_CASE_WITHOUT_HEAD(open_ostream__stdout);
42ATF_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
56ATF_TEST_CASE_WITHOUT_HEAD(open_ostream__stderr);
57ATF_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
71ATF_TEST_CASE_WITHOUT_HEAD(open_ostream__other);
72ATF_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
87ATF_TEST_CASE_WITHOUT_HEAD(stream_length__empty);
88ATF_TEST_CASE_BODY(stream_length__empty)
89{
90    std::istringstream input("");
91    ATF_REQUIRE_EQ(0, utils::stream_length(input));
92}
93
94
95ATF_TEST_CASE_WITHOUT_HEAD(stream_length__some);
96ATF_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
106ATF_TEST_CASE_WITHOUT_HEAD(read_file__ok);
107ATF_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
115ATF_TEST_CASE_WITHOUT_HEAD(read_file__missing_file);
116ATF_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
124ATF_TEST_CASE_WITHOUT_HEAD(read_stream__empty);
125ATF_TEST_CASE_BODY(read_stream__empty)
126{
127    std::istringstream input("");
128    ATF_REQUIRE_EQ("", utils::read_stream(input));
129}
130
131
132ATF_TEST_CASE_WITHOUT_HEAD(read_stream__some);
133ATF_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
143ATF_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