def mock_restart_fails(self):
response = Response(responseCode=ResponseCode.ERROR, messageDEPRECATED='test error')
response.serverInfo = ServerInfo(thriftAPIVersion=THRIFT_API_VERSION)
self.mock_scheduler.restartShards(
JOB.to_thrift(),
IgnoreArg(),
self.lock,
SESSION_KEY).AndReturn(response)
def mock_restart_instances(self, instances, lock=None):
response = Response(responseCode=ResponseCode.OK, messageDEPRECATED='test')
response.serverInfo = ServerInfo(thriftAPIVersion=THRIFT_API_VERSION)
self.mock_scheduler.restartShards(
JOB.to_thrift(),
instances,
lock,
SESSION_KEY).AndReturn(response)
self.mock_instance_watcher.watch(instances).AndReturn([])
def mock_status_active_tasks(self, instance_ids):
tasks = []
for i in instance_ids:
tasks.append(ScheduledTask(
status=ScheduleStatus.RUNNING,
assignedTask=AssignedTask(task=TaskConfig(), instanceId=i)
))
response = Response(responseCode=ResponseCode.OK, messageDEPRECATED='test')
response.serverInfo = ServerInfo(thriftAPIVersion=THRIFT_API_VERSION)
response.result = Result()
response.result.scheduleStatusResult = ScheduleStatusResult(tasks=tasks)
self.mock_scheduler.getTasksWithoutConfigs(IgnoreArg()).AndReturn(response)
def mock_status_no_active_task(self): response = Response(responseCode=ResponseCode.INVALID_REQUEST, messageDEPRECATED='test') response.serverInfo = ServerInfo(thriftAPIVersion=THRIFT_API_VERSION) self.mock_scheduler.getTasksWithoutConfigs(IgnoreArg()).AndReturn(response)
