appium-webdriveragent 4.7.0 → 4.8.0

package/WebDriverAgent.xcodeproj/xcuserdata/elf.xcuserdatad/xcdebugger/Breakpoints_v2.xcbkptlist

@@ -0,0 +1,91 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<Bucket
+   uuid = "2F0174F2-ED3F-4D11-9A17-7C590053D796"
+   type = "1"
+   version = "2.0">
+   <Breakpoints>
+      <BreakpointProxy
+         BreakpointExtensionID = "Xcode.Breakpoint.ExceptionBreakpoint">
+         <BreakpointContent
+            uuid = "471ECA8C-2C53-4C59-88DB-1C1CB3D61601"
+            shouldBeEnabled = "No"
+            ignoreCount = "0"
+            continueAfterRunningActions = "No"
+            breakpointStackSelectionBehavior = "1"
+            scope = "1"
+            stopOnStyle = "0">
+         </BreakpointContent>
+      </BreakpointProxy>
+      <BreakpointProxy
+         BreakpointExtensionID = "Xcode.Breakpoint.FileBreakpoint">
+         <BreakpointContent
+            uuid = "C2E06865-6A52-4686-9771-279AF13ABF8A"
+            shouldBeEnabled = "Yes"
+            ignoreCount = "0"
+            continueAfterRunningActions = "No"
+            filePath = "WebDriverAgentLib/Categories/XCUIElement+FBUtilities.m"
+            timestampString = "0"
+            startingColumnNumber = "9223372036854775807"
+            endingColumnNumber = "9223372036854775807"
+            startingLineNumber = "264"
+            endingLineNumber = "264">
+         </BreakpointContent>
+      </BreakpointProxy>
+      <BreakpointProxy
+         BreakpointExtensionID = "Xcode.Breakpoint.FileBreakpoint">
+         <BreakpointContent
+            uuid = "54216FE6-E240-47F1-88D3-D450CF0AF44E"
+            shouldBeEnabled = "Yes"
+            ignoreCount = "0"
+            continueAfterRunningActions = "No"
+            filePath = "WebDriverAgentLib/Categories/XCUIElement+FBResolve.m"
+            startingColumnNumber = "9223372036854775807"
+            endingColumnNumber = "9223372036854775807"
+            startingLineNumber = "37"
+            endingLineNumber = "37"
+            landmarkName = "-fb_stableInstance"
+            landmarkType = "7">
+            <Locations>
+               <Location
+                  uuid = "54216FE6-E240-47F1-88D3-D450CF0AF44E - b09f4ce1ce926e6"
+                  shouldBeEnabled = "Yes"
+                  ignoreCount = "0"
+                  continueAfterRunningActions = "No"
+                  symbolName = "-[XCUIElement(FBResolve) fb_stableInstance]"
+                  moduleName = "WebDriverAgentLib"
+                  usesParentBreakpointCondition = "Yes"
+                  urlString = "file:///Users/elf/code/WebDriverAgent/WebDriverAgentLib/Categories/XCUIElement+FBResolve.m"
+                  startingColumnNumber = "9223372036854775807"
+                  endingColumnNumber = "9223372036854775807"
+                  startingLineNumber = "37"
+                  endingLineNumber = "37"
+                  offsetFromSymbolStart = "19">
+               </Location>
+               <Location
+                  shouldBeEnabled = "Yes"
+                  ignoreCount = "0"
+                  continueAfterRunningActions = "No"
+                  usesParentBreakpointCondition = "Yes"
+                  offsetFromSymbolStart = "0">
+               </Location>
+            </Locations>
+         </BreakpointContent>
+      </BreakpointProxy>
+      <BreakpointProxy
+         BreakpointExtensionID = "Xcode.Breakpoint.FileBreakpoint">
+         <BreakpointContent
+            uuid = "D3009B57-42BC-4905-99BC-70232544FAFB"
+            shouldBeEnabled = "Yes"
+            ignoreCount = "0"
+            continueAfterRunningActions = "No"
+            filePath = "WebDriverAgentTests/IntegrationTests/FBAlertTests.m"
+            startingColumnNumber = "5"
+            endingColumnNumber = "64"
+            startingLineNumber = "32"
+            endingLineNumber = "32"
+            landmarkName = "-setUp"
+            landmarkType = "7">
+         </BreakpointContent>
+      </BreakpointProxy>
+   </Breakpoints>
+</Bucket>

