67 lines
1.7 KiB
Objective-C
67 lines
1.7 KiB
Objective-C
/**
|
|
* 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 "RCTTestModule.h"
|
|
|
|
#import "FBSnapshotTestController.h"
|
|
#import "RCTAssert.h"
|
|
#import "RCTLog.h"
|
|
#import "RCTUIManager.h"
|
|
|
|
@implementation RCTTestModule
|
|
{
|
|
NSMutableDictionary *_snapshotCounter;
|
|
}
|
|
|
|
@synthesize bridge = _bridge;
|
|
|
|
RCT_EXPORT_MODULE()
|
|
|
|
- (dispatch_queue_t)methodQueue
|
|
{
|
|
return _bridge.uiManager.methodQueue;
|
|
}
|
|
|
|
- (instancetype)init
|
|
{
|
|
if ((self = [super init])) {
|
|
_snapshotCounter = [NSMutableDictionary new];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
RCT_EXPORT_METHOD(verifySnapshot:(RCTResponseSenderBlock)callback)
|
|
{
|
|
RCTAssert(_controller != nil, @"No snapshot controller configured.");
|
|
|
|
[_bridge.uiManager addUIBlock:^(RCTUIManager *uiManager, RCTSparseArray *viewRegistry) {
|
|
|
|
NSString *testName = NSStringFromSelector(_testSelector);
|
|
_snapshotCounter[testName] = [@([_snapshotCounter[testName] integerValue] + 1) stringValue];
|
|
|
|
NSError *error = nil;
|
|
BOOL success = [_controller compareSnapshotOfView:_view
|
|
selector:_testSelector
|
|
identifier:_snapshotCounter[testName]
|
|
error:&error];
|
|
|
|
RCTAssert(success, @"Snapshot comparison failed: %@", error);
|
|
callback(@[]);
|
|
}];
|
|
}
|
|
|
|
RCT_EXPORT_METHOD(markTestCompleted)
|
|
{
|
|
[_bridge.uiManager addUIBlock:^(RCTUIManager *uiManager, RCTSparseArray *viewRegistry) {
|
|
_done = YES;
|
|
}];
|
|
}
|
|
|
|
@end
|