| OLD | NEW |
| 1 # Copyright 2014 The Chromium Authors. All rights reserved. | 1 # Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 from slave import recipe_api | 5 from recipe_engine import recipe_api |
| 6 | 6 |
| 7 class SwarmingHeartbeatApi(recipe_api.RecipeApi): | 7 class SwarmingHeartbeatApi(recipe_api.RecipeApi): |
| 8 """Defines support to run automated heartbeat.""" | 8 """Defines support to run automated heartbeat.""" |
| 9 | 9 |
| 10 def run(self): | 10 def run(self): |
| 11 """Runs job_runs_fine.py to run an heartbeat check.""" | 11 """Runs job_runs_fine.py to run an heartbeat check.""" |
| 12 args = [] | 12 args = [] |
| 13 if self.m.properties.get('target_environment') == 'canary': | 13 if self.m.properties.get('target_environment') == 'canary': |
| 14 args = ['--canary'] | 14 args = ['--canary'] |
| 15 self.m.python( | 15 self.m.python( |
| 16 'job_runs_fine.py', script=self.resource('job_runs_fine.py'), args=args) | 16 'job_runs_fine.py', script=self.resource('job_runs_fine.py'), args=args) |
| OLD | NEW |