content_test_suite_base.cc revision 2a99a7e74a7f215066514fe81d2bfa6639d9eddd 
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/public/common/content_client.h"
15#include "content/public/common/content_paths.h"
16#include "media/base/media.h"
17#include "testing/gtest/include/gtest/gtest.h"
18#include "ui/base/ui_base_paths.h"
19#include "ui/compositor/compositor_setup.h"
20
21#if defined(OS_ANDROID)
22#include "base/android/jni_android.h"
23#include "content/browser/android/browser_jni_registrar.h"
24#include "content/common/android/common_jni_registrar.h"
25#include "net/android/net_jni_registrar.h"
26#include "ui/android/ui_jni_registrar.h"
27#include "ui/shell_dialogs/android/shell_dialogs_jni_registrar.h"
28#endif
29
30namespace content {
31
32class ContentTestSuiteBaseListener : public testing::EmptyTestEventListener {
33 public:
34  ContentTestSuiteBaseListener() {
35  }
36  virtual void OnTestEnd(const testing::TestInfo& test_info) OVERRIDE {
37    BrowserThreadImpl::FlushThreadPoolHelper();
38  }
39 private:
40  DISALLOW_COPY_AND_ASSIGN(ContentTestSuiteBaseListener);
41};
42
43ContentTestSuiteBase::ContentTestSuiteBase(int argc, char** argv)
44    : base::TestSuite(argc, argv),
45      external_libraries_enabled_(true) {
46}
47
48void ContentTestSuiteBase::Initialize() {
49  base::TestSuite::Initialize();
50
51#if defined(OS_ANDROID)
52  // Register JNI bindings for android.
53  JNIEnv* env = base::android::AttachCurrentThread();
54  content::android::RegisterCommonJni(env);
55  content::android::RegisterBrowserJni(env);
56  net::android::RegisterJni(env);
57  ui::android::RegisterJni(env);
58  ui::shell_dialogs::RegisterJni(env);
59#endif
60
61  if (external_libraries_enabled_)
62    media::InitializeMediaLibraryForTesting();
63
64  scoped_ptr<ContentClient> client_for_init(CreateClientForInitialization());
65  SetContentClient(client_for_init.get());
66  RegisterContentSchemes(false);
67  SetContentClient(NULL);
68
69  RegisterPathProvider();
70  ui::RegisterPathProvider();
71
72  // Mock out the compositor on platforms that use it.
73  ui::SetupTestCompositor();
74
75  testing::UnitTest::GetInstance()->listeners().Append(
76      new ContentTestSuiteBaseListener);
77}
78
79}  // namespace content
80