如何使用OCMock測試是否在完成處理程序塊內調用了對象的方法？

Question

我有一個方法：

@implementation SomeClass

- (void)thisMethod:(ObjectA *)objA {
    [APIClient connectToAPIWithCompletionHandler:^(id result){
        if (result) [objA methodOne];
        else [objA methodTwo];
    }];
}

有沒有一種方法來驗證methodOne或methodTwo時將被稱作thisMethod:被稱為？ 基本上我只想存根connectToAPIWithCompletionHandler:方法。 現在我可以通過調用connectToAPIWithCompletionHandler:方法來實現這一點。 但我想知道是否有更好的方法。

我在這里找到了類似的問題，但它使用的是實例方法，而在我的例子中是類方法。

Answer 1

嘗試這個：

- (void)test_thisMethod {
    id mockA = [OCMockObject mockForClass:[ObjectA class]];
    id mockClient = [OCMockObject mockForClass:[APIClient class]];

    // Use class method mocking on APIClient
    [[mockClient expect] andDo:(NSInvocation *invocation) {
        void (^completion)(id result) = [invocation getArgumentAtIndexAsObject:2];
        completion(nil);
    }] connectToAPIWithCompletionHandler:OCMOCK_ANY];

    [[mockA expect] methodTwo];

    [[SomeClass new] thisMethod:mockA];

    [mockA verify];
    [mockClient verify];        
}

請注意，我直接在瀏覽器中鍵入了這個，但希望它接近工作。