Skip to content

Port Xcode 9 XCTAttachment API integration from FB repo #19

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
delebedev wants to merge 5 commits into from
Closed

Port Xcode 9 XCTAttachment API integration from FB repo #19

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
4ea21c4
Implement Xcode 9 integration using XCTAttachement API
SiarheiFedartsou Jun 13, 2017
25feb78
Add tvOS support for Xcode 9 integration & some cleanup
SiarheiFedartsou Jun 22, 2017
951bb0d
s/attachement/attachment
delebedev Mar 10, 2018
d45a6ff
Conform to coding style
delebedev Mar 10, 2018
b85e09e
Pass recordMode as parameter
delebedev Mar 10, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 42 additions & 5 deletions FBSnapshotTestCase/FBSnapshotTestCase.m
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@

#import <FBSnapshotTestCase/FBSnapshotTestCase.h>
#import <FBSnapshotTestCase/FBSnapshotTestController.h>
#import <Availability.h>

@implementation FBSnapshotTestCase
{
Expand Down Expand Up @@ -122,17 +123,53 @@ - (NSString *)snapshotVerifyViewOrLayer:(id)viewOrLayer
}
}
}

if (!testSuccess) {
return [NSString stringWithFormat:@"Snapshot comparison failed: %@", errors.firstObject];
}

[self _addAttachmentsWithErrors:errors identifier:identifier ranInRecordMode:self.recordMode];

if (self.recordMode) {
return @"Test ran in record mode. Reference image is now saved. Disable record mode to perform an actual snapshot comparison!";
if (errors.count > 0) {
return [NSString stringWithFormat:@"Snapshot comparison failed: %@", errors.firstObject];
Copy link
Collaborator

@alanzeino alanzeino Mar 9, 2018

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is this possible? Looking at L102-L105, it is doing this:

     BOOL referenceImageSaved = [self _compareSnapshotOfViewOrLayer:viewOrLayer referenceImagesDirectory:referenceImagesDirectory identifier:(identifier) tolerance:tolerance error:&error];
     if (!referenceImageSaved) {
       [errors addObject:error];
     }

When self.recordMode is on, compareSnapshotOfViewOrLayer:... the BOOL from the function relates to whether or not the reference image was saved.

} else {
return @"Test ran in record mode. Reference image is now saved. Disable record mode to perform an actual snapshot comparison!";
}
} else if (!testSuccess) {
return [NSString stringWithFormat:@"Snapshot comparison failed: %@", errors.firstObject];
}

return nil;
}

- (void)_addAttachmentsWithErrors:(NSArray<NSError *> *)errors identifier:(NSString *)identifier ranInRecordMode:(BOOL)ranInRecordMode
{
#if defined(__IPHONE_11_0) || defined(__TVOS_11_0)
if (ranInRecordMode) {
UIImage *image = [_snapshotController referenceImageForSelector:self.invocation.selector identifier:identifier error:nil];
if (image) {
XCTAttachment *attachment = [XCTAttachment attachmentWithImage:image];
attachment.name = @"Reference Image";
[self addAttachment:attachment];
}
} else if (errors.firstObject != nil) {
NSError *error = errors.firstObject;
if (error.userInfo[FBReferenceImageKey] != nil) {
XCTAttachment *attachment = [XCTAttachment attachmentWithImage:error.userInfo[FBReferenceImageKey]];
attachment.name = @"Reference Image";
[self addAttachment:attachment];
}
if (error.userInfo[FBCapturedImageKey] != nil) {
XCTAttachment *attachment = [XCTAttachment attachmentWithImage:error.userInfo[FBCapturedImageKey]];
attachment.name = @"Captured Image";
[self addAttachment:attachment];
}
if (error.userInfo[FBDiffedImageKey] != nil) {
XCTAttachment *attachment = [XCTAttachment attachmentWithImage:error.userInfo[FBDiffedImageKey]];
attachment.name = @"Diffed Image";
[self addAttachment:attachment];
}
}
#endif
}

