blob: ff43d9252329092611c94a8b137d475a98951a88 [file] [log] [blame]
# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
class _SiteAbstractDrone(object):
This is a site subclass of _BaseAbstractDrone in Any methods
here automatically overload _BaseAbstractDrone and are used to create
_AbstractDrone for consumers.
def __init__(self, timestamp_remote_calls=True):
Add a new private variable _processes_to_kill to _AbstractDrone
@param timestamp_remote_calls: If true, drone_utility is invoked with
the --call_time option and the current time. Currently this is only
used for testing.
super(_SiteAbstractDrone, self).__init__(
self._processes_to_kill = []
def queue_kill_process(self, process):
"""Queue a process to kill/abort.
@param process: Process to kill/abort.
def clear_processes_to_kill(self):
"""Reset the list of processes to kill for this tick."""
self._processes_to_kill = []
def execute_queued_calls(self):
"""Overloads execute_queued_calls().
If there are any processes queued to kill, kill them then process the
remaining queued up calls.
if self._processes_to_kill:
self.queue_call('kill_processes', self._processes_to_kill)
return super(_SiteAbstractDrone, self).execute_queued_calls()