5 files changed, 234 insertions, 0 deletions
diff --git a/testing/gtest/mozilla/GTestRunner.cpp b/testing/gtest/mozilla/GTestRunner.cpp
new file mode 100644
index 000000000..0864db8cb
--- /dev/null
+++ b/testing/gtest/mozilla/GTestRunner.cpp
@@ -0,0 +1,132 @@
+/* -*- 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"
+#ifdef MOZ_CRASHREPORTER
+#include "nsICrashReporter.h"
+#endif
+#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 c = 0;
+  InitGoogleTest(&c, static_cast<char**>(nullptr));
+
+  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
+#ifdef MOZ_CRASHREPORTER
+  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);
+    }
+  }
+#endif
+
+  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
diff --git a/testing/gtest/mozilla/GTestRunner.h b/testing/gtest/mozilla/GTestRunner.h
new file mode 100644
index 000000000..6c0af0cdb
--- /dev/null
+++ b/testing/gtest/mozilla/GTestRunner.h
@@ -0,0 +1,10 @@
+/* -*- 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/. */
+
+namespace mozilla {
+
+extern int (*RunGTest)();
+
+} // namespace mozilla
diff --git a/testing/gtest/mozilla/MozGTestBench.cpp b/testing/gtest/mozilla/MozGTestBench.cpp
new file mode 100644
index 000000000..15e5a7ae1
--- /dev/null
+++ b/testing/gtest/mozilla/MozGTestBench.cpp
@@ -0,0 +1,39 @@
+/* -*- 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 "MozGTestBench.h"
+#include "mozilla/TimeStamp.h"
+#include <stdio.h>
+
+#define MOZ_GTEST_BENCH_FRAMEWORK "platform_microbench"
+
+using mozilla::TimeStamp;
+
+namespace mozilla {
+void GTestBench(const char* aSuite, const char* aName,
+                const mozilla::function<void()>& aTest)
+{
+#ifdef DEBUG
+  // Run the test to make sure that it doesn't fail but don't log
+  // any measurements since it's not an optimized build.
+  aTest();
+#else
+  mozilla::TimeStamp start = TimeStamp::Now();
+
+  aTest();
+
+  int msDuration = (TimeStamp::Now() - start).ToMicroseconds();
+
+  // Print the result for each test. Let perfherder aggregate for us
+  printf("PERFHERDER_DATA: {\"framework\": {\"name\": \"%s\"}, "
+                           "\"suites\": [{\"name\": \"%s\", \"subtests\": "
+                               "[{\"name\": \"%s\", \"value\": %i, \"lowerIsBetter\": true}]"
+                           "}]}\n",
+                           MOZ_GTEST_BENCH_FRAMEWORK, aSuite, aName, msDuration);
+#endif
+}
+
+} // mozilla
+
diff --git a/testing/gtest/mozilla/MozGTestBench.h b/testing/gtest/mozilla/MozGTestBench.h
new file mode 100644
index 000000000..1e9417b29
--- /dev/null
+++ b/testing/gtest/mozilla/MozGTestBench.h
@@ -0,0 +1,22 @@
+/* -*- 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/. */
+
+#ifndef GTEST_MOZGTESTBENCH_H
+#define GTEST_MOZGTESTBENCH_H
+
+#include "mozilla/Function.h"
+
+namespace mozilla {
+
+void GTestBench(const char* aSuite, const char* aName, const mozilla::function<void()>& aTest);
+
+} //mozilla
+
+#define MOZ_GTEST_BENCH(suite, test, lambdaOrFunc) \
+TEST(suite, test) { \
+  mozilla::GTestBench(#suite, #test, lambdaOrFunc); \
+}
+
+#endif // GTEST_MOZGTESTBENCH_H
diff --git a/testing/gtest/mozilla/SanityTest.cpp b/testing/gtest/mozilla/SanityTest.cpp
new file mode 100644
index 000000000..614c9d30d
--- /dev/null
+++ b/testing/gtest/mozilla/SanityTest.cpp
@@ -0,0 +1,31 @@
+/* -*- 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 "gtest/gtest.h"
+#include "gmock/gmock.h"
+
+using ::testing::AtLeast;
+
+// Sanity test to make sure that GTest is hooked into
+// the mozilla build system correctly
+TEST(MozillaGTestSanity, Runs) {
+  EXPECT_EQ(1, 1);
+}
+namespace {
+class TestMock {
+public:
+  TestMock() {}
+  MOCK_METHOD0(MockedCall, void());
+};
+} // namespace
+TEST(MozillaGMockSanity, Runs) {
+  TestMock mockedClass;
+  EXPECT_CALL(mockedClass, MockedCall())
+    .Times(AtLeast(3));
+
+  mockedClass.MockedCall();
+  mockedClass.MockedCall();
+  mockedClass.MockedCall();
+}