summaryrefslogtreecommitdiffstats
path: root/src/seastar/fmt/test/gtest-extra.cc
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 18:24:20 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-27 18:24:20 +0000
commit483eb2f56657e8e7f419ab1a4fab8dce9ade8609 (patch)
treee5d88d25d870d5dedacb6bbdbe2a966086a0a5cf /src/seastar/fmt/test/gtest-extra.cc
parentInitial commit. (diff)
downloadceph-483eb2f56657e8e7f419ab1a4fab8dce9ade8609.tar.xz
ceph-483eb2f56657e8e7f419ab1a4fab8dce9ade8609.zip
Adding upstream version 14.2.21.upstream/14.2.21upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/seastar/fmt/test/gtest-extra.cc')
-rw-r--r--src/seastar/fmt/test/gtest-extra.cc90
1 files changed, 90 insertions, 0 deletions
diff --git a/src/seastar/fmt/test/gtest-extra.cc b/src/seastar/fmt/test/gtest-extra.cc
new file mode 100644
index 00000000..2e5ea3e3
--- /dev/null
+++ b/src/seastar/fmt/test/gtest-extra.cc
@@ -0,0 +1,90 @@
+// Formatting library for C++ - custom Google Test assertions
+//
+// Copyright (c) 2012 - present, Victor Zverovich
+// All rights reserved.
+//
+// For the license information refer to format.h.
+
+#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::system_error(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 *f) : file_(f) {
+ flush();
+ int fd = FMT_POSIX(fileno(f));
+ // 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 += 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::string_view message) {
+ fmt::memory_buffer out;
+ format_system_error(out, error_code, message);
+ return to_string(out);
+}