更改 Cocoa 单元测试的系统日期

Posted 2023-03-08

【中文标题】更改 Cocoa 单元测试的系统日期

【英文标题】：Change system date for Cocoa unit testing

【发布时间】：2012-09-18 06:22:25

【问题描述】：

我正在测试一个调用-[NSDate date] 的方法，并使用系统时间作为参考来执行一些逻辑。我需要在一年中的某些特定日子测试逻辑是否正确，但我不能等到那一天。那么我可以在单元测试时以编程方式更改系统日期吗？

【问题讨论】：

一般情况下，您可以覆盖类别中的-[NSDate date]以返回特定日期。

【参考方案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]);