test_utils.cc revision 116680a4aac90f2aa7413d9095a592090648e557
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/test/test_utils.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 89ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/run_loop.h" 10868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/values.h" 125d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "content/public/browser/browser_child_process_host_iterator.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/notification_service.h" 14a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "content/public/browser/render_frame_host.h" 155d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "content/public/browser/render_process_host.h" 165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)#include "content/public/common/process_type.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/test/test_launcher.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace content { 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Number of times to repost a Quit task so that the MessageLoop finishes up 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// pending tasks and tasks posted by those pending tasks without risking the 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// potential hang behavior of MessageLoop::QuitWhenIdle. 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The criteria for choosing this number: it should be high enough to make the 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// quit act like QuitWhenIdle, while taking into account that any page which is 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// animating may be rendering another frame for each quit deferral. For an 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// animating page, the potential delay to quitting the RunLoop would be 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// kNumQuitDeferrals * frame_render_time. Some perf tests run slow, such as 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 200ms/frame. 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const int kNumQuitDeferrals = 10; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static void DeferredQuitRunLoop(const base::Closure& quit_task, 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int num_quit_deferrals) { 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (num_quit_deferrals <= 0) { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) quit_task.Run(); 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 40c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop::current()->PostTask( 41c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FROM_HERE, 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&DeferredQuitRunLoop, quit_task, num_quit_deferrals - 1)); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunAllPendingMessageAndSendQuit(BrowserThread::ID thread_id, 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Closure& quit_task) { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunAllPendingInMessageLoop(); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::PostTask(thread_id, FROM_HERE, quit_task); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 52116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch// Class used to handle result callbacks for ExecuteScriptAndGetValue. 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class ScriptCallback { 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) public: 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ScriptCallback() { } 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual ~ScriptCallback() { } 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void ResultCallback(const base::Value* result); 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<base::Value> result() { return result_.Pass(); } 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) private: 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<base::Value> result_; 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ScriptCallback); 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ScriptCallback::ResultCallback(const base::Value* result) { 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (result) 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) result_.reset(result->DeepCopy()); 70c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop::current()->Quit(); 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 73116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch// Monitors if any task is processed by the message loop. 74116680a4aac90f2aa7413d9095a592090648e557Ben Murdochclass TaskObserver : public base::MessageLoop::TaskObserver { 75116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch public: 76116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch TaskObserver() : processed_(false) {} 77116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch virtual ~TaskObserver() {} 78116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 79116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch // MessageLoop::TaskObserver overrides. 80116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch virtual void WillProcessTask(const base::PendingTask& pending_task) OVERRIDE { 81116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch } 82116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch virtual void DidProcessTask(const base::PendingTask& pending_task) OVERRIDE { 83116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch processed_ = true; 84116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch } 85116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 86116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch // Returns true if any task was processed. 87116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch bool processed() const { return processed_; } 88116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 89116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch private: 90116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch bool processed_; 91116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch DISALLOW_COPY_AND_ASSIGN(TaskObserver); 92116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch}; 93116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 944e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)// Adapter that makes a WindowedNotificationObserver::ConditionTestCallback from 954e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)// a WindowedNotificationObserver::ConditionTestCallbackWithoutSourceAndDetails 964e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)// by ignoring the notification source and details. 974e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)bool IgnoreSourceAndDetails( 984e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const WindowedNotificationObserver:: 994e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) ConditionTestCallbackWithoutSourceAndDetails& callback, 1004e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const NotificationSource& source, 1014e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const NotificationDetails& details) { 1024e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) return callback.Run(); 1034e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 1044e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunMessageLoop() { 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::RunLoop run_loop; 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunThisRunLoop(&run_loop); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunThisRunLoop(base::RunLoop* run_loop) { 113c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop::ScopedNestableTaskAllower allow( 114c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop::current()); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1167d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // If we're running inside a browser test, we might need to allow the test 1177d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // launcher to do extra work before/after running a nested message loop. 1187d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) TestLauncherDelegate* delegate = NULL; 1197d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#if !defined(OS_IOS) 1207d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) delegate = GetCurrentTestLauncherDelegate(); 1217d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#endif 1227d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if (delegate) 1237d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) delegate->PreRunMessageLoop(run_loop); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) run_loop->Run(); 1257d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) if (delegate) 1267d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) delegate->PostRunMessageLoop(); 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunAllPendingInMessageLoop() { 130c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop::current()->PostTask( 131c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FROM_HERE, base::MessageLoop::QuitWhenIdleClosure()); 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunMessageLoop(); 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunAllPendingInMessageLoop(BrowserThread::ID thread_id) { 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (BrowserThread::CurrentlyOn(thread_id)) { 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunAllPendingInMessageLoop(); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::ID current_thread_id; 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!BrowserThread::GetCurrentThreadIdentifier(¤t_thread_id)) { 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::RunLoop run_loop; 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::PostTask(thread_id, FROM_HERE, 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&RunAllPendingMessageAndSendQuit, current_thread_id, 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) run_loop.QuitClosure())); 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunThisRunLoop(&run_loop); 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 153116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid RunAllBlockingPoolTasksUntilIdle() { 154116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch while (true) { 155116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch content::BrowserThread::GetBlockingPool()->FlushForTesting(); 156116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 157116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch TaskObserver task_observer; 158116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch base::MessageLoop::current()->AddTaskObserver(&task_observer); 159116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch base::RunLoop().RunUntilIdle(); 160116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch base::MessageLoop::current()->RemoveTaskObserver(&task_observer); 161116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 162116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch if (!task_observer.processed()) 163116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch break; 164116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch } 165116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 166116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::Closure GetQuitTaskForRunLoop(base::RunLoop* run_loop) { 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return base::Bind(&DeferredQuitRunLoop, run_loop->QuitClosure(), 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kNumQuitDeferrals); 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)scoped_ptr<base::Value> ExecuteScriptAndGetValue( 173a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) RenderFrameHost* render_frame_host, const std::string& script) { 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ScriptCallback observer; 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 176a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) render_frame_host->ExecuteJavaScript( 1775d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::UTF8ToUTF16(script), 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&ScriptCallback::ResultCallback, base::Unretained(&observer))); 179c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::MessageLoop* loop = base::MessageLoop::current(); 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) loop->Run(); 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return observer.result().Pass(); 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)MessageLoopRunner::MessageLoopRunner() 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) : loop_running_(false), 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) quit_closure_called_(false) { 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)MessageLoopRunner::~MessageLoopRunner() { 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void MessageLoopRunner::Run() { 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Do not run the message loop if our quit closure has already been called. 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // This helps in scenarios where the closure has a chance to run before 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // we Run explicitly. 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (quit_closure_called_) 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) loop_running_ = true; 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunThisRunLoop(&run_loop_); 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)base::Closure MessageLoopRunner::QuitClosure() { 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return base::Bind(&MessageLoopRunner::Quit, this); 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void MessageLoopRunner::Quit() { 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) quit_closure_called_ = true; 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Only run the quit task if we are running the message loop. 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (loop_running_) { 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) GetQuitTaskForRunLoop(&run_loop_).Run(); 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) loop_running_ = false; 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)WindowedNotificationObserver::WindowedNotificationObserver( 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int notification_type, 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const NotificationSource& source) 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : seen_(false), 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) running_(false), 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) source_(NotificationService::AllSources()) { 2234e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) AddNotificationType(notification_type, source); 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2267d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)WindowedNotificationObserver::WindowedNotificationObserver( 2277d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) int notification_type, 2287d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const ConditionTestCallback& callback) 2297d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) : seen_(false), 2307d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) running_(false), 2317d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) callback_(callback), 2327d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) source_(NotificationService::AllSources()) { 2334e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) AddNotificationType(notification_type, source_); 2344e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 2354e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 2364e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)WindowedNotificationObserver::WindowedNotificationObserver( 2374e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) int notification_type, 2384e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const ConditionTestCallbackWithoutSourceAndDetails& callback) 2394e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) : seen_(false), 2404e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) running_(false), 2414e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) callback_(base::Bind(&IgnoreSourceAndDetails, callback)), 2424e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) source_(NotificationService::AllSources()) { 2437d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) registrar_.Add(this, notification_type, source_); 2447d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)} 2457d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)WindowedNotificationObserver::~WindowedNotificationObserver() {} 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2484e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)void WindowedNotificationObserver::AddNotificationType( 2494e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) int notification_type, 2504e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const NotificationSource& source) { 2514e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) registrar_.Add(this, notification_type, source); 2524e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 2534e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void WindowedNotificationObserver::Wait() { 255eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch if (seen_) 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) running_ = true; 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_runner_ = new MessageLoopRunner; 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_runner_->Run(); 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(seen_); 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void WindowedNotificationObserver::Observe( 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int type, 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const NotificationSource& source, 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const NotificationDetails& details) { 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) source_ = source; 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) details_ = details; 2704e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) if (!callback_.is_null() && !callback_.Run(source, details)) 271eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch return; 272eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) seen_ = true; 274eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch if (!running_) 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) message_loop_runner_->Quit(); 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) running_ = false; 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2815d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)InProcessUtilityThreadHelper::InProcessUtilityThreadHelper() 2825d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) : child_thread_count_(0) { 2835d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) RenderProcessHost::SetRunRendererInProcess(true); 2845d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) BrowserChildProcessObserver::Add(this); 2855d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 2865d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 2875d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)InProcessUtilityThreadHelper::~InProcessUtilityThreadHelper() { 2885d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (child_thread_count_) { 2895d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) DCHECK(BrowserThread::IsMessageLoopValid(BrowserThread::UI)); 2905d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) DCHECK(BrowserThread::IsMessageLoopValid(BrowserThread::IO)); 2915d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) runner_ = new MessageLoopRunner; 2925d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) runner_->Run(); 2935d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) } 2945d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) BrowserChildProcessObserver::Remove(this); 2955d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) RenderProcessHost::SetRunRendererInProcess(false); 2965d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 2975d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 2985d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void InProcessUtilityThreadHelper::BrowserChildProcessHostConnected( 2995d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const ChildProcessData& data) { 3005d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) child_thread_count_++; 3015d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 3025d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 3035d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)void InProcessUtilityThreadHelper::BrowserChildProcessHostDisconnected( 3045d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) const ChildProcessData& data) { 3055d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (--child_thread_count_) 3065d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) return; 3075d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 3085d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) if (runner_.get()) 3095d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) runner_->Quit(); 3105d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)} 3115d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace content 313