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([]))
