| OLD | NEW |
| (Empty) | |
| 1 # Copyright 2017 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 import time |
| 5 |
| 6 from telemetry.page import shared_page_state |
| 7 from telemetry import story |
| 8 |
| 9 |
| 10 class _IdleSharedState(shared_page_state.SharedPageState): |
| 11 def __init__(self, test, finder_options, story_set): |
| 12 super(_IdleSharedState, self).__init__(test, finder_options, story_set) |
| 13 self._current_story = None |
| 14 |
| 15 def WillRunStory(self, current_story): |
| 16 self._current_story = current_story |
| 17 assert self.platform.tracing_controller.is_tracing_running |
| 18 |
| 19 def RunStory(self, _): |
| 20 self._current_story.Run(self) |
| 21 |
| 22 def DidRunStory(self, _): |
| 23 self._current_story = None |
| 24 |
| 25 |
| 26 class _IdleStory(story.Story): |
| 27 def __init__(self, name, duration): |
| 28 super(_IdleStory, self).__init__( |
| 29 shared_state_class=_IdleSharedState, name=name) |
| 30 self._duration = duration |
| 31 |
| 32 def Run(self, shared_state): |
| 33 time.sleep(self._duration) |
| 34 |
| 35 |
| 36 class IdleStorySet(story.StorySet): |
| 37 def __init__(self): |
| 38 super(IdleStorySet, self).__init__() |
| 39 self.AddStory(_IdleStory('IdleStory_10s', 10)) |
| 40 self.AddStory(_IdleStory('IdleStory_60s', 60)) |
| 41 self.AddStory(_IdleStory('IdleStory_120s', 120)) |
| OLD | NEW |