Files
Monkey-Test/WebDriverAgent/WebDriverAgentLib/Routing/FBElementCache.m
T
2019-01-25 17:01:52 +08:00

60 lines
1.1 KiB
Objective-C
Executable File

/**
* 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. An additional grant
* of patent rights can be found in the PATENTS file in the same directory.
*/
#import "FBElementCache.h"
#import "FBAlert.h"
#import "XCUIElement.h"
#import "XCUIElement+FBUtilities.h"
@interface FBElementCache ()
@property (atomic, strong) NSMutableDictionary *elementCache;
@end
@implementation FBElementCache
- (instancetype)init
{
self = [super init];
if (!self) {
return nil;
}
_elementCache = [[NSMutableDictionary alloc] init];
return self;
}
- (NSString *)storeElement:(XCUIElement *)element
{
NSString *uuid = [[NSUUID UUID] UUIDString];
self.elementCache[uuid] = element;
return uuid;
}
- (XCUIElement *)elementForUUID:(NSString *)uuid
{
if (!uuid) {
return nil;
}
XCUIElement *element = self.elementCache[uuid];
[element resolve];
return element;
}
- (void)clear
{
[self.elementCache removeAllObjects];
}
- (NSUInteger)count
{
return [self.elementCache count];
}
@end