Có thể đặt lại trạng thái của dispatch_once trong thử nghiệm đơn vị, để làm cho chúng chạy lại

Question

Có thể đặt lại trạng thái mã dispatch_once trong một tearDown thử nghiệm đơn vị không?

Tôi nghĩ sẽ tốt hơn nếu các bài kiểm tra đơn vị của chúng tôi có thể chạy từ trạng thái thực sự sạch sẽ, nhưng chúng tôi đang đấu tranh với dispatch_once và một số đơn được thực hiện với công văn một lần.

Answer 1

Tôi cần lưu ý đầu tiên rằng đây không phải là điều tốt để làm trong bất kỳ tình huống nào ngoài thử nghiệm; thậm chí sau đó, tiến hành cẩn thận - AliSoftware cung cấp một số chi tiết và mã ví dụ trong phần bình luận bên dưới. Xem thêm các câu trả lời thú vị tại Can I declare a dispatch_once_t predicate as a member variable instead of static?, bao gồm một số thông tin quan trọng from the horse's mouth.

dispatch_once_t là typedef d long. Giá trị sai của nó là 0. Nếu bạn đặt lại cờ đó thành 0, dispatch_once() sẽ chạy lại. Vấn đề của bạn là "chỉ" làm thế nào để thay đổi giá trị của một biến tĩnh từ một đơn vị biên dịch khác. Đối với điều này, tôi nghĩ rằng bạn cần một cái móc thử nghiệm debug/đơn vị, như vậy:

MakeWhoopie.h

#import <Foundation/Foundation.h> 

void makeWhoopie(void); 

#ifdef DEBUG 
void resetDispatchOnce(void); 
#endif 

MakeWhoopie.m

#include "MakeWhoopie.h" 

static dispatch_once_t * once_token_debug; 

void makeWhoopie(void) 
{ 

    static dispatch_once_t once_token; 
    once_token_debug = &once_token; // Store address of once_token 
             // to access it in debug function. 
    dispatch_once(&once_token, ^{ 
     NSLog(@"That's what you get, folks."); 
    }); 

    NSLog(@"Making whoopie."); 
} 

#ifdef DEBUG 
void resetDispatchOnce(void) 
{ 
    *once_token_debug = 0; 
} 
#endif 

(Bạn cũng có thể di chuyển once_token lên đến nộp và thay đổi trực tiếp.)

Hãy thử cách này:

#import <Foundation/Foundation.h> 
#import "MakeWhoopie.h" 

int main(int argc, const char * argv[]) 
{ 

    @autoreleasepool { 

     makeWhoopie(); 
     makeWhoopie(); 
     resetDispatchOnce(); 
     makeWhoopie(); 
    } 
    return 0; 
} 

Kết quả trong:

2012-06-07 18: 45: 28,134 ResetDispatchOnce [8628: 403] Đó là những gì bạn nhận được, folks.
2012-06-07 18: 45: 28.163 ResetDispatchOnce [8628: 403] Làm whoopie.
2012-06-07 18: 45: 28.164 ResetDispatchOnce [8628: 403] Làm whoopie.
2012-06-07 18: 45: 28.165 ResetDispatchOnce [8628: 403] Đó là những gì bạn nhận được, mọi người.
2012-06-07 18: 45: 28.165 ResetDispatchOnce [8628: 403] Làm whoopie.

Answer 2

Chúng tôi cũng kiểm tra đơn vị của chúng tôi và đôi khi cần phải thay thế chúng bằng các đối tượng giả hoặc đặt lại chúng. Tôi lấy câu trả lời của Josh và đơn giản hóa nó thêm một chút:

static ArticleManager *_sharedInstance = nil; 
static dispatch_once_t once_token = 0; 

+(ArticleManager *)sharedInstance { 
    dispatch_once(&once_token, ^{ 
     if (_sharedInstance == nil) { 
      _sharedInstance = [[ArticleManager alloc] init]; 
     } 
    }); 
    return _sharedInstance; 
} 

+(void)setSharedInstance:(ArticleManager *)instance { 
    if (instance == nil) once_token = 0; 
    _sharedInstance = instance; 
}