def setUp(self):
self.RETURN_VALUE = 'foo'
test_obj = self
class FakeAuroraClientAPI(object):
def kill_job(self, job_key, instances=None, lock=None):
test_obj.API_CALL = functools.partial(self.kill_job, job_key, instances, lock)
return test_obj.RETURN_VALUE
def restart(self, job_key, shards, restart_settings):
test_obj.API_CALL = functools.partial(self.restart, job_key, shards, restart_settings)
return test_obj.RETURN_VALUE
def start_cronjob(self, job_key):
test_obj.API_CALL = functools.partial(self.start_cronjob, job_key)
return test_obj.RETURN_VALUE
self._patch_bases(NonHookedAuroraClientAPI, (FakeAuroraClientAPI, ))
self.api = NonHookedAuroraClientAPI()
# Test args passed in to check that these are proxied un-modified
self.test_job_key = AuroraJobKey.from_path('a/b/c/d')
self.test_config = 'bar'
self.test_shards = 'baz'
self.test_lock = 'lock'
self.health_check_interval_seconds = 'baa'
