blob: e900b5be82d6c8bd54fec22bdfed1275f95fa8a2 (
plain)
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
|
//
// GTMNSThread+BlocksTest.m
//
// Copyright 2012 Google Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy
// of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations
// under the License.
//
#import "GTMSenTestCase.h"
#import "GTMNSThread+Blocks.h"
#if GTM_IPHONE_SDK || (MAC_OS_X_VERSION_MIN_REQUIRED >= MAC_OS_X_VERSION_10_5)
#import "GTMFoundationUnitTestingUtilities.h"
@interface GTMNSThread_BlocksTest : GTMTestCase {
@private
NSThread *workerThread_;
BOOL workerRunning_;
}
@property (nonatomic, readwrite, getter=isWorkerRunning) BOOL workerRunning;
@end
@implementation GTMNSThread_BlocksTest
@synthesize workerRunning = workerRunning_;
- (void)stopTestRunning:(GTMUnitTestingBooleanRunLoopContext *)context{
[context setShouldStop:YES];
}
- (void)workerMain:(id)object {
NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
while ([self isWorkerRunning]) {
[[NSRunLoop currentRunLoop] runMode:NSDefaultRunLoopMode
beforeDate:[NSDate distantFuture]];
}
[pool drain];
}
- (void)killWorkerThread:(GTMUnitTestingBooleanRunLoopContext *)context {
[self setWorkerRunning:NO];
[context setShouldStop:YES];
}
- (void)setUp {
[self setWorkerRunning:YES];
workerThread_ = [[NSThread alloc] initWithTarget:self
selector:@selector(workerMain:)
object:nil];
[workerThread_ start];
}
- (void)tearDown {
GTMUnitTestingBooleanRunLoopContext *context
= [GTMUnitTestingBooleanRunLoopContext context];
[self performSelector:@selector(killWorkerThread:)
onThread:workerThread_
withObject:context
waitUntilDone:NO];
NSRunLoop *runLoop = [NSRunLoop currentRunLoop];
STAssertTrue([runLoop gtm_runUpToSixtySecondsWithContext:context], nil);
[workerThread_ release];
}
- (void)testPerformBlock {
NSThread *currentThread = [NSThread currentThread];
GTMUnitTestingBooleanRunLoopContext *context
= [GTMUnitTestingBooleanRunLoopContext context];
[workerThread_ gtm_performBlock:^{
[self performSelector:@selector(stopTestRunning:)
onThread:currentThread
withObject:context
waitUntilDone:YES];
}];
NSRunLoop *runLoop = [NSRunLoop currentRunLoop];
STAssertTrue([runLoop gtm_runUpToSixtySecondsWithContext:context], nil);
}
- (void)testPerformBlockWaitUntilDone {
GTMUnitTestingBooleanRunLoopContext *context
= [GTMUnitTestingBooleanRunLoopContext context];
[workerThread_ gtm_performWaitingUntilDone:YES block:^{
[context setShouldStop:YES];
}];
STAssertTrue([context shouldStop], nil);
}
- (void)testPerformBlockInBackground {
NSThread *currentThread = [NSThread currentThread];
GTMUnitTestingBooleanRunLoopContext *context
= [GTMUnitTestingBooleanRunLoopContext context];
[NSThread gtm_performBlockInBackground:^{
[self performSelector:@selector(stopTestRunning:)
onThread:currentThread
withObject:context
waitUntilDone:YES];
}];
NSRunLoop *runLoop = [NSRunLoop currentRunLoop];
STAssertTrue([runLoop gtm_runUpToSixtySecondsWithContext:context], nil);
}
@end
#endif // GTM_IPHONE_SDK || (MAC_OS_X_VERSION_MIN_REQUIRED >= MAC_OS_X_VERSION_10_5)
|
