1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
content / public / test / test_frame_navigation_observer.cc [blame]
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/public/test/test_frame_navigation_observer.h"
#include "content/browser/renderer_host/navigation_entry_impl.h"
#include "content/browser/renderer_host/render_frame_host_impl.h"
#include "content/browser/renderer_host/render_view_host_impl.h"
#include "content/browser/web_contents/web_contents_impl.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/navigation_handle.h"
namespace content {
namespace {
RenderFrameHostImpl* ToRenderFrameHostImpl(const ToRenderFrameHost& frame) {
return static_cast<RenderFrameHostImpl*>(frame.render_frame_host());
}
} // namespace
TestFrameNavigationObserver::TestFrameNavigationObserver(
const ToRenderFrameHost& adapter)
: WebContentsObserver(
WebContents::FromRenderFrameHost(ToRenderFrameHostImpl(adapter))),
frame_tree_node_id_(
ToRenderFrameHostImpl(adapter)->GetFrameTreeNodeId()) {
DCHECK_CURRENTLY_ON(BrowserThread::UI);
}
TestFrameNavigationObserver::~TestFrameNavigationObserver() {}
void TestFrameNavigationObserver::Wait() {
DCHECK_CURRENTLY_ON(BrowserThread::UI);
wait_for_commit_ = false;
run_loop_.Run();
}
void TestFrameNavigationObserver::WaitForCommit() {
DCHECK_CURRENTLY_ON(BrowserThread::UI);
if (has_committed_)
return;
wait_for_commit_ = true;
run_loop_.Run();
}
void TestFrameNavigationObserver::DidStartNavigation(
NavigationHandle* navigation_handle) {
last_navigation_succeeded_ = false;
if (navigation_handle->GetFrameTreeNodeId() == frame_tree_node_id_) {
navigation_started_ = true;
has_committed_ = false;
}
}
void TestFrameNavigationObserver::DidFinishNavigation(
NavigationHandle* navigation_handle) {
if (!navigation_started_)
return;
last_navigation_succeeded_ = !navigation_handle->IsErrorPage();
last_net_error_code_ = navigation_handle->GetNetErrorCode();
if (!navigation_handle->HasCommitted() ||
navigation_handle->IsErrorPage() ||
navigation_handle->GetFrameTreeNodeId() != frame_tree_node_id_) {
return;
}
transition_type_ = navigation_handle->GetPageTransition();
last_committed_url_ = navigation_handle->GetURL();
has_committed_ = true;
if (wait_for_commit_)
run_loop_.Quit();
}
void TestFrameNavigationObserver::DidStopLoading() {
if (!navigation_started_)
return;
navigation_started_ = false;
run_loop_.Quit();
}
} // namespace content