OCMock期待一個類的方法

Question

所以我想能期待一個類的方法被稱爲我的班

@implementation CustomClass 

+ (void)method:(NSString*)string{ 
    [[self class] method:string object:nil]; 

} 

+ (void)method:(NSString *)string object:(id)object { 
    //do something with string and object 
} 

@end 

之一，我想打電話給[CustomClass method:@""]和期望method: string:

我試着方法攪拌，但它似乎只是用於存根。

Answer 1

你可以測試，既使用方法swizzling或OCMock。

隨着法混寫，首先我們在測試中實現文件中聲明以下變量：

static NSString *passedString; 
static id passedObject; 

然後，我們實現一個存根方法（在測試類）和去與混寫：

+ (void)stub_method:(NSString *)string object:(id)object 
{ 
    passedString = string; 
    passedObject = object; 
} 

- (void) test__with_method_swizzling 
{ 
    // Test preparation 
    passedString = nil; 
    passedObject = [NSNull null];// Whatever object to verify that we pass nil 

    Method originalMethod = 
     class_getClassMethod([CustomClass class], @selector(method:object:)); 
    Method stubMethod = 
     class_getClassMethod([self class], @selector(stub_method:object:)); 

    method_exchangeImplementations(originalMethod, stubMethod); 

    NSString * const kFakeString = @"fake string"; 

    // Method to test 
    [CustomClass method:kFakeString]; 

    // Verifications 
    STAssertEquals(passedString, kFakeString, nil); 
    STAssertNil(passedObject, nil); 

    method_exchangeImplementations(stubMethod, originalMethod); 
} 

但是，我們可以在一個更簡單的方法完成同樣與OCMock：

- (void) test__with_OCMock 
{ 
    // Test preparation 
    id mock = [OCMockObject mockForClass:[CustomClass class]]; 

    NSString * const kFakeString = @"fake string"; 
    [[mock expect] method:kFakeString object:nil]; 

    // Method to test 
    [CustomClass method:kFakeString]; 

    // Verifications 
    [mock verify]; 
}