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 Args(parser): """Register flags for this command.""" job_utils.ArgsForJobRefs(parser, nargs='+')
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([])))
def testExtractJobRefsNoneButHasRegion(self): job_utils.ArgsForJobRefs(self.parser, nargs='+') with self.AssertRaisesArgumentErrorMatches( 'argument JOB_ID [JOB_ID ...]: Must be specified.'): job_utils.ExtractJobRef( self.parser.parse_args(['--region=%s' % (REGION)]))
def testExtractJobRefsNone(self): job_utils.ArgsForJobRefs(self.parser, nargs='+') with self.AssertRaisesArgumentErrorMatches( 'argument JOB_ID [JOB_ID ...]: Must be specified.'): job_utils.ExtractJobRef(self.parser.parse_args([]))