1 files changed, 143 insertions, 0 deletions
diff --git a/Foundation/GTMSignalHandlerTest.m b/Foundation/GTMSignalHandlerTest.m
new file mode 100644
index 0000000..9dcabf4
--- /dev/null
+++ b/Foundation/GTMSignalHandlerTest.m
@@ -0,0 +1,143 @@
+//
+//  GTMSignalHandlerTest.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 "GTMSignalHandler.h"
+#import "GTMSenTestCase.h"
+
+@interface GTMSignalHandlerTest : GTMTestCase
+@end  // GTMSignalHandlerTest
+
+@interface SignalCounter : NSObject {
+ @public
+  int signalCount_;
+  int lastSeenSignal_;
+}
+- (int)count;
+- (int)lastSeen;
+- (void)countSignal:(NSNumber *)signo;
++ (id)signalCounter;
+@end // SignalCounter
+
+@implementation SignalCounter
++ (id)signalCounter {
+  return [[[[self class] alloc] init] autorelease];
+}
+- (int)count {
+  return signalCount_;
+}
+- (int)lastSeen {
+  return lastSeenSignal_;
+}
+// Count the number of times this signal handler has fired.
+- (void)countSignal:(NSNumber *)signo {
+  signalCount_++;
+  lastSeenSignal_ = [signo intValue];
+}  // countSignal
+@end  // SignalCounter
+
+@implementation GTMSignalHandlerTest
+
+// Spin the run loop so that the kqueue event notifications will get delivered.
+- (void)giveSomeLove {
+  NSDate *endTime = [NSDate dateWithTimeIntervalSinceNow:0.5];
+  [[NSRunLoop currentRunLoop] runUntilDate:endTime];
+}  // giveSomeLove
+
+
+- (void)testNillage {
+  GTMSignalHandler *handler;
+
+  // Just an init should return nil.
+  handler = [[[GTMSignalHandler alloc] init] autorelease];
+  STAssertNil(handler, nil);
+
+  // Zero signal should return nil as well.
+  handler = [[[GTMSignalHandler alloc] 
+              initWithSignal:0
+                      target:self
+                     handler:@selector(nomnomnom:)] autorelease];
+  STAssertNil(handler, nil);
+
+}  // testNillage
+
+
+- (void)testSingleHandler {
+  SignalCounter *counter = [SignalCounter signalCounter];
+  STAssertNotNil(counter, nil);
+  
+  GTMSignalHandler *handler = [[GTMSignalHandler alloc]
+                                initWithSignal:SIGWINCH
+                                        target:counter
+                                       handler:@selector(countSignal:)];
+  STAssertNotNil(handler, nil);
+  raise(SIGWINCH);
+  [self giveSomeLove];
+
+  STAssertEquals([counter count], 1, nil);
+  STAssertEquals([counter lastSeen], SIGWINCH, nil);
+
+  raise(SIGWINCH);
+  [self giveSomeLove];
+
+  STAssertEquals([counter count], 2, nil);
+  STAssertEquals([counter lastSeen], SIGWINCH, nil);
+
+  // create a second one to make sure we're seding data where we want
+  SignalCounter *counter2 = [SignalCounter signalCounter];
+  STAssertNotNil(counter2, nil);
+  [[[GTMSignalHandler alloc] initWithSignal:SIGUSR1
+                                     target:counter2
+                                    handler:@selector(countSignal:)] autorelease];
+  
+  raise(SIGUSR1);
+  [self giveSomeLove];
+  
+  STAssertEquals([counter count], 2, nil);
+  STAssertEquals([counter lastSeen], SIGWINCH, nil);
+  STAssertEquals([counter2 count], 1, nil);
+  STAssertEquals([counter2 lastSeen], SIGUSR1, nil);
+
+  [handler release];
+
+  // The signal is still ignored (so we shouldn't die), but the
+  // the handler method should not get called.
+  raise(SIGWINCH);
+
+  STAssertEquals([counter count], 2, nil);
+  STAssertEquals([counter lastSeen], SIGWINCH, nil);
+  STAssertEquals([counter2 count], 1, nil);
+  STAssertEquals([counter2 lastSeen], SIGUSR1, nil);
+
+}  // testSingleHandler
+
+
+- (void)testIgnore {
+  SignalCounter *counter = [SignalCounter signalCounter];
+  STAssertNotNil(counter, nil);
+
+  [[[GTMSignalHandler alloc] initWithSignal:SIGUSR1
+                                     target:counter
+                                    handler:NULL] autorelease];
+
+  raise(SIGUSR1);
+  [self giveSomeLove];
+  STAssertEquals([counter count], 0, nil);
+
+}  // testIgnore
+
+@end  // GTMSignalHandlerTest