def Args(parser):
"""Register flags for this command.
Args:
parser: An argparse.ArgumentParser-like object. It is mocked out in order
to capture some information, but behaves like an ArgumentParser.
"""
iap_util.AddIapIamResourceArgs(parser)
base.URI_FLAG.RemoveFromParser(parser)
def testParse(self, args, expected_type):
self.get_client_instance = self.StartObjectPatch(
apis, 'GetClientInstance')
parser = calliope_util.ArgumentParser()
parser.add_argument('--project', help='The project.')
util.AddIapIamResourceArgs(parser)
parsed_args = parser.parse_args(args)
resource = util.ParseIapIamResource(self.track, parsed_args)
self.assertEqual(type(resource), expected_type)
def testParseInvalid(self, args, expected_exception):
self.get_client_instance = self.StartObjectPatch(
apis, 'GetClientInstance')
parser = calliope_util.ArgumentParser()
parser.add_argument('--project', help='The project.')
util.AddIapIamResourceArgs(parser)
with self.assertRaises(expected_exception):
parsed_args = parser.parse_args(args)
util.ParseIapIamResource(self.track, parsed_args)
def Args(parser):
"""Register flags for this command.
Args:
parser: An argparse.ArgumentParser-like object. It is mocked out in order
to capture some information, but behaves like an ArgumentParser.
"""
iap_util.AddIapIamResourceArgs(
parser, use_region_arg=True, use_iap_gateway=True)
iap_util.AddRemoveIamPolicyBindingArgs(parser)
base.URI_FLAG.RemoveFromParser(parser)
