Initialize WDA for UITesting when using devices

Summary: Configurations are copied around and we missed two places. :)

Reviewed By: nqmtuan

Differential Revision: D3834885

fbshipit-source-id: 5ccb320878ca8541560e796ee6588aaee6f3c13e

XCTestBootstrap/Bundles/FBApplicationDataPackage.m

@@ -130,10 +130,11 @@ - (FBApplicationDataPackage *)buildWithError:(NSError **)error
   }
   NSError *innerError;
   package.testConfiguration =
-  [[[[[[FBTestConfigurationBuilder builderWithFileManager:self.fileManager]
-       withModuleName:self.testBundle.name]
-      withSessionIdentifier:self.testBundle.configuration.sessionIdentifier]
-     withTestBundlePath:deviceTestBundlePath]
+  [[[[[[[FBTestConfigurationBuilder builderWithFileManager:self.fileManager]
+        withModuleName:self.testBundle.name]
+       withSessionIdentifier:self.testBundle.configuration.sessionIdentifier]
+      withTestBundlePath:deviceTestBundlePath]
+     withUITesting:self.testBundle.configuration.shouldInitializeForUITesting]
     saveAs:[localTestPlanDirPath stringByAppendingPathComponent:testConfigurationFileName]]
    buildWithError:&innerError];
   if (!package.testConfiguration) {
@@ -144,11 +145,12 @@ - (FBApplicationDataPackage *)buildWithError:(NSError **)error
   }
 
   package.testBundle =
-  [[[[[[FBTestBundleBuilder builderWithFileManager:self.fileManager]
-      withBundlePath:testBundlePath]
-     withSessionIdentifier:self.testBundle.configuration.sessionIdentifier]
-     withWorkingDirectory:workingDirectory]
-    withCodesignProvider:self.codesignProvider]
+  [[[[[[[FBTestBundleBuilder builderWithFileManager:self.fileManager]
+        withBundlePath:testBundlePath]
+       withSessionIdentifier:self.testBundle.configuration.sessionIdentifier]
+      withWorkingDirectory:workingDirectory]
+     withCodesignProvider:self.codesignProvider]
+    withUITesting:self.testBundle.configuration.shouldInitializeForUITesting]
    buildWithError:&innerError];
   if (!package.testBundle) {
     return

XCTestBootstrap/Bundles/FBTestConfiguration.h

@@ -36,6 +36,11 @@
  */
 @property (nonatomic, copy, readonly) NSString *path;
 
+/**
+ Determines whether should initialize for UITesting
+ */
+@property (nonatomic, assign, readonly) BOOL shouldInitializeForUITesting;
+
 @end
 
 /**

XCTestBootstrap/Bundles/FBTestConfiguration.m

@@ -19,6 +19,7 @@ @interface FBTestConfiguration ()
 @property (nonatomic, copy) NSString *moduleName;
 @property (nonatomic, copy) NSString *testBundlePath;
 @property (nonatomic, copy) NSString *path;
+@property (nonatomic, assign) BOOL shouldInitializeForUITesting;
 @end
 
 @implementation FBTestConfiguration
@@ -101,6 +102,7 @@ - (FBTestConfiguration *)buildWithError:(NSError **)error
   configuration.testBundlePath = self.testBundlePath;
   configuration.moduleName = self.moduleName;
   configuration.path = self.savePath;
+  configuration.shouldInitializeForUITesting = self.shouldInitializeForUITesting;
   return configuration;
 }