def testParseClearPullFlags(self): flags.AddUpdatePullQueueFlags(self.parser) expected_config = _ALPHA_MESSAGES_MODULE.Queue( retryConfig=_ALPHA_MESSAGES_MODULE.RetryConfig()) args = self.parser.parse_args( ['--clear-max-attempts', '--clear-max-retry-duration']) actual_config = parsers.ParseCreateOrUpdateQueueArgs( args, constants.PULL_QUEUE, _ALPHA_MESSAGES_MODULE, is_update=True, release_track=base.ReleaseTrack.ALPHA) self.assertEqual(actual_config, expected_config)
def Args(parser): flags.AddQueueResourceArg(parser, 'to update') flags.AddLocationFlag(parser) flags.AddUpdatePullQueueFlags(parser)