RestKit/Code/Testing/RKTestNotificationObserver.m

//
//  RKTestNotificationObserver.m
//  RestKit
//
//  Created by Jeff Arena on 8/23/11.
//  Copyright (c) 2009-2012 RestKit. All rights reserved.
//

#import "RKTestNotificationObserver.h"

@interface RKTestNotificationObserver ()
@property (nonatomic, assign, getter = isAwaitingNotification) BOOL awaitingNotification;
@end

@implementation RKTestNotificationObserver

@synthesize object;
@synthesize name;
@synthesize timeout;
@synthesize awaitingNotification;

+ (void)waitForNotificationWithName:(NSString *)name object:(id)object usingBlock:(void(^)())block
{
    RKTestNotificationObserver *observer = [RKTestNotificationObserver notificationObserverForName:name object:object];
    block();
    [observer waitForNotification];
}

+ (void)waitForNotificationWithName:(NSString *)name usingBlock:(void(^)())block
{
    [self waitForNotificationWithName:name object:nil usingBlock:block];
}

+ (RKTestNotificationObserver *)notificationObserver
{
    return [[[self alloc] init] autorelease];
}

+ (RKTestNotificationObserver *)notificationObserverForName:(NSString *)notificationName object:(id)object
{
    RKTestNotificationObserver *notificationObserver = [self notificationObserver];
    notificationObserver.object = object;
    notificationObserver.name = notificationName;
    return notificationObserver;
}

+ (RKTestNotificationObserver *)notificationObserverForName:(NSString *)notificationName
{
    return [self notificationObserverForName:notificationName object:nil];
}

- (id)init
{
    self = [super init];
    if (self) {
        timeout = 5;
        awaitingNotification = NO;
    }
    return self;
}

- (void)dealloc
{
    [[NSNotificationCenter defaultCenter] removeObserver:self];
    [super dealloc];
}

- (void)waitForNotification
{
    NSAssert(name, @"Notification name cannot be nil");

    [[NSNotificationCenter defaultCenter] addObserver:self
                                             selector:@selector(processNotification:)
                                                 name:self.name
                                               object:self.object];

    awaitingNotification = YES;
    NSDate *startDate = [NSDate date];

    while (self.isAwaitingNotification) {
        [[NSRunLoop currentRunLoop] runUntilDate:[NSDate dateWithTimeIntervalSinceNow:0.1]];
        if ([[NSDate date] timeIntervalSinceDate:startDate] > self.timeout) {
            [NSException raise:nil format:@"*** Operation timed out after %f seconds...", self.timeout];
            awaitingNotification = NO;
        }
    }
}

- (void)processNotification:(NSNotification *)notification
{
    NSAssert([name isEqualToString:notification.name],
             @"Received notification (%@) differs from expected notification (%@)",
             notification.name, name);
    awaitingNotification = NO;
}

@end