ResourceTracking.m [plain text]
/*
* Copyright (c) 2011 Apple Inc. All rights reserved.
*
* @APPLE_APACHE_LICENSE_HEADER_START@
*
* 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.
*
* @APPLE_APACHE_LICENSE_HEADER_END@
*/
//
// ResourceTracking.m
// Copyright (c) 2009-2011 Apple Inc. All rights reserved.
//
#import "BlackBoxTest.h"
static BOOL shouldCollect;
@interface ResourceTracking : BlackBoxTest
{
BOOL _finalized;
BOOL _collected;
BOOL _resourceTrackerQueried;
}
- (void)verifyNoCollection;
- (void)verifyCollection;
@end
@implementation ResourceTracking
static char *tracker_description = "resource tracker unit test";
- (BOOL)shouldCollect
{
boolean_t collect = shouldCollect;
_resourceTrackerQueried = YES;
shouldCollect = NO;
return collect;
}
- (void)processOutputLine:(NSString *)line
{
NSString *expectedString = @"triggering collection due to external resource tracker: resource tracker unit test";
NSRange r = [line rangeOfString:expectedString];
if (r.location == NSNotFound) {
[super processOutputLine:line];
}
}
- (void)allocate
{
// force test object out of thread local set
CFRelease(CFRetain([TestFinalizer new]));
}
- (void)performTest
{
auto_zone_register_resource_tracker([self auto_zone], tracker_description, ^{
return (boolean_t)[self shouldCollect];
});
shouldCollect = NO;
_finalized = NO;
_resourceTrackerQueried = NO;
[self allocate];
[NSThread sleepForTimeInterval:0.3]; // wait long enough that the collector will poll the resource tracker
auto_zone_collect_and_notify([self auto_zone], AUTO_ZONE_COLLECT_NO_OPTIONS, _testQueue, ^{ [self verifyNoCollection]; } ); // should not collect
}
- (void)verifyNoCollection
{
if (_finalized) {
[self fail:@"unexpected collection"];
}
if (!_resourceTrackerQueried) {
[self fail:@"resource tracker was not queried (no collection)"];
}
shouldCollect = YES;
_resourceTrackerQueried = NO;
[NSThread sleepForTimeInterval:0.3]; // wait long enough that the collector will poll the resource tracker
auto_zone_collect_and_notify([self auto_zone], AUTO_ZONE_COLLECT_NO_OPTIONS, _testQueue, ^{ [self verifyCollection]; } ); // should collect
}
- (void)verifyCollection
{
if (!_resourceTrackerQueried) {
[self fail:@"resource tracker was not queried (collection)"];
}
if (_finalized) {
[self passed];
} else {
[self fail:@"did not collect"];
}
auto_zone_unregister_resource_tracker([self auto_zone], tracker_description);
[self testFinished];
}
- (void)didFinalize:(TestFinalizer *)finalizer
{
_finalized = YES;
}
@end