• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 #include "test_assert.h"
2 
3 #include <assert.h>
4 
5 #ifdef _MSC_VER
6 #  include <crtdbg.h>
7 #  include <windows.h>
8 #endif
9 
10 int testing_fails = 0;
11 static TestFailEventListener fail_listener_ = nullptr;
12 
TestFail(const char * expval,const char * val,const char * exp,const char * file,int line,const char * func)13 void TestFail(const char *expval, const char *val, const char *exp,
14               const char *file, int line, const char *func) {
15   TEST_OUTPUT_LINE("EXPECTED: \"%s\"", expval);
16   TEST_OUTPUT_LINE("VALUE: \"%s\"", val);
17   TEST_OUTPUT_LINE("TEST FAILED: %s:%d, %s in %s", file, line, exp,
18                    func ? func : "");
19   testing_fails++;
20 
21   // Notify, emulate 'gtest::OnTestPartResult' event handler.
22   if (fail_listener_) (*fail_listener_)(expval, val, exp, file, line, func);
23 
24   assert(0);  // ignored in Release if NDEBUG defined
25 }
26 
TestEqStr(const char * expval,const char * val,const char * exp,const char * file,int line,const char * func)27 void TestEqStr(const char *expval, const char *val, const char *exp,
28                const char *file, int line, const char *func) {
29   if (strcmp(expval, val) != 0) {
30     TestFail(expval, val, exp, file, line, func);
31   }
32 }
33 
34 #if defined(FLATBUFFERS_MEMORY_LEAK_TRACKING) && defined(_MSC_VER) && \
35     defined(_DEBUG)
36 #  define FLATBUFFERS_MEMORY_LEAK_TRACKING_MSVC
37 #endif
38 
InitTestEngine(TestFailEventListener listener)39 void InitTestEngine(TestFailEventListener listener) {
40   testing_fails = 0;
41   // Disable stdout buffering to prevent information lost on assertion or core
42   // dump.
43   setvbuf(stdout, NULL, _IONBF, 0);
44   setvbuf(stderr, NULL, _IONBF, 0);
45 
46   flatbuffers::SetupDefaultCRTReportMode();
47 
48   // clang-format off
49 
50   #if defined(FLATBUFFERS_MEMORY_LEAK_TRACKING_MSVC)
51     // For more thorough checking:
52     // _CRTDBG_DELAY_FREE_MEM_DF | _CRTDBG_CHECK_ALWAYS_DF
53     auto flags = _CrtSetDbgFlag(_CRTDBG_REPORT_FLAG);
54     _CrtSetDbgFlag(flags | _CRTDBG_ALLOC_MEM_DF);
55   #endif
56   // clang-format on
57 
58   fail_listener_ = listener;
59 }
60 
CloseTestEngine(bool force_report)61 int CloseTestEngine(bool force_report) {
62   if (!testing_fails || force_report) {
63 #if defined(FLATBUFFERS_MEMORY_LEAK_TRACKING_MSVC)
64     auto flags = _CrtSetDbgFlag(_CRTDBG_REPORT_FLAG);
65     flags &= ~_CRTDBG_DELAY_FREE_MEM_DF;
66     flags |= _CRTDBG_LEAK_CHECK_DF;
67     _CrtSetDbgFlag(flags);
68 #endif
69   }
70   return (0 != testing_fails);
71 }
72