Make screenshot more stable

meituan · Nov 5, 2018 · 733712d · 733712d
1 parent 85f450e
commit 733712d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/WebDriverAgentLib/Categories/XCUIDevice+FBHelpers.m b/WebDriverAgentLib/Categories/XCUIDevice+FBHelpers.m
@@ -22,6 +22,7 @@
 
 #import "XCUIDevice.h"
 #import "XCUIScreen.h"
+#import "FBConfiguration.h"
 
 static const NSTimeInterval FBHomeButtonCoolOffTime = 1.;
 static const NSTimeInterval FBScreenLockTimeout = 5.;
@@ -107,7 +108,7 @@ - (NSData *)fb_screenshotWithError:(NSError*__autoreleasing*)error
 {
   FBApplication *activeApplication = FBApplication.fb_activeApplication;
   UIInterfaceOrientation orientation = activeApplication.interfaceOrientation;
-  CGSize screenSize = FBAdjustDimensionsForApplication(activeApplication.frame.size, orientation);
+  CGSize screenSize = [FBConfiguration screenSize];
   CGRect screenRect = CGRectMake(0, 0, screenSize.width, screenSize.height);
   // https://developer.apple.com/documentation/xctest/xctimagequality?language=objc
   // Select lower quality, since XCTest crashes randomly if the maximum quality (zero value) is selected