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)