- (BOOL)compareSnapshotOfLayer:(CALayer *)layer
referenceImagesDirectory:(NSString *)referenceImagesDirectory
identifier:(NSString *)identifier
Expand Down
53 changes: 5 additions & 48 deletions FBSnapshotTestCase/SwiftSupport.swift
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,58 +11,15 @@
#if swift(>=3)
public extension FBSnapshotTestCase {
public func FBSnapshotVerifyView(_ view: UIView, identifier: String = "", suffixes: NSOrderedSet = FBSnapshotTestCaseDefaultSuffixes(), tolerance: CGFloat = 0, file: StaticString = #file, line: UInt = #line) {
FBSnapshotVerifyViewOrLayer(view, identifier: identifier, suffixes: suffixes, tolerance: tolerance, file: file, line: line)
if let errorDescription = snapshotVerifyViewOrLayer(view, identifier: identifier, suffixes: suffixes, tolerance: tolerance, defaultReferenceDirectory: FB_REFERENCE_IMAGE_DIR) {
XCTFail(errorDescription, file: file, line: line)
}
}

public func FBSnapshotVerifyLayer(_ layer: CALayer, identifier: String = "", suffixes: NSOrderedSet = FBSnapshotTestCaseDefaultSuffixes(), tolerance: CGFloat = 0, file: StaticString = #file, line: UInt = #line) {
FBSnapshotVerifyViewOrLayer(layer, identifier: identifier, suffixes: suffixes, tolerance: tolerance, file: file, line: line)
}

private func FBSnapshotVerifyViewOrLayer(_ viewOrLayer: AnyObject, identifier: String = "", suffixes: NSOrderedSet = FBSnapshotTestCaseDefaultSuffixes(), tolerance: CGFloat = 0, file: StaticString = #file, line: UInt = #line) {
Copy link
Collaborator

@alanzeino alanzeino Mar 9, 2018 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I believe there's a good reason why SwiftSupport.swift uses its own FBSnapshotVerifyViewOrLayer function instead of leveraging the Objective-C one. This should probably stay as is, with modifications to support this new attachment feature though.

let envReferenceImageDirectory = self.getReferenceImageDirectory(withDefault: FB_REFERENCE_IMAGE_DIR)
var error: NSError?
var comparisonSuccess = false

if let envReferenceImageDirectory = envReferenceImageDirectory {
for suffix in suffixes {
let referenceImagesDirectory = "\(envReferenceImageDirectory)\(suffix)"
if viewOrLayer.isKind(of: UIView.self) {
do {
try compareSnapshot(of: viewOrLayer as! UIView, referenceImagesDirectory: referenceImagesDirectory, identifier: identifier, tolerance: tolerance)
comparisonSuccess = true
} catch let error1 as NSError {
error = error1
comparisonSuccess = false
}
} else if viewOrLayer.isKind(of: CALayer.self) {
do {
try compareSnapshot(of: viewOrLayer as! CALayer, referenceImagesDirectory: referenceImagesDirectory, identifier: identifier, tolerance: tolerance)
comparisonSuccess = true
} catch let error1 as NSError {
error = error1
comparisonSuccess = false
}
} else {
assertionFailure("Only UIView and CALayer classes can be snapshotted")
}

assert(recordMode == false, message: "Test ran in record mode. Reference image is now saved. Disable record mode to perform an actual snapshot comparison!", file: file, line: line)

if comparisonSuccess || recordMode {
break
}

assert(comparisonSuccess, message: "Snapshot comparison failed: \(String(describing: error))", file: file, line: line)
if let errorDescription = snapshotVerifyViewOrLayer(layer, identifier: identifier, suffixes: suffixes, tolerance: tolerance, defaultReferenceDirectory: FB_REFERENCE_IMAGE_DIR) {
XCTFail(errorDescription, file: file, line: line)
}
} else {
XCTFail("Missing value for referenceImagesDirectory - Set FB_REFERENCE_IMAGE_DIR as Environment variable in your scheme.")
}
}

func assert(_ assertion: Bool, message: String, file: StaticString, line: UInt) {
if !assertion {
XCTFail(message, file: file, line: line)
}
}
}
#else
Expand Down