| // Copyright 2015 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 ASH_TEST_TASK_SWITCH_TIME_TRACKER_TEST_API_H_ |
| #define ASH_TEST_TASK_SWITCH_TIME_TRACKER_TEST_API_H_ |
| |
| #include <memory> |
| #include <string> |
| |
| #include "base/macros.h" |
| #include "base/time/time.h" |
| |
| namespace base { |
| class SimpleTestTickClock; |
| } // namespace base |
| |
| namespace ash { |
| |
| class TaskSwitchTimeTracker; |
| |
| namespace test { |
| |
| // Provides access TaskSwitchTimeTracker's internals. |
| class TaskSwitchTimeTrackerTestAPI { |
| public: |
| // Creates a TaskSwitchTimeTracker with the given |histogram_name| and injects |
| // a base::SimpleTestTickClock that can be controlled. |
| explicit TaskSwitchTimeTrackerTestAPI(const std::string& histogram_name); |
| |
| ~TaskSwitchTimeTrackerTestAPI(); |
| |
| TaskSwitchTimeTracker* time_tracker() { return time_tracker_.get(); } |
| |
| // Advances |time_tracker_|'s TickClock by |time_delta|. |
| void Advance(base::TimeDelta time_delta); |
| |
| // Wrapper function to access TaskSwitchTimeTracker::HasLastActionTime(); |
| bool HasLastActionTime() const; |
| |
| private: |
| // Owned by |time_tracker_|. |
| base::SimpleTestTickClock* tick_clock_; |
| |
| // The TaskSwitchTimeTracker to provide internal access to. |
| std::unique_ptr<TaskSwitchTimeTracker> time_tracker_; |
| |
| DISALLOW_COPY_AND_ASSIGN(TaskSwitchTimeTrackerTestAPI); |
| }; |
| |
| } // namespace test |
| } // namespace ash |
| |
| #endif // ASH_TEST_TASK_SWITCH_TIME_TRACKER_TEST_API_H_ |