gpu/command_buffer/common/unittest_main.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/at_exit.h"
   6 #include "base/bind.h"
   7 #include "base/command_line.h"
   8 #include "base/test/launcher/unit_test_launcher.h"
   9 #include "base/test/test_suite.h"
  10 #include "testing/gmock/include/gmock/gmock.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12 #include "ui/gl/gl_implementation.h"
  13 #include "ui/gl/gl_mock.h"
  14 #include "ui/gl/gl_surface.h"
  15
  16 namespace {
  17
  18 class NoAtExitBaseTestSuite : public base::TestSuite {
  19  public:
  20   NoAtExitBaseTestSuite(int argc, char** argv)
  21       : base::TestSuite(argc, argv, false) {
  22   }
  23
  24   virtual void Initialize() OVERRIDE {
  25     base::TestSuite::Initialize();
  26     gfx::SetGLGetProcAddressProc(gfx::MockGLInterface::GetGLProcAddress);
  27     gfx::GLSurface::InitializeOneOffWithMockBindingsForTests();
  28     gfx::GLSurface::InitializeDynamicMockBindingsForTests(NULL);
  29   }
  30 };
  31
  32 int RunTestSuite(int argc, char** argv) {
  33   return NoAtExitBaseTestSuite(argc, argv).Run();
  34 }
  35
  36 }  // namespace
  37
  38 int main(int argc, char** argv) {
  39   // On Android, AtExitManager is created in
  40   // testing/android/native_test_wrapper.cc before main() is called.
  41   // The same thing is also done in base/test/test_suite.cc
  42 #if !defined(OS_ANDROID)
  43   base::AtExitManager exit_manager;
  44 #endif
  45   CommandLine::Init(argc, argv);
  46   testing::InitGoogleMock(&argc, argv);
  47   return base::LaunchUnitTests(argc,
  48                                argv,
  49                                base::Bind(&RunTestSuite, argc, argv));
  50 }