def testExtractJobRefsTwo(self): job_utils.ArgsForJobRefs(self.parser, nargs='+') jobs = job_utils.ExtractJobRefs( self.ParseArgs('%s %s' % (JOB_1_ID, JOB_2_ID))) self.assertEqual(2, len(jobs)) self.AssertJobRef(jobs[0], JOB_1_ID) self.AssertJobRef(jobs[1], JOB_2_ID)
def testExtractJobRefsTwoWithRegion(self): job_utils.ArgsForJobRefs(self.parser, nargs='+') jobs = job_utils.ExtractJobRefs( self.ParseArgs('--region=%s %s %s' % (REGION, JOB_1_ID, JOB_2_ID))) self.assertEqual(2, len(jobs)) self.AssertJobRef(jobs[0], JOB_1_ID, location=REGION) self.AssertJobRef(jobs[1], JOB_2_ID, location=REGION)
def Run(self, args): """This is what gets called when the user runs this command. Args: args: all the arguments that were provided to this command invocation. """ for job_ref in job_utils.ExtractJobRefs(args.jobs): try: apis.Jobs.Cancel(job_ref.jobId) log.status.Print('Cancelled job [{0}]'.format(job_ref.jobId)) except exceptions.HttpException as error: log.status.Print('Failed to cancel job [{0}]: {1}'.format( job_ref.jobId, error.payload.status_message))
def Run(self, args): """This is what gets called when the user runs this command. Args: args: all the arguments that were provided to this command invocation. """ for job_ref in job_utils.ExtractJobRefs(args): try: apis.Jobs.Drain(job_ref.jobId, project_id=job_ref.projectId, region_id=job_ref.location) log.status.Print('Started draining job [{0}]'.format( job_ref.jobId)) except exceptions.HttpException as error: log.status.Print('Failed to drain job [{0}]: {1}'.format( job_ref.jobId, error.payload.status_message))
def Run(self, args): """This is what gets called when the user runs this command. Args: args: all the arguments that were provided to this command invocation. """ for job_ref in job_utils.ExtractJobRefs(args): try: apis.Jobs.Cancel(job_ref.jobId, project_id=job_ref.projectId, region_id=job_ref.location) log.status.Print('Cancelled job [{0}]'.format(job_ref.jobId)) except exceptions.HttpException as error: log.status.Print( "Failed to cancel job [{0}]: {1} Please ensure you have permission " "to access the job and the `--region` flag, {2}, matches the job\'s" " region.".format(job_ref.jobId, error.payload.status_message, job_ref.location))
def testExtractJobRefsOne(self): job_utils.ArgsForJobRefs(self.parser, nargs='+') jobs = job_utils.ExtractJobRefs(self.parser.parse_args([JOB_1_ID])) self.assertEqual(1, len(jobs)) self.AssertJobRef(jobs[0], JOB_1_ID)
def testExtractJobRefsNoneOptional(self): job_utils.ArgsForJobRefs(self.parser, nargs='*') self.assertEqual([], job_utils.ExtractJobRefs(self.parser.parse_args([])))