初始化提交

WebDriverAgentLib/Routing/FBXCElementSnapshotWrapper.m

@@ -0,0 +1,114 @@
+/**
+ * Copyright (c) 2015-present, Facebook, Inc.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#import "FBXCElementSnapshotWrapper.h"
+
+#import "FBElementUtils.h"
+
+#pragma clang diagnostic push
+#pragma clang diagnostic ignored "-Wobjc-protocol-property-synthesis"
+#pragma clang diagnostic ignored "-Wprotocol"
+
+@implementation FBXCElementSnapshotWrapper
+
+- (instancetype)initWithSnapshot:(id<FBXCElementSnapshot>)snapshot;
+{
+  self->_snapshot = snapshot;
+  return self;
+}
+
++ (instancetype)ensureWrapped:(id<FBXCElementSnapshot>)snapshot
+{
+  if (nil == snapshot) {
+    return nil;
+  }
+  return [(NSObject *)snapshot isKindOfClass:self.class]
+    ? (FBXCElementSnapshotWrapper *)snapshot
+    : [[FBXCElementSnapshotWrapper alloc] initWithSnapshot:snapshot];
+}
+
+// Attributes are queried most often,
+// so we prefer them to have direct accessors defined here
+// rather than to use message forwarding via forwardingTargetForSelector,
+// which is slow
+
+- (NSString *)identifier
+{
+  return self.snapshot.identifier;
+}
+
+- (CGRect)frame
+{
+  return self.snapshot.frame;
+}
+
+- (id)value
+{
+  return self.snapshot.value;
+}
+
+- (NSString *)title
+{
+  return self.snapshot.title;
+}
+
+- (NSString *)label
+{
+  return self.snapshot.label;
+}
+
+- (XCUIElementType)elementType
+{
+  return self.snapshot.elementType;
+}
+
+- (BOOL)isEnabled
+{
+  return self.snapshot.enabled;
+}
+
+- (XCUIUserInterfaceSizeClass)horizontalSizeClass
+{
+  return self.snapshot.horizontalSizeClass;
+}
+
+- (XCUIUserInterfaceSizeClass)verticalSizeClass
+{
+  return self.snapshot.verticalSizeClass;
+}
+
+- (NSString *)placeholderValue
+{
+  return self.snapshot.placeholderValue;
+}
+
+- (BOOL)isSelected
+{
+  return self.snapshot.selected;
+}
+
+#if !TARGET_OS_OSX
+- (BOOL)hasFocus
+{
+  return self.snapshot.hasFocus;
+}
+#endif
+
+- (id)forwardingTargetForSelector:(SEL)aSelector
+{
+  static dispatch_once_t onceToken;
+  static NSSet<NSString *> *names;
+  dispatch_once(&onceToken, ^{
+    names = [FBElementUtils selectorNamesWithProtocol:@protocol(FBXCElementSnapshot)];
+  });
+  return [names containsObject:NSStringFromSelector(aSelector)] ? self.snapshot : nil;
+}
+
+@end
+
+#pragma clang diagnostic pop