libs-base/Tests/base/NSMapTable/weakObjects.m
2024-07-17 11:32:52 +01:00

43 lines
1.2 KiB
Objective-C

#import "ObjectTesting.h"
#import <Foundation/NSAutoreleasePool.h>
#import <Foundation/NSMapTable.h>
#if __has_include(<objc/capabilities.h>)
#include <objc/capabilities.h>
#endif
int main()
{
START_SET("NSMapTable weak objects")
#if !__APPLE__ // We assume that apple systems support zeroing weak pointers
#ifdef OBJC_CAP_ARC
if (!objc_test_capability(OBJC_CAP_ARC))
#endif
{
SKIP("ARC support unavailable")
}
#endif
NSAutoreleasePool *arp = [NSAutoreleasePool new];
NSMapTable *mapTable = [NSMapTable strongToWeakObjectsMapTable];
NSAutoreleasePool *arp2 = [NSAutoreleasePool new];
id testObj1 = [[[NSObject alloc] init] autorelease];
id testObj2 = [[[NSObject alloc] init] autorelease];
[mapTable setObject: testObj1 forKey: @"test"];
PASS([mapTable objectForKey: @"test"] == testObj1,
"Table retains first active weak reference");
[mapTable setObject: testObj2 forKey: @"test"];
PASS([mapTable objectForKey: @"test"] == testObj2,
"Table retains second active weak reference");
[arp2 release]; arp2 = nil;
PASS([mapTable objectForKey: @"test"] == nil,
"Table removes dead weak reference");
[arp release]; arp = nil;
END_SET("NSMapTable weak objects")
return 0;
}