#include "test_assert.h"

#include <assert.h>

#ifdef _MSC_VER

#  include <crtdbg.h>

#  include <windows.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("EXPECTED: \"%s\"", expval);

  TEST_OUTPUT_LINE("VALUE: \"%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, const char *func) {

  if (strcmp(expval, val) != 0) {

    TestFail(expval, val, exp, file, line, func);

  }

}

#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, nullptr, _IONBF, 0);

  setvbuf(stderr, nullptr, _IONBF, 0);

  // clang-format off

  #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);

}