/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-

 * * This Source Code Form is subject to the terms of the Mozilla Public

 * * License, v. 2.0. If a copy of the MPL was not distributed with this

 * * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "GTestRunner.h"

#include "gtest/gtest.h"

#include "mozilla/Attributes.h"

#include "nsICrashReporter.h"

#include "testing/TestHarness.h"

#include "prenv.h"

#ifdef XP_WIN

#include "mozilla/ipc/WindowsMessageLoop.h"

#endif

using ::testing::EmptyTestEventListener;

using ::testing::InitGoogleTest;

using ::testing::Test;

using ::testing::TestCase;

using ::testing::TestEventListeners;

using ::testing::TestInfo;

using ::testing::TestPartResult;

using ::testing::UnitTest;

namespace mozilla {

// See gtest.h for method documentation

class MozillaPrinter : public EmptyTestEventListener

{

public:

  virtual void OnTestProgramStart(const UnitTest& /* aUnitTest */) override {

    printf("TEST-INFO | GTest unit test starting\n");

  }

  virtual void OnTestProgramEnd(const UnitTest& aUnitTest) override {

    printf("TEST-%s | GTest unit test: %s\n",

           aUnitTest.Passed() ? "PASS" : "UNEXPECTED-FAIL",

           aUnitTest.Passed() ? "passed" : "failed");

    printf("Passed: %d\n", aUnitTest.successful_test_count());

    printf("Failed: %d\n", aUnitTest.failed_test_count());

  }

  virtual void OnTestStart(const TestInfo& aTestInfo) override {

    mTestInfo = &aTestInfo;

    printf("TEST-START | %s.%s\n",

        mTestInfo->test_case_name(), mTestInfo->name());

  }

  virtual void OnTestPartResult(const TestPartResult& aTestPartResult) override {

    printf("TEST-%s | %s.%s | %s @ %s:%i\n",

           !aTestPartResult.failed() ? "PASS" : "UNEXPECTED-FAIL",

           mTestInfo ? mTestInfo->test_case_name() : "?", mTestInfo ? mTestInfo->name() : "?",

           aTestPartResult.summary(),

           aTestPartResult.file_name(), aTestPartResult.line_number());

  }

  virtual void OnTestEnd(const TestInfo& aTestInfo) override {

    printf("TEST-%s | %s.%s | test completed (time: %llims)\n",

           aTestInfo.result()->Passed() ? "PASS": "UNEXPECTED-FAIL",

           aTestInfo.test_case_name(), aTestInfo.name(),

           aTestInfo.result()->elapsed_time());

    MOZ_ASSERT(&aTestInfo == mTestInfo);

    mTestInfo = nullptr;

  }

  const TestInfo* mTestInfo;

};

static void ReplaceGTestLogger()

{

  // Replace the GTest logger so that it can be passed

  // by the mozilla test parsers.

  // Code is based on: http://googletest.googlecode.com/svn/trunk/samples/sample9_unittest.cc

  UnitTest& unitTest = *UnitTest::GetInstance();

  TestEventListeners& listeners = unitTest.listeners();

  delete listeners.Release(listeners.default_result_printer());

  listeners.Append(new MozillaPrinter);

}

int RunGTestFunc(int* argc, char** argv)

{

  InitGoogleTest(argc, argv);

  if (getenv("MOZ_TBPL_PARSER")) {

    ReplaceGTestLogger();

  }

  PR_SetEnv("XPCOM_DEBUG_BREAK=stack-and-abort");

  ScopedXPCOM xpcom("GTest");

#ifdef XP_WIN

  mozilla::ipc::windows::InitUIThread();

#endif

  nsCOMPtr<nsICrashReporter> crashreporter;

  char *crashreporterStr = PR_GetEnv("MOZ_CRASHREPORTER");

  if (crashreporterStr && !strcmp(crashreporterStr, "1")) {

    //TODO: move this to an even-more-common location to use in all

    // C++ unittests

    crashreporter = do_GetService("@mozilla.org/toolkit/crash-reporter;1");

    if (crashreporter) {

      std::cerr << "Setting up crash reporting" << std::endl;

      nsCOMPtr<nsIProperties> dirsvc =

          do_GetService(NS_DIRECTORY_SERVICE_CONTRACTID);

      nsCOMPtr<nsIFile> cwd;

      nsresult rv = dirsvc->Get(NS_OS_CURRENT_WORKING_DIR,

                       NS_GET_IID(nsIFile),

                       getter_AddRefs(cwd));

      MOZ_RELEASE_ASSERT(NS_SUCCEEDED(rv));

      crashreporter->SetEnabled(true);

      crashreporter->SetMinidumpPath(cwd);

    }

  }

  return RUN_ALL_TESTS();

}

// We use a static var 'RunGTest' defined in nsAppRunner.cpp.

// RunGTest is initialized to nullptr but if GTest (this file)

// is linked in then RunGTest will be set here indicating

// GTest is supported.

class _InitRunGTest {

public:

  _InitRunGTest() {

    RunGTest = RunGTestFunc;

  }

} InitRunGTest;

} // namespace mozilla