Add pixel threshold pixel option to screenshot matching

kif-framework · Dec 5, 2014 · 10485a9 · 10485a9
1 parent 2215e32
commit 10485a9
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/Lela/KIFUITestActor+Lela.h b/Lela/KIFUITestActor+Lela.h
@@ -8,8 +8,11 @@
 
 #import <KIFUITestActor.h>
 
+extern NSString * const LECompareOptionThresholdPixels;
+
 @interface KIFUITestActor (Lela)
 
 - (void)expectScreenToMatchImageNamed:(NSString *)name;
 - (void)expectScreenToMatchImageNamed:(NSString *)name options:(NSDictionary *)options;
+
 @end
diff --git a/Lela/KIFUITestActor+Lela.m b/Lela/KIFUITestActor+Lela.m
@@ -11,6 +11,8 @@
 #import "Lela.h"
 #import <XCTest/XCTest.h>
 
+NSString * const LECompareOptionThresholdPixels = @"LECompareOptionThresholdPixels";
+
 @implementation KIFUITestActor (Lela)
 
 + (NSString *)lelaTestRunName

diff --git a/Lela/Lela.mm b/Lela/Lela.mm
@@ -121,6 +121,7 @@ + (BOOL)compareExpectedImage:(UIImage *)expected toActual:(UIImage *)actual opti
     args.ImgA = RGBAImage::ReadFromUIImage(expected);
     args.ImgB = RGBAImage::ReadFromUIImage(actual);
     args.ImgDiff = new RGBAImage(args.ImgA->Get_Width(), args.ImgA->Get_Height(), "Output");
+    args.ThresholdPixels = [options[LECompareOptionThresholdPixels] unsignedIntValue];
 
     BOOL success = Yee_Compare(args);