Files
swift-mirror/unittests/runtime/Refcounting.mm
Mike Ash 46309d9794 [Runtime] Rename swift_unknown* functions to swift_unknownObject*.
These functions don't accept local variable heap memory, although the names make it sound like they work on anything. When you try, they mistakenly identify such things as ObjC objects, call through to the equivalent objc_* function, and crash confusingly. This adds Object to the name of each one to make it more clear what they accept.

rdar://problem/37285743
2018-08-15 17:48:23 -04:00

45 lines
1.4 KiB
Plaintext

//===--- Refcounting.mm - Reference-counting for ObjC ---------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2017 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#include <Foundation/NSObject.h>
#include <objc/runtime.h>
#include "swift/Runtime/HeapObject.h"
#include "swift/Runtime/Metadata.h"
#include "gtest/gtest.h"
using namespace swift;
static unsigned DestroyedObjCCount = 0;
/// A trivial class that increments DestroyedObjCCount when deallocated.
@interface ObjCTestClass : NSObject @end
@implementation ObjCTestClass
- (void) dealloc {
DestroyedObjCCount++;
[super dealloc];
}
@end
static HeapObject *make_objc_object() {
return static_cast<HeapObject *>([ObjCTestClass new]);
}
TEST(RefcountingTest, objc_unknown_retain_release_n) {
auto object = make_objc_object();
swift_unknownObjectRetain_n(object, 32);
swift_unknownObjectRetain(object);
swift_unknownObjectRelease_n(object, 31);
swift_unknownObjectRelease(object);
swift_unknownObjectRelease_n(object, 1);
swift_unknownObjectRelease(object);
// The object should be destroyed by now.
EXPECT_EQ(1u, DestroyedObjCCount);
}