1// Copyright 2013 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/test/webkit_support.h" 6 7#include "base/message_loop/message_loop.h" 8#include "base/run_loop.h" 9#include "content/test/test_webkit_platform_support.h" 10#include "third_party/WebKit/public/web/WebCache.h" 11#include "third_party/WebKit/public/web/WebKit.h" 12#include "third_party/WebKit/public/web/WebRuntimeFeatures.h" 13#include "url/url_util.h" 14#include "webkit/common/user_agent/user_agent.h" 15#include "webkit/common/user_agent/user_agent_util.h" 16 17#if defined(OS_ANDROID) 18#include "base/android/jni_android.h" 19#include "net/android/network_library.h" 20#endif 21 22#if defined(OS_MACOSX) 23#include "base/test/mock_chrome_application_mac.h" 24#endif 25 26namespace content { 27 28namespace { 29 30class TestEnvironment { 31 public: 32#if defined(OS_ANDROID) 33 // Android UI message loop goes through Java, so don't use it in tests. 34 typedef base::MessageLoop MessageLoopType; 35#else 36 typedef base::MessageLoopForUI MessageLoopType; 37#endif 38 39 TestEnvironment() { 40 main_message_loop_.reset(new MessageLoopType); 41 42 // TestWebKitPlatformSupport must be instantiated after MessageLoopType. 43 webkit_platform_support_.reset(new TestWebKitPlatformSupport); 44 } 45 46 TestWebKitPlatformSupport* webkit_platform_support() const { 47 return webkit_platform_support_.get(); 48 } 49 50 private: 51 scoped_ptr<MessageLoopType> main_message_loop_; 52 scoped_ptr<TestWebKitPlatformSupport> webkit_platform_support_; 53}; 54 55TestEnvironment* test_environment; 56 57} // namespace 58 59void SetUpTestEnvironmentForUnitTests() { 60 blink::WebRuntimeFeatures::enableStableFeatures(true); 61 blink::WebRuntimeFeatures::enableExperimentalFeatures(true); 62 blink::WebRuntimeFeatures::enableTestOnlyFeatures(true); 63 64#if defined(OS_ANDROID) 65 JNIEnv* env = base::android::AttachCurrentThread(); 66 net::android::RegisterNetworkLibrary(env); 67#endif 68 69#if defined(OS_MACOSX) 70 mock_cr_app::RegisterMockCrApp(); 71#endif 72 73 // Explicitly initialize the GURL library before spawning any threads. 74 // Otherwise crash may happend when different threads try to create a GURL 75 // at same time. 76 url_util::Initialize(); 77 test_environment = new TestEnvironment; 78 webkit_glue::SetUserAgent(webkit_glue::BuildUserAgentFromProduct( 79 "DumpRenderTree/0.0.0.0"), false); 80} 81 82void TearDownTestEnvironment() { 83 // Flush any remaining messages before we kill ourselves. 84 // http://code.google.com/p/chromium/issues/detail?id=9500 85 base::RunLoop().RunUntilIdle(); 86 87 if (RunningOnValgrind()) 88 blink::WebCache::clear(); 89 delete test_environment; 90 test_environment = NULL; 91} 92 93} // namespace content 94