OLD | NEW |
| (Empty) |
1 // Copyright (c) 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 "chrome/test/chromedriver/navigation_tracker.h" | |
6 | |
7 #include "base/logging.h" | |
8 #include "base/stringprintf.h" | |
9 #include "base/values.h" | |
10 #include "chrome/test/chromedriver/devtools_client.h" | |
11 #include "chrome/test/chromedriver/status.h" | |
12 | |
13 NavigationTracker::NavigationTracker(DevToolsClient* client) | |
14 : client_(client), | |
15 loading_state_(kUnknown) { | |
16 client_->AddListener(this); | |
17 } | |
18 | |
19 NavigationTracker::NavigationTracker(DevToolsClient* client, | |
20 LoadingState known_state) | |
21 : client_(client), | |
22 loading_state_(known_state) { | |
23 client_->AddListener(this); | |
24 } | |
25 | |
26 NavigationTracker::~NavigationTracker() {} | |
27 | |
28 Status NavigationTracker::IsPendingNavigation(const std::string& frame_id, | |
29 bool* is_pending) { | |
30 if (loading_state_ == kUnknown) { | |
31 // If the loading state is unknown (which happens after first connecting), | |
32 // force loading to start and set the state to loading. This will | |
33 // cause a frame start event to be received, and the frame stop event | |
34 // will not be received until all frames are loaded. | |
35 // Loading is forced to start by attaching a temporary iframe. | |
36 // Forcing loading to start is not necessary if the main frame is not yet | |
37 // loaded. | |
38 const char kStartLoadingIfMainFrameNotLoading[] = | |
39 "var isLoaded = document.readyState == 'complete' ||" | |
40 " document.readyState == 'interactive';" | |
41 "if (isLoaded) {" | |
42 " var frame = document.createElement('iframe');" | |
43 " frame.src = 'about:blank';" | |
44 " document.body.appendChild(frame);" | |
45 " window.setTimeout(function() {" | |
46 " document.body.removeChild(frame);" | |
47 " }, 0);" | |
48 "}"; | |
49 base::DictionaryValue params; | |
50 params.SetString("expression", kStartLoadingIfMainFrameNotLoading); | |
51 scoped_ptr<base::DictionaryValue> result; | |
52 Status status = client_->SendCommandAndGetResult( | |
53 "Runtime.evaluate", params, &result); | |
54 if (status.IsError()) | |
55 return Status(kUnknownError, "cannot determine loading status", status); | |
56 | |
57 // Between the time the JavaScript is evaluated and SendCommandAndGetResult | |
58 // returns, OnEvent may have received info about the loading state. | |
59 // This is only possible during a nested command. Only set the loading state | |
60 // if the loading state is still unknown. | |
61 if (loading_state_ == kUnknown) | |
62 loading_state_ = kLoading; | |
63 } | |
64 *is_pending = (loading_state_ == kLoading) || | |
65 scheduled_frame_set_.count(frame_id) > 0; | |
66 return Status(kOk); | |
67 } | |
68 | |
69 Status NavigationTracker::OnConnected() { | |
70 loading_state_ = kUnknown; | |
71 scheduled_frame_set_.clear(); | |
72 | |
73 // Enable page domain notifications to allow tracking navigation state. | |
74 base::DictionaryValue empty_params; | |
75 return client_->SendCommand("Page.enable", empty_params); | |
76 } | |
77 | |
78 void NavigationTracker::OnEvent(const std::string& method, | |
79 const base::DictionaryValue& params) { | |
80 // Chrome does not send Page.frameStoppedLoading until all frames have | |
81 // run their onLoad handlers (including frames created during the handlers). | |
82 // When it does, it only sends one stopped event for all frames. | |
83 if (method == "Page.frameStartedLoading") { | |
84 loading_state_ = kLoading; | |
85 } else if (method == "Page.frameStoppedLoading") { | |
86 loading_state_ = kNotLoading; | |
87 } else if (method == "Page.frameScheduledNavigation") { | |
88 double delay; | |
89 if (!params.GetDouble("delay", &delay)) { | |
90 LOG(ERROR) << "missing or invalid 'delay'"; | |
91 return; | |
92 } | |
93 std::string frame_id; | |
94 if (!params.GetString("frameId", &frame_id)) { | |
95 LOG(ERROR) << "missing or invalid 'frameId'"; | |
96 return; | |
97 } | |
98 // WebDriver spec says to ignore redirects over 1s. | |
99 if (delay > 1) | |
100 return; | |
101 scheduled_frame_set_.insert(frame_id); | |
102 } else if (method == "Page.frameClearedScheduledNavigation") { | |
103 std::string frame_id; | |
104 if (!params.GetString("frameId", &frame_id)) { | |
105 LOG(ERROR) << "missing or invalid 'frameId'"; | |
106 return; | |
107 } | |
108 scheduled_frame_set_.erase(frame_id); | |
109 } else if (method == "Page.frameNavigated") { | |
110 // Note: in some cases Page.frameNavigated may be received for subframes | |
111 // without a frameStoppedLoading (for example cnn.com). | |
112 | |
113 // If the main frame just navigated, discard any pending scheduled | |
114 // navigations. For some reasons at times the cleared event is not | |
115 // received when navigating. | |
116 // See crbug.com/180742. | |
117 const base::Value* unused_value; | |
118 if (!params.Get("frame.parentId", &unused_value)) | |
119 scheduled_frame_set_.clear(); | |
120 } | |
121 } | |
OLD | NEW |