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