367 lines
11 KiB
Objective-C
367 lines
11 KiB
Objective-C
//
|
|
// GTMSenTestCase.m
|
|
//
|
|
// Copyright 2007-2008 Google Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
|
// use this file except in compliance with the License. You may obtain a copy
|
|
// of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
// License for the specific language governing permissions and limitations under
|
|
// the License.
|
|
//
|
|
|
|
#import "GTMSenTestCase.h"
|
|
#import <unistd.h>
|
|
|
|
#if !GTM_IPHONE_SDK
|
|
#import "GTMGarbageCollection.h"
|
|
#endif // !GTM_IPHONE_SDK
|
|
|
|
#if GTM_IPHONE_SDK
|
|
#import <stdarg.h>
|
|
|
|
@interface NSException (GTMSenTestPrivateAdditions)
|
|
+ (NSException *)failureInFile:(NSString *)filename
|
|
atLine:(int)lineNumber
|
|
reason:(NSString *)reason;
|
|
@end
|
|
|
|
@implementation NSException (GTMSenTestPrivateAdditions)
|
|
+ (NSException *)failureInFile:(NSString *)filename
|
|
atLine:(int)lineNumber
|
|
reason:(NSString *)reason {
|
|
NSDictionary *userInfo =
|
|
[NSDictionary dictionaryWithObjectsAndKeys:
|
|
[NSNumber numberWithInteger:lineNumber], SenTestLineNumberKey,
|
|
filename, SenTestFilenameKey,
|
|
nil];
|
|
|
|
return [self exceptionWithName:SenTestFailureException
|
|
reason:reason
|
|
userInfo:userInfo];
|
|
}
|
|
@end
|
|
|
|
@implementation NSException (GTMSenTestAdditions)
|
|
|
|
+ (NSException *)failureInFile:(NSString *)filename
|
|
atLine:(int)lineNumber
|
|
withDescription:(NSString *)formatString, ... {
|
|
|
|
NSString *testDescription = @"";
|
|
if (formatString) {
|
|
va_list vl;
|
|
va_start(vl, formatString);
|
|
testDescription =
|
|
[[[NSString alloc] initWithFormat:formatString arguments:vl] autorelease];
|
|
va_end(vl);
|
|
}
|
|
|
|
NSString *reason = testDescription;
|
|
|
|
return [self failureInFile:filename atLine:lineNumber reason:reason];
|
|
}
|
|
|
|
+ (NSException *)failureInCondition:(NSString *)condition
|
|
isTrue:(BOOL)isTrue
|
|
inFile:(NSString *)filename
|
|
atLine:(int)lineNumber
|
|
withDescription:(NSString *)formatString, ... {
|
|
|
|
NSString *testDescription = @"";
|
|
if (formatString) {
|
|
va_list vl;
|
|
va_start(vl, formatString);
|
|
testDescription =
|
|
[[[NSString alloc] initWithFormat:formatString arguments:vl] autorelease];
|
|
va_end(vl);
|
|
}
|
|
|
|
NSString *reason = [NSString stringWithFormat:@"'%@' should be %s. %@",
|
|
condition, isTrue ? "TRUE" : "FALSE", testDescription];
|
|
|
|
return [self failureInFile:filename atLine:lineNumber reason:reason];
|
|
}
|
|
|
|
+ (NSException *)failureInEqualityBetweenObject:(id)left
|
|
andObject:(id)right
|
|
inFile:(NSString *)filename
|
|
atLine:(int)lineNumber
|
|
withDescription:(NSString *)formatString, ... {
|
|
|
|
NSString *testDescription = @"";
|
|
if (formatString) {
|
|
va_list vl;
|
|
va_start(vl, formatString);
|
|
testDescription =
|
|
[[[NSString alloc] initWithFormat:formatString arguments:vl] autorelease];
|
|
va_end(vl);
|
|
}
|
|
|
|
NSString *reason =
|
|
[NSString stringWithFormat:@"'%@' should be equal to '%@'. %@",
|
|
[left description], [right description], testDescription];
|
|
|
|
return [self failureInFile:filename atLine:lineNumber reason:reason];
|
|
}
|
|
|
|
+ (NSException *)failureInEqualityBetweenValue:(NSValue *)left
|
|
andValue:(NSValue *)right
|
|
withAccuracy:(NSValue *)accuracy
|
|
inFile:(NSString *)filename
|
|
atLine:(int)lineNumber
|
|
withDescription:(NSString *)formatString, ... {
|
|
|
|
NSString *testDescription = @"";
|
|
if (formatString) {
|
|
va_list vl;
|
|
va_start(vl, formatString);
|
|
testDescription =
|
|
[[[NSString alloc] initWithFormat:formatString arguments:vl] autorelease];
|
|
va_end(vl);
|
|
}
|
|
|
|
NSString *reason;
|
|
if (accuracy) {
|
|
reason =
|
|
[NSString stringWithFormat:@"'%@' should be equal to '%@'. %@",
|
|
left, right, testDescription];
|
|
} else {
|
|
reason =
|
|
[NSString stringWithFormat:@"'%@' should be equal to '%@' +/-'%@'. %@",
|
|
left, right, accuracy, testDescription];
|
|
}
|
|
|
|
return [self failureInFile:filename atLine:lineNumber reason:reason];
|
|
}
|
|
|
|
+ (NSException *)failureInRaise:(NSString *)expression
|
|
inFile:(NSString *)filename
|
|
atLine:(int)lineNumber
|
|
withDescription:(NSString *)formatString, ... {
|
|
|
|
NSString *testDescription = @"";
|
|
if (formatString) {
|
|
va_list vl;
|
|
va_start(vl, formatString);
|
|
testDescription =
|
|
[[[NSString alloc] initWithFormat:formatString arguments:vl] autorelease];
|
|
va_end(vl);
|
|
}
|
|
|
|
NSString *reason = [NSString stringWithFormat:@"'%@' should raise. %@",
|
|
expression, testDescription];
|
|
|
|
return [self failureInFile:filename atLine:lineNumber reason:reason];
|
|
}
|
|
|
|
+ (NSException *)failureInRaise:(NSString *)expression
|
|
exception:(NSException *)exception
|
|
inFile:(NSString *)filename
|
|
atLine:(int)lineNumber
|
|
withDescription:(NSString *)formatString, ... {
|
|
|
|
NSString *testDescription = @"";
|
|
if (formatString) {
|
|
va_list vl;
|
|
va_start(vl, formatString);
|
|
testDescription =
|
|
[[[NSString alloc] initWithFormat:formatString arguments:vl] autorelease];
|
|
va_end(vl);
|
|
}
|
|
|
|
NSString *reason;
|
|
if ([[exception name] isEqualToString:SenTestFailureException]) {
|
|
// it's our exception, assume it has the right description on it.
|
|
reason = [exception reason];
|
|
} else {
|
|
// not one of our exception, use the exceptions reason and our description
|
|
reason = [NSString stringWithFormat:@"'%@' raised '%@'. %@",
|
|
expression, [exception reason], testDescription];
|
|
}
|
|
|
|
return [self failureInFile:filename atLine:lineNumber reason:reason];
|
|
}
|
|
|
|
@end
|
|
|
|
NSString *STComposeString(NSString *formatString, ...) {
|
|
NSString *reason = @"";
|
|
if (formatString) {
|
|
va_list vl;
|
|
va_start(vl, formatString);
|
|
reason =
|
|
[[[NSString alloc] initWithFormat:formatString arguments:vl] autorelease];
|
|
va_end(vl);
|
|
}
|
|
return reason;
|
|
}
|
|
|
|
NSString *const SenTestFailureException = @"SenTestFailureException";
|
|
NSString *const SenTestFilenameKey = @"SenTestFilenameKey";
|
|
NSString *const SenTestLineNumberKey = @"SenTestLineNumberKey";
|
|
|
|
@interface SenTestCase (SenTestCasePrivate)
|
|
// our method of logging errors
|
|
+ (void)printException:(NSException *)exception fromTestName:(NSString *)name;
|
|
@end
|
|
|
|
@implementation SenTestCase
|
|
- (void)failWithException:(NSException*)exception {
|
|
[exception raise];
|
|
}
|
|
|
|
- (void)setUp {
|
|
}
|
|
|
|
- (void)performTest:(SEL)sel {
|
|
currentSelector_ = sel;
|
|
@try {
|
|
[self invokeTest];
|
|
} @catch (NSException *exception) {
|
|
[[self class] printException:exception
|
|
fromTestName:NSStringFromSelector(sel)];
|
|
[exception raise];
|
|
}
|
|
}
|
|
|
|
+ (void)printException:(NSException *)exception fromTestName:(NSString *)name {
|
|
NSDictionary *userInfo = [exception userInfo];
|
|
NSString *filename = [userInfo objectForKey:SenTestFilenameKey];
|
|
NSNumber *lineNumber = [userInfo objectForKey:SenTestLineNumberKey];
|
|
NSString *className = NSStringFromClass([self class]);
|
|
if ([filename length] == 0) {
|
|
filename = @"Unknown.m";
|
|
}
|
|
fprintf(stderr, "%s:%ld: error: -[%s %s] : %s\n",
|
|
[filename UTF8String],
|
|
(long)[lineNumber integerValue],
|
|
[className UTF8String],
|
|
[name UTF8String],
|
|
[[exception reason] UTF8String]);
|
|
fflush(stderr);
|
|
}
|
|
|
|
- (void)invokeTest {
|
|
NSException *e = nil;
|
|
@try {
|
|
// Wrap things in autorelease pools because they may
|
|
// have an STMacro in their dealloc which may get called
|
|
// when the pool is cleaned up
|
|
NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
|
|
// We don't log exceptions here, instead we let the person that called
|
|
// this log the exception. This ensures they are only logged once but the
|
|
// outer layers get the exceptions to report counts, etc.
|
|
@try {
|
|
[self setUp];
|
|
@try {
|
|
[self performSelector:currentSelector_];
|
|
} @catch (NSException *exception) {
|
|
e = [exception retain];
|
|
}
|
|
[self tearDown];
|
|
} @catch (NSException *exception) {
|
|
e = [exception retain];
|
|
}
|
|
[pool release];
|
|
} @catch (NSException *exception) {
|
|
e = [exception retain];
|
|
}
|
|
if (e) {
|
|
[e autorelease];
|
|
[e raise];
|
|
}
|
|
}
|
|
|
|
- (void)tearDown {
|
|
}
|
|
|
|
- (NSString *)description {
|
|
// This matches the description OCUnit would return to you
|
|
return [NSString stringWithFormat:@"-[%@ %@]", [self class],
|
|
NSStringFromSelector(currentSelector_)];
|
|
}
|
|
@end
|
|
|
|
#endif // GTM_IPHONE_SDK
|
|
|
|
@implementation GTMTestCase : SenTestCase
|
|
- (void)invokeTest {
|
|
Class devLogClass = NSClassFromString(@"GTMUnitTestDevLog");
|
|
if (devLogClass) {
|
|
[devLogClass performSelector:@selector(enableTracking)];
|
|
[devLogClass performSelector:@selector(verifyNoMoreLogsExpected)];
|
|
|
|
}
|
|
[super invokeTest];
|
|
if (devLogClass) {
|
|
[devLogClass performSelector:@selector(verifyNoMoreLogsExpected)];
|
|
[devLogClass performSelector:@selector(disableTracking)];
|
|
}
|
|
}
|
|
@end
|
|
|
|
// Leak detection
|
|
#if !GTM_IPHONE_DEVICE
|
|
// Don't want to get leaks on the iPhone Device as the device doesn't
|
|
// have 'leaks'. The simulator does though.
|
|
|
|
// COV_NF_START
|
|
// We don't have leak checking on by default, so this won't be hit.
|
|
static void _GTMRunLeaks(void) {
|
|
// This is an atexit handler. It runs leaks for us to check if we are
|
|
// leaking anything in our tests.
|
|
const char* cExclusionsEnv = getenv("GTM_LEAKS_SYMBOLS_TO_IGNORE");
|
|
NSMutableString *exclusions = [NSMutableString string];
|
|
if (cExclusionsEnv) {
|
|
NSString *exclusionsEnv = [NSString stringWithUTF8String:cExclusionsEnv];
|
|
NSArray *exclusionsArray = [exclusionsEnv componentsSeparatedByString:@","];
|
|
NSString *exclusion;
|
|
NSCharacterSet *wcSet = [NSCharacterSet whitespaceCharacterSet];
|
|
GTM_FOREACH_OBJECT(exclusion, exclusionsArray) {
|
|
exclusion = [exclusion stringByTrimmingCharactersInSet:wcSet];
|
|
[exclusions appendFormat:@"-exclude \"%@\" ", exclusion];
|
|
}
|
|
}
|
|
NSString *string
|
|
= [NSString stringWithFormat:@"/usr/bin/leaks %@%d"
|
|
@"| /usr/bin/sed -e 's/Leak: /Leaks:0: warning: Leak /'",
|
|
exclusions, getpid()];
|
|
int ret = system([string UTF8String]);
|
|
if (ret) {
|
|
fprintf(stderr, "%s:%d: Error: Unable to run leaks. 'system' returned: %d",
|
|
__FILE__, __LINE__, ret);
|
|
fflush(stderr);
|
|
}
|
|
}
|
|
// COV_NF_END
|
|
|
|
static __attribute__((constructor)) void _GTMInstallLeaks(void) {
|
|
BOOL checkLeaks = YES;
|
|
#if !GTM_IPHONE_SDK
|
|
checkLeaks = GTMIsGarbageCollectionEnabled() ? NO : YES;
|
|
#endif // !GTM_IPHONE_SDK
|
|
if (checkLeaks) {
|
|
checkLeaks = getenv("GTM_ENABLE_LEAKS") ? YES : NO;
|
|
if (checkLeaks) {
|
|
// COV_NF_START
|
|
// We don't have leak checking on by default, so this won't be hit.
|
|
fprintf(stderr, "Leak Checking Enabled\n");
|
|
fflush(stderr);
|
|
int ret = atexit(&_GTMRunLeaks);
|
|
_GTMDevAssert(ret == 0,
|
|
@"Unable to install _GTMRunLeaks as an atexit handler (%d)",
|
|
errno);
|
|
// COV_NF_END
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif // !GTM_IPHONE_DEVICE
|