OCMock模拟协议的静态类方法。

Question

OCMock 3中的新功能可以剔除class methods。

是否可以模拟协议中定义的类方法？即

@protocol AViewControllerProtocol <NSObject> 
+ (Type)typeForViewController; 
@end 

在我的单元测试类

- (void)testProtocolClassMethod { 
    id mockedViewController = OCMProtocolMock(@protocol(AViewControllerProtocol)); 

    //This line compiles fine, but throws an exception at run time. 
    OCMStub([mockedViewController typeForViewController]).andReturn(SomeType); 
} 

异常抛出

NSInvalidArgumentException：不能存根/期望/验证方法 'typeForViewController'，因为没有这样的方法在嘲笑

类存在

Answer 1

这看起来像是一个oversig在OCMock 3.1中，但如果你愿意，你可以自己修复。

// OCProtocolMockObject.m 
- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector 
{ 
    struct objc_method_description methodDescription = protocol_getMethodDescription(mockedProtocol, aSelector, YES, YES); 
    if(methodDescription.name == NULL) 
    { 
     methodDescription = protocol_getMethodDescription(mockedProtocol, aSelector, NO, YES); 
    } 
    // Add this case for required class methods 
    if (methodDescription.name == NULL) 
    { 
     methodDescription = protocol_getMethodDescription(mockedProtocol, aSelector, YES, NO); 
    } 
    // Add this case for optional class methods 
    if (methodDescription.name == NULL) 
    { 
     methodDescription = protocol_getMethodDescription(mockedProtocol, aSelector, NO, NO); 
    } 
    if(methodDescription.name == NULL) 
    { 
     return nil; 
    } 
    return [NSMethodSignature signatureWithObjCTypes:methodDescription.types]; 
} 

我验证了这个解决与此测试：

- (void)testProtocolClassMethod { 
    id mockedViewController = OCMProtocolMock(@protocol(AViewControllerProtocol)); 

    // FIXED: This line compiles fine, but throws an exception at run time. 
    OCMStub([mockedViewController typeForViewController]).andReturn(SomeType); 

    Type type = [mockedViewController typeForViewController]; 

    XCTAssertEqual(type, SomeType, @"Not equal!"); 

    OCMVerify([mockedViewController typeForViewController]); 
} 

我会把请求在项目页面上此。