root/open-source/SimpleRegexSupport/trunk/RegexTest.m

//
//  Created by Martin HÀcker on 09.10.08.
//  This Code is in the public domain.
//

#import "RegexTest.h"

/// For the regex syntax see <http://www.icu-project.org/userguide/regexp.html>
@interface NSString (SimpleRegexMatching)
- (BOOL) matchesRegex:(NSString *)aRegex;
- (BOOL) containsRegex:(NSString *)aRegex;
@end

@implementation NSString (SimpleRegexMatching)
- (BOOL) matchesRegex:(NSString *)aRegex;
{
    id tigerFix = [NSString stringWithFormat:@"(?m)%@", aRegex];
    id predicate = [NSPredicate predicateWithFormat:@"self matches %@", tigerFix]; //aRegex];
    return [predicate evaluateWithObject:self];
}
- (BOOL) containsRegex:(NSString *)aRegex;
{
    return [self matchesRegex:[NSString stringWithFormat:@".*%@.*", aRegex]];
}
@end

@implementation RegexTest

- (void) testSmoke
{
    STAssertNotNil(@"", nil);
}

- (void) testHasMatchingMethod
{
    STAssertTrue([@"" respondsToSelector:@selector(matchesRegex:)], nil);
}

- (void) testEmptyStringNeverMatches
{
    STAssertFalse([@"" matchesRegex:nil], nil);
    STAssertFalse([@"nil" matchesRegex:nil], nil);
    STAssertFalse([@"(nil)" matchesRegex:nil], nil);

    // TODO: Sure this is an edge case but still, I had different results when I ran this in non tiger compatibility mode. Find out why.
    STAssertTrue([@"" matchesRegex:@""], nil);
}

- (void) testSimpleMatches
{
    STAssertTrue([@"nil" matchesRegex:@"nil"], nil);
    STAssertTrue([@" " matchesRegex:@" "], nil);
}

- (void) testDotStarMatchesEverything
{
    STAssertTrue([@"" matchesRegex:@".*"], nil);
    STAssertTrue([@"fnord" matchesRegex:@".*"], nil);
    STAssertTrue([@"fnord\nfnord" matchesRegex:@".*"], nil);
}

- (void) testMoreComplexMatches
{
    STAssertTrue([@"aaa" matchesRegex:@"a{3}"], nil);
    STAssertFalse([@"aa" matchesRegex:@"a{3}"], nil);
    STAssertFalse([@"aaaa" matchesRegex:@"a{3}"], nil);

    STAssertTrue([@"argh fnord argh" matchesRegex:@".*\\bfnord\\b.*"], nil);
    STAssertFalse([@"argh fnorda argh" matchesRegex:@".*\\bfnord\\b.*"], nil);
    STAssertFalse([@"argh afnord argh" matchesRegex:@".*\\bfnord\\b.*"], nil);
    
    STAssertTrue([@"aaa" matchesRegex:@"\\w{3}"], nil);
    STAssertFalse([@"aa" matchesRegex:@"\\w{3}"], nil);
}

- (void) testCanFindContainedRegexes
{
    STAssertTrue([@"aaa" containsRegex:@"a"], nil);
    STAssertTrue([@"aaa" containsRegex:@"aa"], nil);
    STAssertTrue([@"aaa" containsRegex:@"aaa"], nil);
    STAssertTrue([@"aaa" containsRegex:@"^a"], nil);
    STAssertTrue([@"aaa" containsRegex:@"a$"], nil);
    STAssertFalse([@"aaa" containsRegex:@"b"], nil);
    STAssertFalse([@"aaa" containsRegex:@"aaaa"], nil);

    STAssertTrue([@"aaab" containsRegex:@"^a"], nil);
}

@end