1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
|
// 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.
#import <UIKit/UIKit.h>
#include "base/debug/debugger.h"
#include "base/logging.h"
#include "base/mac/scoped_nsautorelease_pool.h"
#include "base/mac/scoped_nsobject.h"
#include "base/message_loop/message_loop.h"
#include "base/message_loop/message_pump_default.h"
#include "base/test/test_suite.h"
// Springboard will kill any iOS app that fails to check in after launch within
// a given time. Starting a UIApplication before invoking TestSuite::Run
// prevents this from happening.
// InitIOSRunHook saves the TestSuite and argc/argv, then invoking
// RunTestsFromIOSApp calls UIApplicationMain(), providing an application
// delegate class: ChromeUnitTestDelegate. The delegate implements
// application:didFinishLaunchingWithOptions: to invoke the TestSuite's Run
// method.
// 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 MainHook are being
// run in a row, this provides an indication of which one is currently running.
static base::TestSuite* g_test_suite = NULL;
static int g_argc;
static char** g_argv;
@interface UIApplication (Testing)
- (void) _terminateWithStatus:(int)status;
@end
@interface ChromeUnitTestDelegate : NSObject {
@private
base::scoped_nsobject<UIWindow> window_;
}
- (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.
window_.reset([[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 = NSTextAlignmentCenter;
[window_ addSubview:label];
if ([self shouldRedirectOutputToFile])
[self redirectOutput];
// Queue up the test run.
[self performSelector:@selector(runTests)
withObject:nil
afterDelay:0.1];
return YES;
}
// Returns true if the gtest output should be redirected to a file, then sent
// to NSLog when compleete. This redirection is used because gtest only writes
// output to stdout, but results must be written to NSLog in order to show up in
// the device log that is retrieved from the device by the host.
- (BOOL)shouldRedirectOutputToFile {
#if !TARGET_IPHONE_SIMULATOR
return !base::debug::BeingDebugged();
#endif // TARGET_IPHONE_SIMULATOR
return NO;
}
// Returns the path to the directory to store gtest output files.
- (NSString*)outputPath {
NSArray* searchPath =
NSSearchPathForDirectoriesInDomains(NSDocumentDirectory,
NSUserDomainMask,
YES);
CHECK([searchPath count] > 0) << "Failed to get the Documents folder";
return [searchPath objectAtIndex:0];
}
// Returns the path to file that stdout is redirected to.
- (NSString*)stdoutPath {
return [[self outputPath] stringByAppendingPathComponent:@"stdout.log"];
}
// Returns the path to file that stderr is redirected to.
- (NSString*)stderrPath {
return [[self outputPath] stringByAppendingPathComponent:@"stderr.log"];
}
// Redirects stdout and stderr to files in the Documents folder in the app's
// sandbox.
- (void)redirectOutput {
freopen([[self stdoutPath] UTF8String], "w+", stdout);
freopen([[self stderrPath] UTF8String], "w+", stderr);
}
// Reads the redirected gtest output from a file and writes it to NSLog.
- (void)writeOutputToNSLog {
// Close the redirected stdout and stderr files so that the content written to
// NSLog doesn't end up in these files.
fclose(stdout);
fclose(stderr);
for (NSString* path in @[ [self stdoutPath], [self stderrPath]]) {
NSString* content = [NSString stringWithContentsOfFile:path
encoding:NSUTF8StringEncoding
error:NULL];
NSArray* lines = [content componentsSeparatedByCharactersInSet:
[NSCharacterSet newlineCharacterSet]];
NSLog(@"Writing contents of %@ to NSLog", path);
for (NSString* line in lines) {
NSLog(@"%@", line);
}
}
}
- (void)runTests {
int exitStatus = g_test_suite->Run();
if ([self shouldRedirectOutputToFile])
[self writeOutputToNSLog];
// 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.
// TODO(ios): crbug.com/137010 Figure out how much time is actually needed,
// and sleep only to make sure that much time has elapsed since launch.
[NSThread sleepUntilDate:[NSDate dateWithTimeIntervalSinceNow:2.0]];
window_.reset();
// 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];
[application _terminateWithStatus:exitStatus];
exit(exitStatus);
}
@end
namespace {
scoped_ptr<base::MessagePump> CreateMessagePumpForUIForTests() {
// A default MessagePump will do quite nicely in tests.
return scoped_ptr<base::MessagePump>(new base::MessagePumpDefault());
}
} // namespace
namespace base {
void InitIOSTestMessageLoop() {
MessageLoop::InitMessagePumpForUIFactory(&CreateMessagePumpForUIForTests);
}
void InitIOSRunHook(TestSuite* suite, int argc, char* argv[]) {
g_test_suite = suite;
g_argc = argc;
g_argv = argv;
}
void RunTestsFromIOSApp() {
// When TestSuite::Run is invoked it calls RunTestsFromIOSApp(). On the first
// invocation, this method fires up an iOS app via UIApplicationMain. Since
// UIApplicationMain does not return until the app exits, control does not
// return to the initial TestSuite::Run invocation, so the app invokes
// TestSuite::Run a second time and since |ran_hook| is true at this point,
// this method is a no-op and control returns to TestSuite:Run so that test
// are executed. Once the app exits, RunTestsFromIOSApp calls exit() so that
// control is not returned to the initial invocation of TestSuite::Run.
static bool ran_hook = false;
if (!ran_hook) {
ran_hook = true;
mac::ScopedNSAutoreleasePool pool;
int exit_status = UIApplicationMain(g_argc, g_argv, nil,
@"ChromeUnitTestDelegate");
exit(exit_status);
}
}
} // namespace base
|
