[chromedriver] Apply page load timeout to slow cross-process navigations

chrome/test/chromedriver/util.h

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_TEST_CHROMEDRIVER_UTIL_H_
 #define CHROME_TEST_CHROMEDRIVER_UTIL_H_
 
 #include <string>
 
+#include "base/time/time.h"
+
 namespace base {
 class FilePath;
 class ListValue;
 }
 
 struct Session;
 class Status;
 class WebView;
 
+// A helper for tracking the time spent executing a task.  Creating a new
+// instance marks the beginning of the task.  Task duration limit may be set
+// in the contructor or at any time later, but only once.
+class Timeout {
+ public:
+  Timeout();
+  explicit Timeout(const base::TimeDelta& duration);
+
+  // To be used when executing a sub-task with its own timeout, keeping track
+  // of the parent task timeout at the same time. Creates a new Timeout whose
+  // deadline is either Now() + |duration| or deadline of |outer|, whichever is
+  // smaller. |outer| may be nullptr. Note: setting duration on |outer| won't
+  // affect sub-timeouts that were created earlier!
+  Timeout(const base::TimeDelta& duration, const Timeout* outer);
+
+  // Sets the deadline by adding |duration| to the start time recored at
+  // contruction.  Should not be called if the deadline is already set.
+  void SetDuration(const base::TimeDelta& duration);
+
+  bool is_set() const { return !deadline_.is_null(); }
+
+  // Whether the remaining time delta is less than or equal to zero.
+  bool IsExpired() const;
+
+  // Returns the duration if set, otherwise returns TimeDelta::Max().
+  base::TimeDelta GetDuration() const;
+
+  // Returns the remaining time if duration is set, otherwise returns
+  // TimeDelta::Max().
+  base::TimeDelta GetRemainingTime() const;
+
+ private:
+  base::TimeTicks start_;
+  base::TimeTicks deadline_;
+};
+
 // Generates a random, 32-character hexidecimal ID.
 std::string GenerateId();
 
 // Send a sequence of key strokes to the active Element in window.
 Status SendKeysOnWindow(
     WebView* web_view,
     const base::ListValue* key_list,
     bool release_modifiers,
     int* sticky_modifiers);
 
@@ skipping 10 matching lines @@
 Status UnzipSoleFile(const base::FilePath& unzip_dir,
                      const std::string& bytes,
                      base::FilePath* file);
 
 // Calls BeforeCommand for each of |session|'s |CommandListener|s.
 // If an error is encountered, will mark |session| for deletion and return.
 Status NotifyCommandListenersBeforeCommand(Session* session,
                                            const std::string& command_name);
 
 #endif  // CHROME_TEST_CHROMEDRIVER_UTIL_H_