#include "test_assert.h"
|
|
#include <assert.h>
|
|
#ifdef _MSC_VER
|
# include <crtdbg.h>
|
#endif
|
|
int testing_fails = 0;
|
static TestFailEventListener fail_listener_ = nullptr;
|
|
void TestFail(const char *expval, const char *val, const char *exp,
|
const char *file, int line, const char *func) {
|
TEST_OUTPUT_LINE("VALUE: \"%s\"", expval);
|
TEST_OUTPUT_LINE("EXPECTED: \"%s\"", val);
|
TEST_OUTPUT_LINE("TEST FAILED: %s:%d, %s in %s", file, line, exp,
|
func ? func : "");
|
testing_fails++;
|
|
// Notify, emulate 'gtest::OnTestPartResult' event handler.
|
if (fail_listener_) (*fail_listener_)(expval, val, exp, file, line, func);
|
|
assert(0); // ignored in Release if NDEBUG defined
|
}
|
|
void TestEqStr(const char *expval, const char *val, const char *exp,
|
const char *file, int line) {
|
if (strcmp(expval, val) != 0) { TestFail(expval, val, exp, file, line); }
|
}
|
|
#if defined(FLATBUFFERS_MEMORY_LEAK_TRACKING) && defined(_MSC_VER) && \
|
defined(_DEBUG)
|
#define FLATBUFFERS_MEMORY_LEAK_TRACKING_MSVC
|
#endif
|
|
void InitTestEngine(TestFailEventListener listener) {
|
testing_fails = 0;
|
// Disable stdout buffering to prevent information lost on assertion or core
|
// dump.
|
setvbuf(stdout, NULL, _IONBF, 0);
|
setvbuf(stderr, NULL, _IONBF, 0);
|
|
// clang-format off
|
|
#ifdef _MSC_VER
|
// Send all reports to STDOUT.
|
// CrtDebug reports to _CRT_WARN channel.
|
_CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE | _CRTDBG_MODE_DEBUG);
|
_CrtSetReportFile(_CRT_WARN, _CRTDBG_FILE_STDOUT);
|
// The assert from <assert.h> reports to _CRT_ERROR channel
|
_CrtSetReportMode(_CRT_ERROR, _CRTDBG_MODE_FILE | _CRTDBG_MODE_DEBUG);
|
_CrtSetReportFile(_CRT_ERROR, _CRTDBG_FILE_STDOUT);
|
// Internal CRT assert channel?
|
_CrtSetReportMode(_CRT_ASSERT, _CRTDBG_MODE_FILE | _CRTDBG_MODE_DEBUG);
|
_CrtSetReportFile(_CRT_ASSERT, _CRTDBG_FILE_STDOUT);
|
#endif
|
|
#if defined(FLATBUFFERS_MEMORY_LEAK_TRACKING_MSVC)
|
// For more thorough checking:
|
// _CRTDBG_DELAY_FREE_MEM_DF | _CRTDBG_CHECK_ALWAYS_DF
|
auto flags = _CrtSetDbgFlag(_CRTDBG_REPORT_FLAG);
|
_CrtSetDbgFlag(flags | _CRTDBG_ALLOC_MEM_DF);
|
#endif
|
// clang-format on
|
|
fail_listener_ = listener;
|
}
|
|
int CloseTestEngine(bool force_report) {
|
if (!testing_fails || force_report) {
|
#if defined(FLATBUFFERS_MEMORY_LEAK_TRACKING_MSVC)
|
auto flags = _CrtSetDbgFlag(_CRTDBG_REPORT_FLAG);
|
flags &= ~_CRTDBG_DELAY_FREE_MEM_DF;
|
flags |= _CRTDBG_LEAK_CHECK_DF;
|
_CrtSetDbgFlag(flags);
|
#endif
|
}
|
return (0 != testing_fails);
|
}
|
