content_test_suite_base.cc revision 5c02ac1a9c1b504631c0a3d2b6e737b5d738bae1 
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 "content/public/test/content_test_suite_base.h"
6
7#include "base/basictypes.h"
8#include "base/compiler_specific.h"
9#include "base/memory/scoped_ptr.h"
10#include "base/test/test_suite.h"
11#include "base/threading/sequenced_worker_pool.h"
12#include "content/browser/browser_thread_impl.h"
13#include "content/common/url_schemes.h"
14#include "content/gpu/in_process_gpu_thread.h"
15#include "content/public/common/content_client.h"
16#include "content/renderer/in_process_renderer_thread.h"
17#include "content/utility/in_process_utility_thread.h"
18#include "testing/gtest/include/gtest/gtest.h"
19#include "ui/base/ui_base_paths.h"
20
21#if !defined(OS_IOS)
22#include "content/browser/gpu/gpu_process_host.h"
23#include "content/browser/renderer_host/render_process_host_impl.h"
24#include "content/browser/utility_process_host_impl.h"
25#endif
26
27#if defined(OS_ANDROID)
28#include "base/android/jni_android.h"
29#include "content/browser/android/browser_jni_registrar.h"
30#include "content/common/android/common_jni_registrar.h"
31#include "media/base/android/media_jni_registrar.h"
32#include "net/android/net_jni_registrar.h"
33#include "ui/base/android/ui_base_jni_registrar.h"
34#include "ui/gfx/android/gfx_jni_registrar.h"
35#include "ui/gl/android/gl_jni_registrar.h"
36#include "ui/shell_dialogs/android/shell_dialogs_jni_registrar.h"
37#endif
38
39namespace content {
40
41class ContentTestSuiteBaseListener : public testing::EmptyTestEventListener {
42 public:
43  ContentTestSuiteBaseListener() {
44  }
45  virtual void OnTestEnd(const testing::TestInfo& test_info) OVERRIDE {
46    BrowserThreadImpl::FlushThreadPoolHelper();
47  }
48 private:
49  DISALLOW_COPY_AND_ASSIGN(ContentTestSuiteBaseListener);
50};
51
52ContentTestSuiteBase::ContentTestSuiteBase(int argc, char** argv)
53    : base::TestSuite(argc, argv) {
54}
55
56void ContentTestSuiteBase::Initialize() {
57  base::TestSuite::Initialize();
58
59#if defined(OS_ANDROID)
60  // Register JNI bindings for android.
61  JNIEnv* env = base::android::AttachCurrentThread();
62  content::android::RegisterCommonJni(env);
63  content::android::RegisterBrowserJni(env);
64  gfx::android::RegisterJni(env);
65  media::RegisterJni(env);
66  net::android::RegisterJni(env);
67  ui::android::RegisterJni(env);
68  ui::shell_dialogs::RegisterJni(env);
69#endif
70
71  testing::UnitTest::GetInstance()->listeners().Append(
72      new ContentTestSuiteBaseListener);
73}
74
75void ContentTestSuiteBase::RegisterContentSchemes(
76    ContentClient* content_client) {
77  SetContentClient(content_client);
78  content::RegisterContentSchemes(false);
79  SetContentClient(NULL);
80}
81
82void ContentTestSuiteBase::RegisterInProcessThreads() {
83#if !defined(OS_IOS)
84  UtilityProcessHostImpl::RegisterUtilityMainThreadFactory(
85      CreateInProcessUtilityThread);
86  RenderProcessHostImpl::RegisterRendererMainThreadFactory(
87      CreateInProcessRendererThread);
88  GpuProcessHost::RegisterGpuMainThreadFactory(CreateInProcessGpuThread);
89#endif
90}
91
92}  // namespace content
93