Create window in unittests with Scene API

Change-Id: I4996afbbc8b56b59e979a5f2e2c5a136f281bf6a
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2403361
Commit-Queue: Olivier Robin <olivierrobin@chromium.org>
Reviewed-by: Mark Cogan <marq@chromium.org>
Reviewed-by: Rohit Rao <rohitrao@chromium.org>
Auto-Submit: Olivier Robin <olivierrobin@chromium.org>
Cr-Commit-Position: refs/heads/master@{#806106}

base/test/test_support_ios.mm

@@ -35,6 +35,28 @@ static base::TestSuite* g_test_suite = NULL;
 static int g_argc;
 static char** g_argv;
 
+namespace {
+void PopulateUIWindow(UIWindow* window) {
+  [window setBackgroundColor:[UIColor whiteColor]];
+  [window makeKeyAndVisible];
+  CGRect bounds = [[UIScreen mainScreen] bounds];
+  // Add a label with the app name.
+  UILabel* label = [[[UILabel alloc] initWithFrame:bounds] autorelease];
+  label.text = [[NSProcessInfo processInfo] processName];
+  label.textAlignment = NSTextAlignmentCenter;
+  [window addSubview:label];
+
+  // An NSInternalInconsistencyException is thrown if the app doesn't have a
+  // root view controller. Set an empty one here.
+  [window setRootViewController:[[[UIViewController alloc] init] autorelease]];
+}
+
+bool IsSceneStartupEnabled() {
+  return [[NSBundle mainBundle].infoDictionary
+      objectForKey:@"UIApplicationSceneManifest"];
+}
+}
+
 @interface UIApplication (Testing)
 - (void)_terminateWithStatus:(int)status;
 @end
@@ -54,10 +76,10 @@ static char** g_argv;
 // the application delegate, so they need to be separate objects (the same
 // object can't be both the app and scene delegate, since new scene delegates
 // are created for each scene).
-@interface ChromeUnitTestSceneDelegate : NSObject <UISceneDelegate>
-@end
+@interface ChromeUnitTestSceneDelegate : NSObject <UIWindowSceneDelegate> {
+  base::scoped_nsobject<UIWindow> _window;
+}
 
-@implementation ChromeUnitTestSceneDelegate
 @end
 
 @interface ChromeUnitTestDelegate : NSObject <GoogleTestRunnerDelegate> {
@@ -66,6 +88,26 @@ static char** g_argv;
 - (void)runTests;
 @end
 
+@implementation ChromeUnitTestSceneDelegate
+
+- (void)scene:(UIScene*)scene
+    willConnectToSession:(UISceneSession*)session
+                 options:(UISceneConnectionOptions*)connectionOptions
+    API_AVAILABLE(ios(13)) {
+  // Unittests do not support multiple scenes.
+  DCHECK(![[UIApplication sharedApplication] supportsMultipleScenes]);
+  // Yes, this is leaked, it's just to make what's running visible.
+  _window.reset([[UIWindow alloc]
+      initWithWindowScene:static_cast<UIWindowScene*>(scene)]);
+  PopulateUIWindow(_window);
+}
+
+- (void)sceneDidDisconnect:(UIScene*)scene API_AVAILABLE(ios(13)) {
+  _window.reset();
+}
+
+@end
+
 @implementation ChromeUnitTestDelegate
 
 - (BOOL)application:(UIApplication *)application
@@ -80,22 +122,13 @@ static char** g_argv;
   [[UIKeyboardImpl sharedInstance] setSoftwareKeyboardShownByTouch:YES];
 #endif  // TARGET_IPHONE_SIMULATOR
 
-  CGRect bounds = [[UIScreen mainScreen] bounds];
+  if (!IsSceneStartupEnabled()) {
+    CGRect bounds = [[UIScreen mainScreen] bounds];
 
-  // Yes, this is leaked, it's just to make what's running visible.
-  _window.reset([[UIWindow alloc] initWithFrame:bounds]);
-  [_window setBackgroundColor:[UIColor whiteColor]];
-  [_window makeKeyAndVisible];
-
-  // Add a label with the app name.
-  UILabel* label = [[[UILabel alloc] initWithFrame:bounds] autorelease];
-  label.text = [[NSProcessInfo processInfo] processName];
-  label.textAlignment = NSTextAlignmentCenter;
-  [_window addSubview:label];
-
-  // An NSInternalInconsistencyException is thrown if the app doesn't have a
-  // root view controller. Set an empty one here.
-  [_window setRootViewController:[[[UIViewController alloc] init] autorelease]];
+    // Yes, this is leaked, it's just to make what's running visible.
+    _window.reset([[UIWindow alloc] initWithFrame:bounds]);
+    PopulateUIWindow(_window);
+  }
 
   if ([self shouldRedirectOutputToFile])
     [self redirectOutput];