Skip to content
This repository has been archived by the owner on Mar 29, 2018. It is now read-only.

Commit

Permalink
A central repository for storing user-defined matchers.
Browse files Browse the repository at this point in the history
  • Loading branch information
Luke Redpath committed Jun 17, 2011
1 parent 7d01f4e commit 0b80c7b
Show file tree
Hide file tree
Showing 3 changed files with 71 additions and 7 deletions.
22 changes: 21 additions & 1 deletion Kiwi/KWMatchers.h
Expand Up @@ -8,6 +8,26 @@

#import <Foundation/Foundation.h>

@interface KWMatchers : NSObject
@class KWUserDefinedMatcherBuilder;

typedef void (^KWMatchersBuildingBlock)(KWUserDefinedMatcherBuilder *);

@class KWUserDefinedMatcher;

@interface KWMatchers : NSObject {
NSMutableDictionary *userDefinedMatchers;
}
+ (id)matchers;

#pragma mark -
#pragma mark Defining Matchers

+ (void)defineMatcher:(NSString *)selectorString as:(KWMatchersBuildingBlock)block;
- (void)defineMatcher:(NSString *)selectorString as:(KWMatchersBuildingBlock)block;
- (void)addUserDefinedMatcherBuilder:(KWUserDefinedMatcherBuilder *)builder;

#pragma mark -
#pragma mark Building Matchers

- (KWUserDefinedMatcher *)matcherForSelector:(SEL)selector subject:(id)subject;
@end
55 changes: 49 additions & 6 deletions Kiwi/KWMatchers.m
Expand Up @@ -7,17 +7,60 @@
//

#import "KWMatchers.h"
#import "KWUserDefinedMatcher.h"

@implementation KWMatchers

- (id)init
{
self = [super init];
if (self) {
// Initialization code here.
#pragma mark -
#pragma mark Singleton implementation

static id sharedMatchers = nil;

+ (void)initialize {
if (self == [KWMatchers class]) {
sharedMatchers = [[self alloc] init];
}
}

+ (id)matchers {
return sharedMatchers;
}

- (id)init {
if ((self = [super init])) {
userDefinedMatchers = [[NSMutableDictionary alloc] init];
}

return self;
}

#pragma mark -
#pragma mark Defining Matchers

+ (void)defineMatcher:(NSString *)selectorString as:(KWMatchersBuildingBlock)block {
[[self matchers] defineMatcher:selectorString as:block];
}

- (void)defineMatcher:(NSString *)selectorString as:(KWMatchersBuildingBlock)block {
KWUserDefinedMatcherBuilder *builder = [KWUserDefinedMatcherBuilder builderForSelector:NSSelectorFromString(selectorString)];
block(builder);
[userDefinedMatchers setObject:builder forKey:builder.key];
}

- (void)addUserDefinedMatcherBuilder:(KWUserDefinedMatcherBuilder *)builder {
[userDefinedMatchers setObject:builder forKey:builder.key];
}

#pragma mark -
#pragma mark Building Matchers

- (KWUserDefinedMatcher *)matcherForSelector:(SEL)selector subject:(id)subject {
KWUserDefinedMatcherBuilder *builder = [userDefinedMatchers objectForKey:NSStringFromSelector(selector)];

if (builder == nil)
return nil;

return [builder buildMatcherWithSubject:subject];
}


@end
1 change: 1 addition & 0 deletions Kiwi/Kiwi.h
Expand Up @@ -65,6 +65,7 @@
#import "KWStringUtilities.h"
#import "KWStub.h"
#import "KWTestCase.h"
#import "KWUserDefinedMatcher.h"
#import "KWValue.h"
#import "KWVerifying.h"

Expand Down

0 comments on commit 0b80c7b

Please sign in to comment.