Rename getTokenWithHandler

4 files changed, 26 insertions, 5 deletions

diff --git a/src/objective-c/GRPCClient/GRPCCall.m b/src/objective-c/GRPCClient/GRPCCall.m
index 589b52031a..18f79311a6 100644
--- a/src/objective-c/GRPCClient/GRPCCall.m
+++ b/src/objective-c/GRPCClient/GRPCCall.m
@@ -885,7 +885,7 @@ const char *kCFStreamVarName = "grpc_cfstream";
     @synchronized(self) {
       self.isWaitingForToken = YES;
     }
-    [_callOptions.authTokenProvider getTokenWithHandler:^(NSString *token) {
+    void (^tokenHandler)(NSString *token) = ^(NSString *token) {
       @synchronized(self) {
         if (self.isWaitingForToken) {
           if (token) {
@@ -895,7 +895,17 @@ const char *kCFStreamVarName = "grpc_cfstream";
           self.isWaitingForToken = NO;
         }
       }
-    }];
+    };
+    id<GRPCAuthorizationProtocol> authTokenProvider = _callOptions.authTokenProvider;
+    if ([authTokenProvider respondsToSelector:@selector(provideTokenToHandler:)]) {
+      [_callOptions.authTokenProvider provideTokenToHandler:tokenHandler];
+    } else {
+      NSAssert([authTokenProvider respondsToSelector:@selector(getTokenWithHandler:)],
+               @"authTokenProvider has no usable method");
+      if ([authTokenProvider respondsToSelector:@selector(getTokenWithHandler:)]) {
+        [_callOptions.authTokenProvider getTokenWithHandler:tokenHandler];
+      }
+    }
   } else {
     [self startCallWithWriteable:writeable];
   }
diff --git a/src/objective-c/GRPCClient/GRPCCallOptions.h b/src/objective-c/GRPCClient/GRPCCallOptions.h
index 85786c7417..b7f08480dc 100644
--- a/src/objective-c/GRPCClient/GRPCCallOptions.h
+++ b/src/objective-c/GRPCClient/GRPCCallOptions.h
@@ -58,13 +58,24 @@ typedef NS_ENUM(NSUInteger, GRPCTransportType) {
 /**
  * Implement this protocol to provide a token to gRPC when a call is initiated.
  */
-@protocol GRPCAuthorizationProtocol
+@protocol GRPCAuthorizationProtocol<NSObject>
+
+@optional
+
+/**
+ * This method is called when gRPC is about to start the call. When OAuth token is acquired,
+ * \a handler is expected to be called with \a token being the new token to be used for this call.
+ */
+- (void)provideTokenToHandler:(void (^_Nullable)(NSString *_Nullable token))handler;
 
 /**
+ * This method is deprecated. Please use \a provideTokenToHandler.
+ *
  * This method is called when gRPC is about to start the call. When OAuth token is acquired,
  * \a handler is expected to be called with \a token being the new token to be used for this call.
  */
 - (void)getTokenWithHandler:(void (^_Nullable)(NSString *_Nullable token))handler;
+
 @end
 
 @interface GRPCCallOptions : NSObject<NSCopying, NSMutableCopying>
diff --git a/src/objective-c/ProtoRPC/ProtoRPC.m b/src/objective-c/ProtoRPC/ProtoRPC.m
index da53435178..92d7fce33c 100644
--- a/src/objective-c/ProtoRPC/ProtoRPC.m
+++ b/src/objective-c/ProtoRPC/ProtoRPC.m
@@ -168,7 +168,7 @@ static NSError *ErrorForBadProto(id proto, Class expectedClass, NSError *parsing
 }
 
 - (void)writeMessage:(GPBMessage *)message {
-  NSAssert([message isKindOfClass:[GPBMessage class]]);
+  NSAssert([message isKindOfClass:[GPBMessage class]], @"Parameter message must be a GPBMessage");
   if (![message isKindOfClass:[GPBMessage class]]) {
     NSLog(@"Failed to send a message that is non-proto.");
     return;
diff --git a/src/objective-c/tests/APIv2Tests/APIv2Tests.m b/src/objective-c/tests/APIv2Tests/APIv2Tests.m
index ca7bf47283..fd472aafeb 100644
--- a/src/objective-c/tests/APIv2Tests/APIv2Tests.m
+++ b/src/objective-c/tests/APIv2Tests/APIv2Tests.m
@@ -241,7 +241,7 @@ static const NSTimeInterval kTestTimeout = 16;
   [self waitForExpectationsWithTimeout:kTestTimeout handler:nil];
 }
 
-- (void)getTokenWithHandler:(void (^)(NSString *token))handler {
+- (void)provideTokenToHandler:(void (^)(NSString *token))handler {
   dispatch_queue_t queue = dispatch_queue_create(NULL, DISPATCH_QUEUE_SERIAL);
   dispatch_sync(queue, ^{
     handler(@"test-access-token");