KWAsyncVerifier.m

//
//  KWAsyncVerifier.m
//  iOSFalconCore
//
//  Created by Luke Redpath on 13/01/2011.
//  Copyright 2011 LJR Software Limited. All rights reserved.
//

#import "KWAsyncVerifier.h"
#import "KWFailure.h"
#import "KWMatching.h"
#import "KWReporting.h"
#import "KWProbePoller.h"

@implementation KWAsyncVerifier

@synthesize timeout;

+ (id)asyncVerifierWithExpectationType:(KWExpectationType)anExpectationType callSite:(KWCallSite *)aCallSite matcherFactory:(KWMatcherFactory *)aMatcherFactory reporter:(id<KWReporting>)aReporter probeTimeout:(NSTimeInterval)probeTimeout;
{
  KWAsyncVerifier *verifier = [[self alloc] initWithExpectationType:anExpectationType callSite:aCallSite matcherFactory:aMatcherFactory reporter:aReporter];
  verifier.timeout = probeTimeout;
  return [verifier autorelease];
}

- (id)initWithExpectationType:(KWExpectationType)anExpectationType callSite:(KWCallSite *)aCallSite matcherFactory:(KWMatcherFactory *)aMatcherFactory reporter:(id<KWReporting>)aReporter {
  if ((self = [super initWithExpectationType:anExpectationType callSite:aCallSite matcherFactory:aMatcherFactory reporter:aReporter])) {
    self.timeout = kKW_DEFAULT_PROBE_TIMEOUT;
  }
  return self;
}

- (void)verifyWithProbe:(KWAsyncMatcherProbe *)aProbe {
  @try {
    KWProbePoller *poller = [[KWProbePoller alloc] initWithTimeout:self.timeout delay:kKW_DEFAULT_PROBE_DELAY];

    if (![poller check:aProbe]) {
      if (self.expectationType == KWExpectationTypeShould) {
        NSString *message = [aProbe.matcher failureMessageForShould];
        KWFailure *failure = [KWFailure failureWithCallSite:self.callSite message:message];
        [self.reporter reportFailure:failure];
      } else if (self.expectationType == KWExpectationTypeShouldNot) {
        NSString *message = [aProbe.matcher failureMessageForShouldNot];
        KWFailure *failure = [KWFailure failureWithCallSite:self.callSite message:message];
        [self.reporter reportFailure:failure];
      } else if (self.expectationType == KWExpectationTypeMaybe) {
        // don't do anything
      }
    }
    [poller release];

  } @catch (NSException *exception) {
    KWFailure *failure = [KWFailure failureWithCallSite:self.callSite message:[exception description]];
    [self.reporter reportFailure:failure];
  }
}

- (void)verifyWithMatcher:(id<KWMatching>)aMatcher {
  KWAsyncMatcherProbe *probe = [[[KWAsyncMatcherProbe alloc] initWithMatcher:aMatcher] autorelease];
  [self verifyWithProbe:probe];
}

@end

@implementation KWAsyncMatcherProbe

@synthesize matcher;

- (id)initWithMatcher:(id<KWMatching>)aMatcher;
{
  if ((self = [super init])) {
    matcher = [aMatcher retain];

    // make sure the matcher knows we are going to evaluate it multiple times
    if ([aMatcher respondsToSelector:@selector(willEvaluateMultipleTimes)]) {
      [aMatcher setWillEvaluateMultipleTimes:YES];
    }
  }
  return self;
}

- (void)dealloc
{
  [matcher release];
  [super dealloc];
}

- (BOOL)isSatisfied;
{
  return matchResult;
}

- (void)sample;
{
  matchResult = [matcher evaluate];
}

@end