Шаблон для асинхронной очереди модульного тестирования, которая вызывает основную очередь по завершении

Это связано с моим предыдущим вопросом , но достаточно отличается, чтобы я решил, что бросил бы его в новый. У меня есть код, который запускает асинхронно в настраиваемой очереди, а затем по завершении выполняет блок завершения в основном потоке. Я хотел бы написать модульный тест вокруг этого метода. Мой метод MyObject выглядит так.

+ (void)doSomethingAsyncThenRunCompletionBlockOnMainQueue:(void (^)())completionBlock {

    dispatch_queue_t customQueue = dispatch_queue_create("com.myObject.myCustomQueue", 0);

    dispatch_async(customQueue, ^(void) {

        dispatch_queue_t currentQueue = dispatch_get_current_queue();
        dispatch_queue_t mainQueue = dispatch_get_main_queue();

        if (currentQueue == mainQueue) {
            NSLog(@"already on main thread");
            completionBlock();
        } else {
            dispatch_async(mainQueue, ^(void) {
                NSLog(@"NOT already on main thread");
                completionBlock();
        }); 
    }
});

}

Я добавил тест основной очереди для дополнительной безопасности, но он всегда попадает в dispatch_async . Мой модульный тест выглядит следующим образом.

- (void)testDoSomething {

    dispatch_semaphore_t sema = dispatch_semaphore_create(0);

    void (^completionBlock)(void) = ^(void){        
        NSLog(@"Completion Block!");
        dispatch_semaphore_signal(sema);
    }; 

    [MyObject doSomethingAsyncThenRunCompletionBlockOnMainQueue:completionBlock];

    // Wait for async code to finish
    dispatch_semaphore_wait(sema, DISPATCH_TIME_FOREVER);
    dispatch_release(sema);

    STFail(@"I know this will fail, thanks");
}

Я создаю семафор, чтобы заблокировать завершение теста до того, как это сделает асинхронный код. Это было бы отлично, если бы мне не требовалось, чтобы блок завершения работал в основном потоке. Однако, как указали несколько человек в вопросе, на который я ссылался выше, тот факт, что тест выполняется в основном потоке, а затем я помещаю блок завершения в основной поток, означает, что я просто буду зависать навсегда.

Вызов основной очереди из асинхронной очереди - это шаблон, который я часто вижу для обновления пользовательского интерфейса и тому подобного. Есть ли у кого-нибудь лучший образец для тестирования асинхронного кода, который обращается к основной очереди?

32
задан Community 23 May 2017 в 12:34
поделиться