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 39 namespace content { 40 41 class ContentTestSuiteBaseListener : public testing::EmptyTestEventListener { 42 public: ContentTestSuiteBaseListener()43 ContentTestSuiteBaseListener() { 44 } OnTestEnd(const testing::TestInfo & test_info)45 virtual void OnTestEnd(const testing::TestInfo& test_info) OVERRIDE { 46 BrowserThreadImpl::FlushThreadPoolHelper(); 47 } 48 private: 49 DISALLOW_COPY_AND_ASSIGN(ContentTestSuiteBaseListener); 50 }; 51 ContentTestSuiteBase(int argc,char ** argv)52ContentTestSuiteBase::ContentTestSuiteBase(int argc, char** argv) 53 : base::TestSuite(argc, argv) { 54 } 55 Initialize()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 RegisterContentSchemes(ContentClient * content_client)75void ContentTestSuiteBase::RegisterContentSchemes( 76 ContentClient* content_client) { 77 SetContentClient(content_client); 78 content::RegisterContentSchemes(false); 79 SetContentClient(NULL); 80 } 81 RegisterInProcessThreads()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