如何让 ARC 下的 OCMock 停止使用弱属性清空 NSProxy 子类集？

Question

在 ARC 下，我有一个对象 Child，它有一个 weak 属性 parent。我正在尝试为 Child 编写一些测试，并使用 OCMock 模拟其 parent 属性。

在 ARC 下，使用合成的弱属性设置器设置 NSProxy 子类不会设置该属性...设置弱属性后的行，检查它发现它已经nil.这是具体的例子：

@interface Child : NSObject
@property (nonatomic, weak) id <ParentInterface>parent;
@end

@implementation Child
@synthesize parent = parent_;
@end

//  ... later, inside a test class ...

- (void)testParentExists
{
    // `mockForProtocol` returns an `NSProxy` subclass
    //
    OCMockObject *aParent = [OCMockObject mockForProtocol:@protocol(ParentInterface)];
    assertThat(aParent, notNilValue());

    // `Child` is the class under test
    //
    Child *child = [[Child alloc] init];
    assertThat(child, notNilValue());

    assertThat(child.parent, nilValue());
    child.parent = (id<ParentInterface>)aParent;
    assertThat([child parent], notNilValue());  // <-- This assertion fails
    [aParent self]; // <-- Added this reference just to ensure `aParent` was valid until the end of the test.
}

我知道我可以使用 assign 属性而不是 weak 属性来解决这个问题，让 Child 引用 < code>Parent，但是当我完成它时，我必须nil出parent（就像某种穴居人），这正是ARC 应该避免这种情况。

关于如何在不更改我的应用程序代码的情况下通过此测试有什么建议吗？

编辑：如果我使 aParent 成为一个实例，这似乎与 OCMockObject 是一个 NSProxy 有关NSObject，child.parent 弱引用“保存”一个非零值。仍在寻找一种无需更改应用程序代码即可通过此测试的方法。

编辑2：接受布莱克的回答后，我在我的项目中实现了一个预处理器宏，该宏有条件地将我的属性从weak -> 更改为weak -> 。分配。您的里程可能会有所不同：

#if __has_feature(objc_arc)
#define BBE_WEAK_PROPERTY(type, name) @property (weak, nonatomic) type name
#else
#define BBE_WEAK_PROPERTY(type, name) @property (assign, nonatomic) type name
#endif

Answer 1

我们一直在努力解决同样的问题，它确实与 ARC 和对 NSProxy 派生对象的弱引用之间的不兼容有关。我建议使用预处理器指令有条件地编译弱委托引用以在测试套件中分配，以便您可以通过 OCMock 测试它们。

Answer 2

我找到了与条件宏不同的解决方案，因为我正在测试无法更改代码的代码。

我编写了一个扩展 NSObject（而不是 NSProxy）的简单类，它将所有选择器调用转发到 OCMockProxy。

CCWeakMockProxy.h：

#import <Foundation/Foundation.h>

/**
 * This class is a hack around the fact that ARC weak references are immediately nil'd if the referent is an NSProxy
 * See: http://stackoverflow.com/questions/9104544/how-can-i-get-ocmock-under-arc-to-stop-nilling-an-nsproxy-subclass-set-using-a-w
 */
@interface CCWeakMockProxy : NSObject

@property (strong, nonatomic) id mock;

- (id)initWithMock:(id)mockObj;

+ (id)mockForClass:(Class)aClass;
+ (id)mockForProtocol:(Protocol *)aProtocol;
+ (id)niceMockForClass:(Class)aClass;
+ (id)niceMockForProtocol:(Protocol *)aProtocol;
+ (id)observerMock;
+ (id)partialMockForObject:(NSObject *)anObject;

@end

CCWeakMockProxy.m：

#import "CCWeakMockProxy.h"
#import <OCMock/OCMock.h>


#pragma mark Implementation
@implementation CCWeakMockProxy

#pragma mark Properties
@synthesize mock;

#pragma mark Memory Management
- (id)initWithMock:(id)mockObj {
    if (self = [super init]) {
        self.mock = mockObj;
    }
    return self;
}

#pragma mark NSObject
- (id)forwardingTargetForSelector:(SEL)aSelector {
    return self.mock;
}

- (BOOL)respondsToSelector:(SEL)aSelector {
    return [self.mock respondsToSelector:aSelector];
}

#pragma mark Public Methods
+ (id)mockForClass:(Class)aClass {
    return [[CCWeakMockProxy alloc] initWithMock:[OCMockObject mockForClass:aClass]];
}

+ (id)mockForProtocol:(Protocol *)aProtocol {
    return [[CCWeakMockProxy alloc] initWithMock:[OCMockObject mockForProtocol:aProtocol]];
}

+ (id)niceMockForClass:(Class)aClass {
    return [[CCWeakMockProxy alloc] initWithMock:[OCMockObject niceMockForClass:aClass]];
}

+ (id)niceMockForProtocol:(Protocol *)aProtocol {
    return [[CCWeakMockProxy alloc] initWithMock:[OCMockObject niceMockForProtocol:aProtocol]];
}

+ (id)observerMock {
    return [[CCWeakMockProxy alloc] initWithMock:[OCMockObject observerMock]];
}

+ (id)partialMockForObject:(NSObject *)anObject {
    return [[CCWeakMockProxy alloc] initWithMock:[OCMockObject partialMockForObject:anObject]];
}

@end

只需像使用常规 OCMockObject 一样使用生成的对象即可！

Answer 3

当然。它会变成 nil，因为在分配 child.parent 之后，你的代理对象本身就会被你的测试释放（因为它不再被引用），这会导致对 nil 的弱引用出去。因此，解决方案是在测试期间保持代理对象处于活动状态。 插入对 的调用来简单地完成此操作

[aParent self];

您可以通过在方法末尾 。该函数调用不执行任何操作（-self 仅返回 self），但它将确保 ARC 保持对象处于活动状态。

另一种方法是将 aParent 声明更改为 __autoreleasing，这使得它的行为更像 MRR，因为 ARC 只会在该插槽中留下自动释放的引用，而不是显式地保留当变量超出范围时释放对象。您可以这样做

__autoreleasing OCMockObject *aParent = ...

，也就是说，第一个解决方案可能更干净，因为您在测试期间明确地保持对象处于活动状态。