package/WebDriverAgent.xcodeproj/xcuserdata/elf.xcuserdatad/xcschemes/xcschememanagement.plist

@@ -0,0 +1,59 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+	<key>SchemeUserState</key>
+	<dict>
+		<key>IntegrationApp.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>0</integer>
+		</dict>
+		<key>IntegrationTests_1.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>1</integer>
+		</dict>
+		<key>IntegrationTests_2.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>2</integer>
+		</dict>
+		<key>IntegrationTests_3.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>3</integer>
+		</dict>
+		<key>WebDriverAgentLib copy.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>9</integer>
+		</dict>
+		<key>WebDriverAgentLib.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>4</integer>
+		</dict>
+		<key>WebDriverAgentLib_tvOS.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>5</integer>
+		</dict>
+		<key>WebDriverAgentRunner-nodebug.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>7</integer>
+		</dict>
+		<key>WebDriverAgentRunner.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>6</integer>
+		</dict>
+		<key>WebDriverAgentRunner_tvOS.xcscheme_^#shared#^_</key>
+		<dict>
+			<key>orderHint</key>
+			<integer>8</integer>
+		</dict>
+	</dict>
+</dict>
+</plist>

package/WebDriverAgentLib/Categories/{XCElementSnapshot+FBHelpers.h → FBXCElementSnapshotWrapper+Helpers.h}

@@ -1,5 +1,5 @@
 /**
- * Copyright (c) 2015-present, Facebook, Inc.
+ * Copyright (c) 2018-present, Facebook, Inc.
  * All rights reserved.
  *
  * This source code is licensed under the BSD-style license found in the
@@ -7,11 +7,11 @@
  * of patent rights can be found in the PATENTS file in the same directory.
  */
 
-#import <WebDriverAgentLib/XCElementSnapshot.h>
+#import "FBXCElementSnapshotWrapper.h"
 
 NS_ASSUME_NONNULL_BEGIN
 
-@interface XCElementSnapshot (FBHelpers)
+@interface FBXCElementSnapshotWrapper (Helpers)
 
 /**
  Returns an array of descendants matching given type
@@ -19,7 +19,7 @@ NS_ASSUME_NONNULL_BEGIN
  @param type requested descendant type
  @return an array of descendants matching given type
  */
-- (NSArray<XCElementSnapshot *> *)fb_descendantsMatchingType:(XCUIElementType)type;
+- (NSArray<id<FBXCElementSnapshot>> *)fb_descendantsMatchingType:(XCUIElementType)type;
 
 /**
  Returns first (going up element tree) parent that matches given type. If non found returns nil.
@@ -27,7 +27,7 @@ NS_ASSUME_NONNULL_BEGIN
  @param type requested parent type
  @return parent element matching given type
  */
-- (nullable XCElementSnapshot *)fb_parentMatchingType:(XCUIElementType)type;
+- (nullable id<FBXCElementSnapshot>)fb_parentMatchingType:(XCUIElementType)type;
 
 /**
  Returns first (going up element tree) parent that matches one of given types. If non found returns nil.
@@ -35,7 +35,7 @@ NS_ASSUME_NONNULL_BEGIN
  @param types possible parent types
  @return parent element matching one of given types
  */
