[Author: aharper]

Add a log filter that filters by simple level comparisons.

R=dmaclach,thomasvl
APPROVED=dmaclach

1 files changed, 66 insertions, 0 deletions

diff --git a/Foundation/GTMLoggerTest.m b/Foundation/GTMLoggerTest.m
index 0dc7689..2d783ca 100644
--- a/Foundation/GTMLoggerTest.m
+++ b/Foundation/GTMLoggerTest.m
@@ -474,6 +474,72 @@
   STAssertTrue([filter filterAllowsMessage:nil level:kGTMLoggerLevelDebug], nil);
 }
 
+- (void)testCustomFilter {
+  // Default level is kGTMLoggerLevelError
+  id<GTMLogFilter> filter = [[[GTMLogCustomLevelFilter alloc] init]
+                                autorelease];
+  STAssertNotNil(filter, nil);
+  STAssertFalse([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelUnknown],
+                nil);
+  STAssertFalse([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelDebug],
+                nil);
+  STAssertFalse([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelInfo],
+                nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelError],
+               nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelAssert],
+               nil);
+
+  // Custom level is possible
+  filter = [[[GTMLogCustomLevelFilter alloc]
+               initWithFilterLevel:kGTMLoggerLevelInfo] autorelease];
+  STAssertNotNil(filter, nil);
+  STAssertFalse([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelUnknown],
+                nil);
+  STAssertFalse([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelDebug],
+                nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelInfo],
+               nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelError],
+               nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelAssert],
+               nil);
+
+  // Custom level is possible
+  filter = [[[GTMLogCustomLevelFilter alloc]
+               initWithFilterLevel:kGTMLoggerLevelInfo] autorelease];
+  STAssertNotNil(filter, nil);
+  STAssertFalse([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelUnknown],
+                nil);
+  STAssertFalse([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelDebug],
+                nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelInfo],
+               nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelError],
+               nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelAssert],
+               nil);
+
+  filter = [[[GTMLogCustomLevelFilter alloc]
+               initWithFilterLevel:kGTMLoggerLevelDebug] autorelease];
+  STAssertNotNil(filter, nil);
+  STAssertFalse([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelUnknown],
+                nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelDebug],
+               nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelInfo],
+               nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelError],
+               nil);
+  STAssertTrue([filter filterAllowsMessage:@"hi" level:kGTMLoggerLevelAssert],
+               nil);
+
+  // Cannot exceed assert level filter
+  filter = [[[GTMLogCustomLevelFilter alloc]
+               initWithFilterLevel:kGTMLoggerLevelAssert + 1] autorelease];
+  STAssertNil(filter, nil);
+}
+
 - (void)testFileHandleCreation {
   NSFileHandle *fh = nil;