| // Copyright (c) 2016 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_NET_TIMEOUT_H_ |
| #define CHROME_TEST_CHROMEDRIVER_NET_TIMEOUT_H_ |
| |
| #include "base/time/time.h" |
| |
| // 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, unless |
| // the duration is exactly the same. |
| 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_; |
| }; |
| |
| #endif // CHROME_TEST_CHROMEDRIVER_NET_TIMEOUT_H_ |