-- (nullable XCElementSnapshot *)fb_parentMatchingOneOfTypes:(NSArray<NSNumber *> *)types;
+- (nullable id<FBXCElementSnapshot>)fb_parentMatchingOneOfTypes:(NSArray<NSNumber *> *)types;
 
 /**
  Returns first (going up element tree) visible parent that matches one of given types and has more than one child. If non found returns nil.
@@ -44,14 +44,14 @@ NS_ASSUME_NONNULL_BEGIN
  @param filter will filter results even further after matching one of given types
  @return parent element matching one of given types and satisfying filter condition
  */
-- (nullable XCElementSnapshot *)fb_parentMatchingOneOfTypes:(NSArray<NSNumber *> *)types filter:(BOOL(^)(XCElementSnapshot *snapshot))filter;
+- (nullable id<FBXCElementSnapshot>)fb_parentMatchingOneOfTypes:(NSArray<NSNumber *> *)types filter:(BOOL(^)(id<FBXCElementSnapshot> snapshot))filter;
 
 /**
  Retrieves the list of all element ancestors in the snapshot hierarchy.
  
  @return the list of element ancestors or an empty list if the snapshot has no parent.
  */
-- (NSArray<XCElementSnapshot *> *)fb_ancestors;
+- (NSArray<id<FBXCElementSnapshot>> *)fb_ancestors;
 
 /**
  Returns value for given accessibility property identifier.
@@ -68,21 +68,21 @@ NS_ASSUME_NONNULL_BEGIN
  @param snapshot element's snapshot to compare against
  @return YES, if they match otherwise NO
  */
-- (BOOL)fb_framelessFuzzyMatchesElement:(XCElementSnapshot *)snapshot;
+- (BOOL)fb_framelessFuzzyMatchesElement:(id<FBXCElementSnapshot>)snapshot;
 
 /**
  Returns an array of descendants cell snapshots
  
  @return an array of descendants cell snapshots
  */
-- (NSArray<XCElementSnapshot *> *)fb_descendantsCellSnapshots;
+- (NSArray<id<FBXCElementSnapshot>> *)fb_descendantsCellSnapshots;
 
 /**
  Returns itself if it is either XCUIElementTypeIcon or XCUIElementTypeCell. Otherwise, returns first (going up element tree) parent that matches cell (XCUIElementTypeCell or  XCUIElementTypeIcon). If non found returns nil.
  
  @return parent element matching either XCUIElementTypeIcon or XCUIElementTypeCell.
  */
-- (nullable XCElementSnapshot *)fb_parentCellSnapshot;
+- (nullable id<FBXCElementSnapshot>)fb_parentCellSnapshot;
 
 /**! Human-readable snapshot description */
 - (NSString *)fb_description;
@@ -94,6 +94,13 @@ NS_ASSUME_NONNULL_BEGIN
  */
 - (CGRect)fb_visibleFrameWithFallback;
 
+/**
+ Wrapper for Apple's hitpoint, thats resolves few known issues
+
+ @return Element's hitpoint if exists nil otherwise
+ */
+- (nullable NSValue *)fb_hitPoint;
+
 @end
 
 NS_ASSUME_NONNULL_END

package/WebDriverAgentLib/Categories/{XCElementSnapshot+FBHelpers.m → FBXCElementSnapshotWrapper+Helpers.m}

@@ -1,5 +1,5 @@
 /**
- * Copyright (c) 2015-present, Facebook, Inc.
+ * Copyright (c) 2018-present, Facebook, Inc.
  * All rights reserved.
  *
  * This source code is licensed under the BSD-style license found in the
@@ -7,20 +7,23 @@
  * of patent rights can be found in the PATENTS file in the same directory.
  */
 
-#import "XCElementSnapshot+FBHelpers.h"
+#import "FBXCElementSnapshotWrapper+Helpers.h"
 
 #import "FBFindElementCommands.h"
 #import "FBRunLoopSpinner.h"
 #import "FBLogger.h"
+#import "FBXCElementSnapshot.h"
 #import "FBXCAXClientProxy.h"
 #import "XCTestDriver.h"
 #import "XCTestPrivateSymbols.h"
 #import "XCUIElement.h"
 #import "XCUIElement+FBWebDriverAttributes.h"
