Fix removed method in SimDeviceNotifier

badoo · Jul 16, 2019 · 4fdc14f · 4fdc14f
1 parent dc9d279
commit 4fdc14f
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/FBSimulatorControl/Notifiers/FBCoreSimulatorNotifier.m b/FBSimulatorControl/Notifiers/FBCoreSimulatorNotifier.m
@@ -46,11 +46,22 @@ - (instancetype)initWithNotifier:(id<SimDeviceNotifier>)notifier queue:(dispatch
   }
 
   _notifier = notifier;
-  _handle = [notifier registerNotificationHandler:^(NSDictionary *info) {
-    dispatch_async(queue, ^{
-      block(info);
-    });
-  }];
+
+  BOOL isNewApi = [(NSObject *)notifier respondsToSelector:@selector(registerNotificationHandlerOnQueue:handler:)];
+
+  if (isNewApi) {
+    _handle = [notifier registerNotificationHandlerOnQueue:queue handler:^(NSDictionary *info) {
+      dispatch_async(queue, ^{
+        block(info);
+      });
+    }];
+  } else {
+    _handle = [notifier registerNotificationHandler:^(NSDictionary *info) {
+      dispatch_async(queue, ^{
+        block(info);
+      });
+    }];
+  }
 
   return self;
 }