1 files changed, 110 insertions, 0 deletions
diff --git a/DebugUtils/GTMDebugThreadValidationTest.m b/DebugUtils/GTMDebugThreadValidationTest.m
new file mode 100644
index 0000000..66bd51b
--- /dev/null
+++ b/DebugUtils/GTMDebugThreadValidationTest.m
@@ -0,0 +1,110 @@
+//
+//  GTMDebugThreadValidationTest.m
+//  Copyright 2008 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 "GTMDebugThreadValidation.h"
+
+// GTMDebugThreadValidation only happens on debug builds
+#if DEBUG
+
+@interface GTMDebugThreadValidationTest : GTMTestCase
+@end
+
+// A cheap flag for knowing when our thread has run
+
+static volatile BOOL gGTMDebugThreadValidationTestDone = NO;
+
+// This is an assertion handler that just records that an assertion has fired.
+@interface GTMDebugThreadValidationCheckAssertionHandler : NSAssertionHandler {
+ @private
+  BOOL handledAssertion_;
+}
+- (void)handleFailureInMethod:(SEL)selector 
+                       object:(id)object 
+                         file:(NSString *)fileName 
+                   lineNumber:(NSInteger)line 
+                  description:(NSString *)format,...;
+
+- (void)handleFailureInFunction:(NSString *)functionName 
+                           file:(NSString *)fileName 
+                     lineNumber:(NSInteger)line 
+                    description:(NSString *)format,...;
+- (BOOL)didHandleAssertion;
+@end
+
+@implementation GTMDebugThreadValidationTest
+- (void)testOnMainThread {
+  STAssertNoThrow(GTMAssertRunningOnMainThread(), nil);
+}
+
+- (void)threadFunc:(NSMutableString *)result {
+  NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
+  // We'll insert our own assertion handler that will get called on the assert
+  // so that we don't have to worry about the log, and exception being thrown.
+  GTMDebugThreadValidationCheckAssertionHandler *handler = 
+    [[[GTMDebugThreadValidationCheckAssertionHandler alloc] init] autorelease];
+  NSMutableDictionary *threadDictionary 
+    = [[NSThread currentThread] threadDictionary];
+  [threadDictionary setObject:handler forKey:@"NSAssertionHandler"];
+  GTMAssertRunningOnMainThread();
+  if ([handler didHandleAssertion]) {
+    [result setString:@"ASSERTED"];
+  }
+  [threadDictionary removeObjectForKey:@"NSAssertionHandler"];
+  gGTMDebugThreadValidationTestDone = YES;
+  [pool release];
+}
+
+- (void)testOnOtherThread {
+  NSMutableString *result = [NSMutableString string];
+  gGTMDebugThreadValidationTestDone = NO;
+  [NSThread detachNewThreadSelector:@selector(threadFunc:)
+                           toTarget:self
+                         withObject:result];
+  NSRunLoop *loop = [NSRunLoop currentRunLoop];
+  
+  while (!gGTMDebugThreadValidationTestDone) {
+    NSDate *date = [NSDate dateWithTimeIntervalSinceNow:0.01];
+    [loop runUntilDate:date];
+  }
+  STAssertEqualStrings(result, @"ASSERTED", @"GTMAssertRunningOnMainThread did "
+                       @"not assert while running on another thread");
+}
+@end
+
+@implementation GTMDebugThreadValidationCheckAssertionHandler
+
+- (void)handleFailureInMethod:(SEL)selector 
+                       object:(id)object 
+                         file:(NSString *)fileName 
+                   lineNumber:(NSInteger)line 
+                  description:(NSString *)format,... {
+  handledAssertion_ = YES;
+}
+
+- (void)handleFailureInFunction:(NSString *)functionName 
+                           file:(NSString *)fileName 
+                     lineNumber:(NSInteger)line 
+                    description:(NSString *)format,... {
+  handledAssertion_ = YES;
+}
+
+- (BOOL)didHandleAssertion {
+  return handledAssertion_;
+}
+@end
+#endif  // DEBUG