Log an error if for some reason we can't add a method to a class (#24)

Maybe it was already swizzled, or some other issue. Either way, it would be good to log this error, as it might help us / developers diagnose an issue.

1 files changed, 10 insertions, 4 deletions

diff --git a/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m b/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m
index 5432c79..214bd41 100644
--- a/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m
+++ b/Firebase/Messaging/FIRMessagingRemoteNotificationsProxy.m
@@ -368,10 +368,16 @@ static NSString *kUserNotificationWillPresentSelectorString =
     struct objc_method_description method_description =
         protocol_getMethodDescription(protocol, originalSelector, NO, YES);
 
-    class_addMethod(klass,
-                    originalSelector,
-                    swizzledImplementation,
-                    method_description.types);
+    BOOL methodAdded = class_addMethod(klass,
+                                       originalSelector,
+                                       swizzledImplementation,
+                                       method_description.types);
+    if (!methodAdded) {
+      FIRMessagingLoggerError(kFIRMessagingMessageCodeRemoteNotificationsProxyMethodNotAdded,
+                              @"Could not add method for %@ to class %@",
+                              NSStringFromSelector(originalSelector),
+                              NSStringFromClass(klass));
+    }
   }
   [self trackSwizzledSelector:originalSelector ofClass:klass];
 }