更改可可單元測試的系統日期

Question

我正在測試一個調用-[NSDate date]的方法，並使用系統時間作爲參考來執行某些邏輯。我需要測試一年中某些特定日子的邏輯是否正確，但我不能等到那一天。那麼我可以在單元測試時以編程方式更改系統日期嗎？

Answer 1

您需要使用Mock object，而在Objective C中，只需重寫您需要的方法即可輕鬆實現。例如，你可以使用以下類別：

// ------------- File: NSDate+NSDateMock.h 
@interface NSDate (NSDateMock) 

+(void)setMockDate:(NSString *)mockDate; 
+(NSDate *) mockCurrentDate; 

@end 

// -------------- File: NSDate+NSDateMock.m 

#import "NSDate+NSDateMock.h" 

@implementation NSDate (NSDateMock) 

static NSDate *_mockDate; 

+(NSDate *)mockCurrentDate 
{ 
    return _mockDate; 
} 

+(void)setMockDate:(NSString *)mockDate 
{ 
    _mockDate = [NSDate dateWithString:mockDate]; 
} 

@end 

，另外你需要SwizzleClassMethod

void SwizzleClassMethod(Class c, SEL orig, SEL new) { 

    Method origMethod = class_getClassMethod(c, orig); 
    Method newMethod = class_getClassMethod(c, new); 

    c = object_getClass((id)c); 

    if(class_addMethod(c, orig, method_getImplementation(newMethod), method_getTypeEncoding(newMethod))) 
     class_replaceMethod(c, new, method_getImplementation(origMethod), method_getTypeEncoding(origMethod)); 
    else 
     method_exchangeImplementations(origMethod, newMethod); 
} 

，然後在你的單元測試，你將能夠使用它像這樣

SwizzleClassMethod([NSDate class], @selector(date), @selector(mockCurrentDate)); 
[NSDate setMockDate:@"2007-03-24 10:45:32 +0200"]; 
NSLog(@"Date is: %@", [NSDate date]);