def Run(self, args): """This is what gets called when the user runs this command. Args: args: an argparse namespace. All the arguments that were provided to this command invocation. Returns: Some value that we want to have printed later. """ op = instances.Upgrade(args.instance) if args.async_: result = op else: op_ref = resources.REGISTRY.ParseRelativeName( op.name, collection='bigtableadmin.operations') message = 'Upgrading bigtable instance {0}'.format(args.instance) result = util.AwaitInstance(op_ref, message) log.UpdatedResource(args.instance, kind='instance', is_async=args.async_) return result
def testUpgrade(self): response = self.msgs.Operation() self.ExpectInstanceUpdateRequest(response) self.assertEqual(instances.Upgrade('my-instance'), response)