fmt/test/gtest-extra.h

162 lines
7.1 KiB
C
Raw Normal View History

2018-03-04 17:16:51 +00:00
// 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.
2014-05-01 23:49:08 +00:00
#ifndef FMT_GTEST_EXTRA_H_
#define FMT_GTEST_EXTRA_H_
2014-05-02 14:23:25 +00:00
2014-05-03 18:26:46 +00:00
#include <string>
#include "gmock.h"
2014-05-01 23:49:08 +00:00
#include "fmt/core.h"
2014-06-09 15:33:48 +00:00
#ifndef FMT_USE_FILE_DESCRIPTORS
2019-01-13 02:27:38 +00:00
# define FMT_USE_FILE_DESCRIPTORS 0
2014-06-09 15:33:48 +00:00
#endif
#if FMT_USE_FILE_DESCRIPTORS
2019-01-13 02:27:38 +00:00
# include "fmt/posix.h"
2014-06-09 15:33:48 +00:00
#endif
2014-05-01 23:49:08 +00:00
#define FMT_TEST_THROW_(statement, expected_exception, expected_message, fail) \
2019-01-13 02:27:38 +00:00
GTEST_AMBIGUOUS_ELSE_BLOCKER_ \
if (::testing::AssertionResult gtest_ar = ::testing::AssertionSuccess()) { \
std::string gtest_expected_message = expected_message; \
bool gtest_caught_expected = false; \
try { \
GTEST_SUPPRESS_UNREACHABLE_CODE_WARNING_BELOW_(statement); \
} catch (expected_exception const& e) { \
if (gtest_expected_message != e.what()) { \
gtest_ar << #statement \
" throws an exception with a different message.\n" \
<< "Expected: " << gtest_expected_message << "\n" \
<< " Actual: " << e.what(); \
goto GTEST_CONCAT_TOKEN_(gtest_label_testthrow_, __LINE__); \
} \
gtest_caught_expected = true; \
} catch (...) { \
gtest_ar << "Expected: " #statement \
" throws an exception of type " #expected_exception \
".\n Actual: it throws a different type."; \
goto GTEST_CONCAT_TOKEN_(gtest_label_testthrow_, __LINE__); \
} \
if (!gtest_caught_expected) { \
gtest_ar << "Expected: " #statement \
" throws an exception of type " #expected_exception \
".\n Actual: it throws nothing."; \
goto GTEST_CONCAT_TOKEN_(gtest_label_testthrow_, __LINE__); \
} \
} else \
GTEST_CONCAT_TOKEN_(gtest_label_testthrow_, __LINE__) \
: fail(gtest_ar.failure_message())
2014-05-01 23:49:08 +00:00
// Tests that the statement throws the expected exception and the exception's
// what() method returns expected message.
#define EXPECT_THROW_MSG(statement, expected_exception, expected_message) \
2019-01-13 02:27:38 +00:00
FMT_TEST_THROW_(statement, expected_exception, expected_message, \
GTEST_NONFATAL_FAILURE_)
2014-05-02 14:23:25 +00:00
std::string format_system_error(int error_code, fmt::string_view message);
2014-05-06 13:11:39 +00:00
#define EXPECT_SYSTEM_ERROR(statement, error_code, message) \
2019-01-13 02:27:38 +00:00
EXPECT_THROW_MSG(statement, fmt::system_error, \
format_system_error(error_code, message))
2014-05-06 13:11:39 +00:00
#if FMT_USE_FILE_DESCRIPTORS
2014-05-03 23:47:00 +00:00
// Captures file output by redirecting it to a pipe.
// The output it can handle is limited by the pipe capacity.
class OutputRedirect {
private:
2019-01-13 02:27:38 +00:00
FILE* file_;
2018-05-19 17:32:53 +00:00
fmt::file original_; // Original file passed to redirector.
fmt::file read_end_; // Read end of the pipe where the output is redirected.
GTEST_DISALLOW_COPY_AND_ASSIGN_(OutputRedirect);
void flush();
void restore();
2014-05-03 23:47:00 +00:00
public:
2019-01-13 02:27:38 +00:00
explicit OutputRedirect(FILE* file);
~OutputRedirect() FMT_NOEXCEPT;
2014-05-03 23:47:00 +00:00
// Restores the original file, reads output from the pipe into a string
// and returns it.
std::string restore_and_read();
};
2019-01-13 02:27:38 +00:00
# define FMT_TEST_WRITE_(statement, expected_output, file, fail) \
GTEST_AMBIGUOUS_ELSE_BLOCKER_ \
if (::testing::AssertionResult gtest_ar = ::testing::AssertionSuccess()) { \
std::string gtest_expected_output = expected_output; \
OutputRedirect gtest_redir(file); \
GTEST_SUPPRESS_UNREACHABLE_CODE_WARNING_BELOW_(statement); \
std::string gtest_output = gtest_redir.restore_and_read(); \
if (gtest_output != gtest_expected_output) { \
gtest_ar << #statement " produces different output.\n" \
<< "Expected: " << gtest_expected_output << "\n" \
<< " Actual: " << gtest_output; \
goto GTEST_CONCAT_TOKEN_(gtest_label_testthrow_, __LINE__); \
} \
} else \
GTEST_CONCAT_TOKEN_(gtest_label_testthrow_, __LINE__) \
: fail(gtest_ar.failure_message())
// Tests that the statement writes the expected output to file.
2019-01-13 02:27:38 +00:00
# define EXPECT_WRITE(file, statement, expected_output) \
FMT_TEST_WRITE_(statement, expected_output, file, GTEST_NONFATAL_FAILURE_)
2019-01-13 02:27:38 +00:00
# ifdef _MSC_VER
// Suppresses Windows assertions on invalid file descriptors, making
// POSIX functions return proper error codes instead of crashing on Windows.
class SuppressAssert {
private:
_invalid_parameter_handler original_handler_;
int original_report_mode_;
2019-01-13 02:27:38 +00:00
static void handle_invalid_parameter(const wchar_t*, const wchar_t*,
const wchar_t*, unsigned, uintptr_t) {}
public:
SuppressAssert()
2019-01-13 02:27:38 +00:00
: original_handler_(
_set_invalid_parameter_handler(handle_invalid_parameter)),
original_report_mode_(_CrtSetReportMode(_CRT_ASSERT, 0)) {}
~SuppressAssert() {
_set_invalid_parameter_handler(original_handler_);
_CrtSetReportMode(_CRT_ASSERT, original_report_mode_);
}
};
2019-01-13 02:27:38 +00:00
# define SUPPRESS_ASSERT(statement) \
{ \
SuppressAssert sa; \
statement; \
}
# else
# define SUPPRESS_ASSERT(statement) statement
# endif // _MSC_VER
2019-01-13 02:27:38 +00:00
# define EXPECT_SYSTEM_ERROR_NOASSERT(statement, error_code, message) \
EXPECT_SYSTEM_ERROR(SUPPRESS_ASSERT(statement), error_code, message)
// Attempts to read count characters from a file.
2019-01-13 02:27:38 +00:00
std::string read(fmt::file& f, std::size_t count);
2019-01-13 02:27:38 +00:00
# define EXPECT_READ(file, expected_content) \
EXPECT_EQ(expected_content, read(file, std::strlen(expected_content)))
#endif // FMT_USE_FILE_DESCRIPTORS
2019-01-13 02:27:38 +00:00
template <typename Mock> struct ScopedMock : testing::StrictMock<Mock> {
2016-05-19 22:04:25 +00:00
ScopedMock() { Mock::instance = this; }
2018-10-24 13:34:28 +00:00
~ScopedMock() { Mock::instance = FMT_NULL; }
2016-05-19 22:04:25 +00:00
};
#endif // FMT_GTEST_EXTRA_H_