Index: base/test/main_hook_ios.mm |
diff --git a/base/test/main_hook_ios.mm b/base/test/main_hook_ios.mm |
new file mode 100644 |
index 0000000000000000000000000000000000000000..c94998f0f2a01758e96178b0856bb2a4ee7e40bc |
--- /dev/null |
+++ b/base/test/main_hook_ios.mm |
@@ -0,0 +1,97 @@ |
+// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "base/logging.h" |
+#include "base/test/main_hook.h" |
+#include "base/debug/debugger.h" |
+ |
+#import <UIKit/UIKit.h> |
+ |
+// Springboard will kill any iOS app that fails to check in after launch within |
+// a given time. These two classes prevent this from happening. |
+ |
+// MainHook saves the chrome main() and calls UIApplicationMain(), |
+// providing an application delegate class: ChromeUnitTestDelegate. The delegate |
+// listens for UIApplicationDidFinishLaunchingNotification. When the |
+// notification is received, it fires main() again to have the real work done. |
+ |
+// Since the executable isn't likely to be a real iOS UI, the delegate puts up a |
+// window displaying the app name. If a bunch of apps using this are being run |
+// in a row, this provides an indication of which one is currently running. |
+// (i.e.-Run All Tests) |
stuartmorgan
2012/07/12 09:36:31
Remove this last line since it's subject to becomi
leng
2012/07/12 10:44:42
Done.
|
+ |
+// NOTE: This takes care of one of the iOS timers, but the watchdog can still |
+// kill the app if the messageloop is starved too long. A potential solution to |
+// this second problem would be to break the running of the main logic |
+// (unittest, etc.) into chunks via a timer so the iOS eventloop can drive it. |
stuartmorgan
2012/07/12 09:36:31
I think this comment predates test_listener_ios; i
leng
2012/07/12 10:44:42
Done.
|
+ |
+static MainHook::MainType g_main_func = NULL; |
+static int g_argc; |
+static char** g_argv; |
+ |
+@interface ChromeUnitTestDelegate : NSObject |
+- (void)runTests; |
+@end |
+ |
+@implementation ChromeUnitTestDelegate |
+ |
+- (BOOL)application:(UIApplication *)application |
+ didFinishLaunchingWithOptions:(NSDictionary *)launchOptions { |
+ |
+ CGRect bounds = [[UIScreen mainScreen] bounds]; |
+ |
+ // Yes, this is leaked, it's just to make what's running visible. |
+ UIWindow* window = [[UIWindow alloc] initWithFrame:bounds]; |
+ [window makeKeyAndVisible]; |
+ |
+ // Add a label with the app name. |
+ UILabel* label = [[[UILabel alloc] initWithFrame:bounds] autorelease]; |
+ label.text = [[NSProcessInfo processInfo] processName]; |
+ label.textAlignment = UITextAlignmentCenter; |
+ [window addSubview:label]; |
+ |
+ // Queue up the test run. |
+ [self performSelector:@selector(runTests) |
+ withObject:nil |
+ afterDelay:0.1]; |
+ return YES; |
+} |
+ |
+- (void)runTests { |
+ int exitStatus = g_main_func(g_argc, g_argv); |
+ |
+ // If a test app is too fast, it will exit before Instruments has has a |
+ // a chance to initialize and no test results will be seen. |
+ [NSThread sleepUntilDate: [NSDate dateWithTimeIntervalSinceNow:2.0]]; |
stuartmorgan
2012/07/12 09:36:31
Remove the space after :
We should add a TODO: wi
leng
2012/07/12 10:44:42
Done.
|
+ |
+ // Use the hidden selector to try and cleanly take down the app (otherwise |
+ // things can think the app crashed even on a zero exit status). |
+ UIApplication* application = [UIApplication sharedApplication]; |
+ if ([application respondsToSelector:@selector(_terminateWithStatus:)]) { |
+ [application performSelector:@selector(_terminateWithStatus:) |
+ withObject:(id)exitStatus]; |
+ } |
+ exit(exitStatus); |
+} |
+ |
+@end |
+ |
+#pragma mark - |
+ |
+MainHook::MainHook(MainType main_func, int argc, char* argv[]) { |
+ static bool ran_hook; |
stuartmorgan
2012/07/12 09:36:31
= false
leng
2012/07/12 10:44:42
Done.
|
+ if (!ran_hook) { |
+ ran_hook = true; |
+ |
+ g_main_func = main_func; |
+ g_argc = argc; |
+ g_argv = argv; |
+ |
+ NSAutoreleasePool * pool = [[NSAutoreleasePool alloc] init]; |
stuartmorgan
2012/07/12 09:36:31
We may as well use a scoped autorelease pool here.
leng
2012/07/12 10:44:42
Done.
|
+ int exitStatus = UIApplicationMain(argc, argv, nil, |
+ @"ChromeUnitTestDelegate"); |
+ [pool release]; |
+ exit(exitStatus); |
+ } |
+} |