objective-c - 如何模拟类方法（+）？

Question

下面代码需要写单元测试，我想对类方法canMakePayments做mock，返回yes或者no，目前还没有找到好的方法，由于canMakePayments是类方法（+），好像所有的OCMock方法都用于实例方法 (-)。

你们的任何建议或讨论将不胜感激。谢谢。

// SKPaymentQueue.h
// StoreKit
if ([SKPaymentQueue canMakePayments]){
   ....
}
else{
   ...
}

Answer 1

一种方法是将类方法包装在您自己的实例方法中：

-(BOOL)canMakePayments {
    return [SKPaymentQueue canMakePayments];
}

然后你模拟那个方法：

-(void)testCanHandlePaymentsDisabled {
    Foo *foo = [[Foo alloc] init];
    id mockFoo = [OCMockObject partialMockForObject:foo];
    BOOL paymentsEnabled = NO;
    [[[mockFoo stub] andReturnValue:OCMOCK_VALUE(paymentsEnabled)] canMakePayments];

    // set up expectations for payments disabled case
    ...

    [foo attemptPurchase];
}

Answer 2

由于无法通过提供不同的实例来拦截方法，因此您可以为类方法做的就是提供不同的类。像这样的东西：

+ (Class)paymentQueueClass
{
    return [SKPaymentQueue class];
}

然后调用点变为：

Class paymentQueueClass = [[self class] paymentQueueClass];
if ([paymentQueueClass canMakePayments])
...

这引入了“测试接缝”或控制点，允许我们指定除SKPaymentQueue. 现在让我们进行替换：

static BOOL fakeCanMakePayments;

@interface FakePaymentQueue : SKPaymentQueue
@end

@implementation FakePaymentQueue

+ (void)setFakeCanMakePayments:(BOOL)fakeValue
{
    fakeCanMakePayments = fakeValue;
}

+ (BOOL)canMakePayments
{
    return fakeCanMakePayments;
}

@end

严格来说，这不是一个“模拟对象”——它是一个“假对象”。不同之处在于模拟对象验证它是如何被调用的。伪造的对象只是提供存根的结果。

现在让我们创建一个我们想要测试的原始类的测试子类。

@interface TestingSubclass : OriginalClass
@end

@implementation TestingSubclass

+ (Class)paymentQueueClass
{
    return [FakePaymentQueue class];
}

@end

所以你看，这替换SKPaymentQueue为FakePaymentQueue. 您的测试现在可以针对TestingSubclass.