+#import "XCUIHitPointResult.h"
 
-inline static BOOL isSnapshotTypeAmongstGivenTypes(XCElementSnapshot* snapshot, NSArray<NSNumber *> *types);
+inline static BOOL isSnapshotTypeAmongstGivenTypes(id<FBXCElementSnapshot> snapshot,
+                                                   NSArray<NSNumber *> *types);
 
-@implementation XCElementSnapshot (FBHelpers)
+@implementation FBXCElementSnapshotWrapper (Helpers)
 
 - (NSString *)fb_description
 {
@@ -31,29 +34,30 @@ inline static BOOL isSnapshotTypeAmongstGivenTypes(XCElementSnapshot* snapshot,
   return result;
 }
 
-- (NSArray<XCElementSnapshot *> *)fb_descendantsMatchingType:(XCUIElementType)type
+- (NSArray<id<FBXCElementSnapshot>> *)fb_descendantsMatchingType:(XCUIElementType)type
 {
-  return [self descendantsByFilteringWithBlock:^BOOL(XCElementSnapshot *snapshot) {
+  return [self descendantsByFilteringWithBlock:^BOOL(id<FBXCElementSnapshot> snapshot) {
     return snapshot.elementType == type;
   }];
 }
 
-- (XCElementSnapshot *)fb_parentMatchingType:(XCUIElementType)type
+- (id<FBXCElementSnapshot>)fb_parentMatchingType:(XCUIElementType)type
 {
   NSArray *acceptedParents = @[@(type)];
   return [self fb_parentMatchingOneOfTypes:acceptedParents];
 }
 
-- (XCElementSnapshot *)fb_parentMatchingOneOfTypes:(NSArray<NSNumber *> *)types
+- (id<FBXCElementSnapshot>)fb_parentMatchingOneOfTypes:(NSArray<NSNumber *> *)types
 {
-  return [self fb_parentMatchingOneOfTypes:types filter:^(XCElementSnapshot *snapshot) {
+  return [self fb_parentMatchingOneOfTypes:types filter:^(id<FBXCElementSnapshot> snapshot) {
     return YES;
   }];
 }
 
-- (XCElementSnapshot *)fb_parentMatchingOneOfTypes:(NSArray<NSNumber *> *)types filter:(BOOL(^)(XCElementSnapshot *snapshot))filter
+- (id<FBXCElementSnapshot>)fb_parentMatchingOneOfTypes:(NSArray<NSNumber *> *)types
+                                              filter:(BOOL(^)(id<FBXCElementSnapshot> snapshot))filter
 {
-  XCElementSnapshot *snapshot = self.parent;
+  id<FBXCElementSnapshot> snapshot = self.parent;
   while (snapshot && !(isSnapshotTypeAmongstGivenTypes(snapshot, types) && filter(snapshot))) {
     snapshot = snapshot.parent;
   }
@@ -73,12 +77,15 @@ inline static BOOL areValuesEqualOrBlank(id value1, id value2);
 
 inline static BOOL isNilOrEmpty(id value);
 
-- (BOOL)fb_framelessFuzzyMatchesElement:(XCElementSnapshot *)snapshot
+- (BOOL)fb_framelessFuzzyMatchesElement:(id<FBXCElementSnapshot>)snapshot
 {
     // Pure payload-based comparison sometimes yield false negatives, therefore relying on it only if all of the identifying properties are blank
-  if (isNilOrEmpty(self.identifier) && isNilOrEmpty(self.title) && isNilOrEmpty(self.label) &&
-      isNilOrEmpty(self.value) && isNilOrEmpty(self.placeholderValue)) {
-    return [self.wdUID isEqualToString:(snapshot.wdUID ?: @"")];
+  if (isNilOrEmpty(self.identifier)
+      && isNilOrEmpty(self.title)
+      && isNilOrEmpty(self.label)
+      && isNilOrEmpty(self.value)
+      && isNilOrEmpty(self.placeholderValue)) {
+    return [self.wdUID isEqualToString:([FBXCElementSnapshotWrapper ensureWrapped:snapshot].wdUID ?: @"")];
   }
   
   // Sometimes value and placeholderValue of a correct match from different snapshots are not the same (one is nil and one is a blank string)
@@ -91,27 +98,27 @@ inline static BOOL isNilOrEmpty(id value);
     areValuesEqualOrBlank(self.placeholderValue, snapshot.placeholderValue);
 }
 
-- (NSArray<XCElementSnapshot *> *)fb_descendantsCellSnapshots
+- (NSArray<id<FBXCElementSnapshot>> *)fb_descendantsCellSnapshots
 {
-  NSArray<XCElementSnapshot *> *cellSnapshots = [self fb_descendantsMatchingType:XCUIElementTypeCell];
-    
+  NSArray<id<FBXCElementSnapshot>> *cellSnapshots = [self fb_descendantsMatchingType:XCUIElementTypeCell];
+  
   if (cellSnapshots.count == 0) {
-      // For the home screen, cells are actually of type XCUIElementTypeIcon
-      cellSnapshots = [self fb_descendantsMatchingType:XCUIElementTypeIcon];
+    // For the home screen, cells are actually of type XCUIElementTypeIcon
+    cellSnapshots = [self fb_descendantsMatchingType:XCUIElementTypeIcon];
   }
-   
+  
   if (cellSnapshots.count == 0) {
-      // In some cases XCTest will not report Cell Views. In that case grab all descendants and try to figure out scroll directon from them.
-      cellSnapshots = self._allDescendants;
+    // In some cases XCTest will not report Cell Views. In that case grab all descendants and try to figure out scroll directon from them.
+    cellSnapshots = self._allDescendants;
   }
   
-    return cellSnapshots;
+  return cellSnapshots;
 }
 
-- (NSArray<XCElementSnapshot *> *)fb_ancestors
+- (NSArray<id<FBXCElementSnapshot>> *)fb_ancestors
 {
-  NSMutableArray<XCElementSnapshot *> *ancestors = [NSMutableArray array];
-  XCElementSnapshot *parent = self.parent;
+  NSMutableArray<id<FBXCElementSnapshot>> *ancestors = [NSMutableArray array];
+  id<FBXCElementSnapshot> parent = self.parent;
   while (parent) {
     [ancestors addObject:parent];
     parent = parent.parent;
@@ -119,18 +126,18 @@ inline static BOOL isNilOrEmpty(id value);
   return ancestors.copy;
 }
 
-- (XCElementSnapshot *)fb_parentCellSnapshot
+- (id<FBXCElementSnapshot>)fb_parentCellSnapshot
 {
-    XCElementSnapshot *targetCellSnapshot = self;
-    // XCUIElementTypeIcon is the cell type for homescreen icons
-    NSArray<NSNumber *> *acceptableElementTypes = @[
-                                                    @(XCUIElementTypeCell),
-                                                    @(XCUIElementTypeIcon),
-                                                    ];
-    if (self.elementType != XCUIElementTypeCell && self.elementType != XCUIElementTypeIcon) {
-        targetCellSnapshot = [self fb_parentMatchingOneOfTypes:acceptableElementTypes];
-    }
-    return targetCellSnapshot;
+  id<FBXCElementSnapshot> targetCellSnapshot = self.snapshot;
+  // XCUIElementTypeIcon is the cell type for homescreen icons
+  NSArray<NSNumber *> *acceptableElementTypes = @[
+                                                  @(XCUIElementTypeCell),
+                                                  @(XCUIElementTypeIcon),
+                                                  ];
+  if (self.elementType != XCUIElementTypeCell && self.elementType != XCUIElementTypeIcon) {
+      targetCellSnapshot = [self fb_parentMatchingOneOfTypes:acceptableElementTypes];
+  }
+  return targetCellSnapshot;
 }
 
 - (CGRect)fb_visibleFrameWithFallback
@@ -156,9 +163,20 @@ inline static BOOL isNilOrEmpty(id value);
   return thisVisibleFrame;
 }
 
+- (NSValue *)fb_hitPoint
+{
+  NSError *error;
+  XCUIHitPointResult *result = [self hitPoint:&error];
+  if (nil != error) {
+    [FBLogger logFmt:@"Failed to fetch hit point for %@ - %@", self.fb_description, error.localizedDescription];
+    return nil;
+  }
+  return [NSValue valueWithCGPoint:result.hitPoint];
+}
+
 @end
 
-inline static BOOL isSnapshotTypeAmongstGivenTypes(XCElementSnapshot* snapshot, NSArray<NSNumber *> *types)
+inline static BOOL isSnapshotTypeAmongstGivenTypes(id<FBXCElementSnapshot> snapshot, NSArray<NSNumber *> *types)
 {
   for (NSUInteger i = 0; i < types.count; i++) {
    if([@(snapshot.elementType) isEqual: types[i]] || [types[i] isEqual: @(XCUIElementTypeAny)]){

package/WebDriverAgentLib/Categories/NSExpression+FBFormat.m

@@ -8,6 +8,7 @@
  */
 
 #import "NSExpression+FBFormat.h"
+
 #import "FBElementUtils.h"
 
 @implementation NSExpression (FBFormat)
@@ -17,14 +18,18 @@
   if ([input expressionType] != NSKeyPathExpressionType) {
     return input;
   }
+  
   NSString *propName = [input keyPath];
   NSUInteger dotPos = [propName rangeOfString:@"."].location;
-  if (NSNotFound != dotPos) {
+  NSString *wdPropName;
+  if (NSNotFound == dotPos) {
+    wdPropName = [FBElementUtils wdAttributeNameForAttributeName:propName];
+  } else {
     NSString *actualPropName = [propName substringToIndex:dotPos];
     NSString *suffix = [propName substringFromIndex:(dotPos + 1)];
-    return [NSExpression expressionForKeyPath:[NSString stringWithFormat:@"%@.%@", [FBElementUtils wdAttributeNameForAttributeName:actualPropName], suffix]];
+    wdPropName = [NSString stringWithFormat:@"%@.%@", [FBElementUtils wdAttributeNameForAttributeName:actualPropName], suffix];
   }
-  return [NSExpression expressionForKeyPath:[FBElementUtils wdAttributeNameForAttributeName:propName]];
+  return [NSExpression expressionForKeyPath:wdPropName];
 }
 
 @end

package/WebDriverAgentLib/Categories/XCUIApplication+FBAlert.m

@@ -10,6 +10,7 @@
 #import "XCUIApplication+FBAlert.h"
 
 #import "FBMacros.h"
+#import "FBXCElementSnapshotWrapper+Helpers.h"
 #import "FBXCodeCompatibility.h"
 #import "XCUIElement+FBUtilities.h"
 
@@ -21,10 +22,10 @@ NSString *const FB_SAFARI_APP_NAME = @"Safari";
 @implementation XCUIApplication (FBAlert)
 
 - (nullable XCUIElement *)fb_alertElementFromSafariWithScrollView:(XCUIElement *)scrollView
-                                                     viewSnapshot:(XCElementSnapshot *)viewSnapshot
+                                                     viewSnapshot:(id<FBXCElementSnapshot>)viewSnapshot
 {
   CGRect appFrame = viewSnapshot.frame;
-  NSPredicate *dstViewMatchPredicate = [NSPredicate predicateWithBlock:^BOOL(XCElementSnapshot *snapshot, NSDictionary *bindings) {
+  NSPredicate *dstViewMatchPredicate = [NSPredicate predicateWithBlock:^BOOL(id<FBXCElementSnapshot> snapshot, NSDictionary *bindings) {
     CGRect curFrame = snapshot.frame;
     if (!CGRectEqualToRect(appFrame, curFrame)
         && curFrame.origin.x > 0 && curFrame.size.width < appFrame.size.width) {
@@ -62,8 +63,8 @@ NSString *const FB_SAFARI_APP_NAME = @"Safari";
   // and conatins at least one text view
   __block NSUInteger buttonsCount = 0;
   __block NSUInteger textViewsCount = 0;
-  XCElementSnapshot *snapshot = candidate.fb_cachedSnapshot ?: candidate.fb_takeSnapshot;
-  [snapshot enumerateDescendantsUsingBlock:^(XCElementSnapshot *descendant) {
+  id<FBXCElementSnapshot> snapshot = candidate.fb_cachedSnapshot ?: candidate.fb_takeSnapshot;
+  [snapshot enumerateDescendantsUsingBlock:^(id<FBXCElementSnapshot> descendant) {
     XCUIElementType curType = descendant.elementType;
     if (curType == XCUIElementTypeButton) {
       buttonsCount++;
@@ -83,7 +84,7 @@ NSString *const FB_SAFARI_APP_NAME = @"Safari";
   if (nil == alert) {
     return nil;
   }
-  XCElementSnapshot *alertSnapshot = alert.fb_cachedSnapshot ?: alert.fb_takeSnapshot;
+  id<FBXCElementSnapshot> alertSnapshot = alert.fb_cachedSnapshot ?: alert.fb_takeSnapshot;
 
   if (alertSnapshot.elementType == XCUIElementTypeAlert) {
     return alert;
@@ -96,7 +97,7 @@ NSString *const FB_SAFARI_APP_NAME = @"Safari";
 
     // In case of iPad we want to check if sheet isn't contained by popover.
     // In that case we ignore it.
-    XCElementSnapshot *ancestor = alertSnapshot.parent;
+    id<FBXCElementSnapshot> ancestor = alertSnapshot.parent;
     while (nil != ancestor) {
       if (nil != ancestor.identifier && [ancestor.identifier isEqualToString:@"PopoverDismissRegion"]) {
         return nil;
@@ -107,7 +108,7 @@ NSString *const FB_SAFARI_APP_NAME = @"Safari";
   }
 
   if (alertSnapshot.elementType == XCUIElementTypeScrollView) {
-    XCElementSnapshot *app = [alertSnapshot fb_parentMatchingType:XCUIElementTypeApplication];
+    id<FBXCElementSnapshot> app = [[FBXCElementSnapshotWrapper ensureWrapped:alertSnapshot] fb_parentMatchingType:XCUIElementTypeApplication];
     if (nil != app && [app.label isEqualToString:FB_SAFARI_APP_NAME]) {
       // Check alert presence in Safari web view
       return [self fb_alertElementFromSafariWithScrollView:alert viewSnapshot:alertSnapshot];

package/WebDriverAgentLib/Categories/XCUIApplication+FBHelpers.h

@@ -10,7 +10,7 @@
 #import <XCTest/XCTest.h>
 
 @class XCElementSnapshot;
-@class XCAccessibilityElement;
+@protocol FBXCAccessibilityElement;
 @class FBXMLGenerationOptions;
 
 NS_ASSUME_NONNULL_BEGIN
@@ -83,7 +83,7 @@ NS_ASSUME_NONNULL_BEGIN
  @param axElements the list of accessibility elements
  @returns The list of dictionaries. Each dictionary contains `bundleId` and `pid` items
  */
-+ (NSArray<NSDictionary<NSString *, id> *> *)fb_appsInfoWithAxElements:(NSArray<XCAccessibilityElement *> *)axElements;
++ (NSArray<NSDictionary<NSString *, id> *> *)fb_appsInfoWithAxElements:(NSArray<id<FBXCAccessibilityElement>> *)axElements;
 
 /**
  Retrieves the information about the currently active apps