//--------------------------------------------------------------------------------------- // $Id$ // Copyright (c) 2007-2010 by Mulle Kybernetik. See License file for details. //--------------------------------------------------------------------------------------- #import <OCMock/OCMConstraint.h> @implementation OCMConstraint + (id)constraint { return [[[self alloc] init] autorelease]; } - (BOOL)evaluate:(id)value { return NO; } + (id)constraintWithSelector:(SEL)aSelector onObject:(id)anObject { OCMInvocationConstraint *constraint = [OCMInvocationConstraint constraint]; NSMethodSignature *signature = [anObject methodSignatureForSelector:aSelector]; if(signature == nil) [NSException raise:NSInvalidArgumentException format:@"Unkown selector %@ used in constraint.", NSStringFromSelector(aSelector)]; NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:signature]; [invocation setTarget:anObject]; [invocation setSelector:aSelector]; constraint->invocation = invocation; return constraint; } + (id)constraintWithSelector:(SEL)aSelector onObject:(id)anObject withValue:(id)aValue { OCMInvocationConstraint *constraint = [self constraintWithSelector:aSelector onObject:anObject]; if([[constraint->invocation methodSignature] numberOfArguments] < 4) [NSException raise:NSInvalidArgumentException format:@"Constraint with value requires selector with two arguments."]; [constraint->invocation setArgument:&aValue atIndex:3]; return constraint; } @end #pragma mark - @implementation OCMAnyConstraint - (BOOL)evaluate:(id)value { return YES; } @end #pragma mark - @implementation OCMIsNilConstraint - (BOOL)evaluate:(id)value { return value == nil; } @end #pragma mark - @implementation OCMIsNotNilConstraint - (BOOL)evaluate:(id)value { return value != nil; } @end #pragma mark - @implementation OCMIsNotEqualConstraint - (BOOL)evaluate:(id)value { return ![value isEqual:testValue]; } @end #pragma mark - @implementation OCMInvocationConstraint - (BOOL)evaluate:(id)value { [invocation setArgument:&value atIndex:2]; // should test if constraint takes arg [invocation invoke]; BOOL returnValue; [invocation getReturnValue:&returnValue]; return returnValue; } @end #pragma mark - #if NS_BLOCKS_AVAILABLE @implementation OCMBlockConstraint - (id)initWithConstraintBlock:(BOOL (^)(id))aBlock; { self = [super init]; block = aBlock; return self; } - (BOOL)evaluate:(id)value { return block(value); } @end #endif