110 lines
3.4 KiB
C++
110 lines
3.4 KiB
C++
/*
|
|
Custom Google Test assertions.
|
|
|
|
Copyright (c) 2012-2014, Victor Zverovich
|
|
All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are met:
|
|
|
|
1. Redistributions of source code must retain the above copyright notice, this
|
|
list of conditions and the following disclaimer.
|
|
2. Redistributions in binary form must reproduce the above copyright notice,
|
|
this list of conditions and the following disclaimer in the documentation
|
|
and/or other materials provided with the distribution.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
|
|
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
|
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "gtest-extra.h"
|
|
|
|
#if FMT_USE_FILE_DESCRIPTORS
|
|
|
|
using fmt::File;
|
|
|
|
void OutputRedirect::flush() {
|
|
#if EOF != -1
|
|
# error "FMT_RETRY assumes return value of -1 indicating failure"
|
|
#endif
|
|
int result = 0;
|
|
FMT_RETRY(result, fflush(file_));
|
|
if (result != 0)
|
|
throw fmt::SystemError(errno, "cannot flush stream");
|
|
}
|
|
|
|
void OutputRedirect::restore() {
|
|
if (original_.descriptor() == -1)
|
|
return; // Already restored.
|
|
flush();
|
|
// Restore the original file.
|
|
original_.dup2(FMT_POSIX(fileno(file_)));
|
|
original_.close();
|
|
}
|
|
|
|
OutputRedirect::OutputRedirect(FILE *file) : file_(file) {
|
|
flush();
|
|
int fd = FMT_POSIX(fileno(file));
|
|
// Create a File object referring to the original file.
|
|
original_ = File::dup(fd);
|
|
// Create a pipe.
|
|
File write_end;
|
|
File::pipe(read_end_, write_end);
|
|
// Connect the passed FILE object to the write end of the pipe.
|
|
write_end.dup2(fd);
|
|
}
|
|
|
|
OutputRedirect::~OutputRedirect() FMT_NOEXCEPT {
|
|
try {
|
|
restore();
|
|
} catch (const std::exception &e) {
|
|
std::fputs(e.what(), stderr);
|
|
}
|
|
}
|
|
|
|
std::string OutputRedirect::restore_and_read() {
|
|
// Restore output.
|
|
restore();
|
|
|
|
// Read everything from the pipe.
|
|
std::string content;
|
|
if (read_end_.descriptor() == -1)
|
|
return content; // Already read.
|
|
enum { BUFFER_SIZE = 4096 };
|
|
char buffer[BUFFER_SIZE];
|
|
std::size_t count = 0;
|
|
do {
|
|
count = read_end_.read(buffer, BUFFER_SIZE);
|
|
content.append(buffer, count);
|
|
} while (count != 0);
|
|
read_end_.close();
|
|
return content;
|
|
}
|
|
|
|
std::string read(File &f, std::size_t count) {
|
|
std::string buffer(count, '\0');
|
|
std::size_t n = 0, offset = 0;
|
|
do {
|
|
n = f.read(&buffer[offset], count - offset);
|
|
// We can't read more than size_t bytes since count has type size_t.
|
|
offset += static_cast<std::size_t>(n);
|
|
} while (offset < count && n != 0);
|
|
buffer.resize(offset);
|
|
return buffer;
|
|
}
|
|
|
|
#endif // FMT_USE_FILE_DESCRIPTORS
|
|
|
|
std::string format_system_error(int error_code, fmt::StringRef message) {
|
|
fmt::MemoryWriter out;
|
|
fmt::format_system_error(out, error_code, message);
|
|
return out.str();
|
|
}
|