| Index: components/dom_distiller/content/distiller_page_web_contents_browsertest.cc
|
| diff --git a/components/dom_distiller/content/distiller_page_web_contents_browsertest.cc b/components/dom_distiller/content/distiller_page_web_contents_browsertest.cc
|
| index 884a75142baaaa076f20da55aaed3c146d709e40..7e7aa0210f7e291548abe2866eb71d8c4c3aaefc 100644
|
| --- a/components/dom_distiller/content/distiller_page_web_contents_browsertest.cc
|
| +++ b/components/dom_distiller/content/distiller_page_web_contents_browsertest.cc
|
| @@ -131,11 +131,9 @@ class WebContentsMainFrameHelper : public content::WebContentsObserver {
|
| WebContentsMainFrameHelper(content::WebContents* web_contents,
|
| const base::Closure& callback,
|
| bool wait_for_document_loaded)
|
| - : web_contents_(web_contents),
|
| + : WebContentsObserver(web_contents),
|
| callback_(callback),
|
| - wait_for_document_loaded_(wait_for_document_loaded) {
|
| - content::WebContentsObserver::Observe(web_contents);
|
| - }
|
| + wait_for_document_loaded_(wait_for_document_loaded) {}
|
|
|
| virtual void DidCommitProvisionalLoadForFrame(
|
| content::RenderFrameHost* render_frame_host,
|
| @@ -148,18 +146,14 @@ class WebContentsMainFrameHelper : public content::WebContentsObserver {
|
| }
|
|
|
| virtual void DocumentLoadedInFrame(
|
| - int64 frame_id,
|
| - content::RenderViewHost* render_view_host) OVERRIDE {
|
| + content::RenderFrameHost* render_frame_host) OVERRIDE {
|
| if (wait_for_document_loaded_) {
|
| - if (web_contents_ &&
|
| - frame_id == web_contents_->GetMainFrame()->GetRoutingID()) {
|
| + if (!render_frame_host->GetParent())
|
| callback_.Run();
|
| - }
|
| }
|
| }
|
|
|
| private:
|
| - content::WebContents* web_contents_;
|
| base::Closure callback_;
|
| bool wait_for_document_loaded_;
|
| };
|
|
|