OCMock异步块回调

时间:2011-02-01 18:01:55

标签: cocoa unit-testing objective-c-blocks ocmock

我正在构建一个小型库来处理文件上传和下载操作,并尝试将一套测试集成到其中。我没有使用委托回调方法,而是在完成处理程序块中处理异步响应,如下所示:

  SyncKit *engine = [[SyncKit alloc] init];
  NSURL *localFilePath = [NSURL URLWithString:@"/Users/justin/Desktop/FileTest.png"];

  [engine uploadFileWithFilename:@"FileTest.png" toRemotePath:@"/" fromLocalPath:localFilePath withCompletionHandler:^(id response, NSError *error) {
    if (error)
    {
      NSLog(@"error = %@", error);
      return;
    }

    NSLog(@"File uploaded and return JSON response = %@", response);
  }];  

基础uploadFileWithFilename...方法如下:

- (void)uploadFileWithFilename:(NSString *)filename toRemotePath:(NSString *)remotePath fromLocalPath:(NSURL *)localPath withCompletionHandler:(SKCompletionHandler)handler
{
  if ((![[NSFileManager defaultManager] fileExistsAtPath:[localPath path]]))
  {
    NSDictionary *userInfo = [NSDictionary dictionaryWithObject:localPath forKey:@"localPath"];
    NSError *error = [NSError errorWithDomain:SKDropboxErrorDomain code:SKDropboxErrorFileNotFound userInfo:userInfo];
    handler(nil, error);
    return;
  }

  // path is the directory the file will be uploaded to, make sure it doesn't have a trailing /
  // (unless it's the root dir) and is properly escaped
  NSString *trimmedPath;
  if (([remotePath length] > 1) && ([remotePath characterAtIndex:[remotePath length] - 1] == '/')) 
  {
    trimmedPath = [remotePath substringToIndex:[remotePath length] - 1];
  } 
  else if ([remotePath isEqualToString:@"/"])
  {
    trimmedPath = @"//";
  }
  else 
  {
    trimmedPath = remotePath;
  }

  NSString *escapedPath = [NSString escapePath:trimmedPath];
  NSString *fullPath = [NSString stringWithFormat:@"/files/dropbox%@", escapedPath];      
  NSString *urlString = [NSString stringWithFormat:@"%@://%@/%@%@", kSKProtocolHTTPS, kSKDropboxAPIContentHost, kSKDropboxAPIVersion, fullPath];

  NSDictionary *params = [NSDictionary dictionaryWithObjectsAndKeys:filename, @"file", nil];
  NSString *body = [params convertToURIParameterString];

  NSURL *url = nil;
  if ([body length] == 0)
  {
    url = [NSURL URLWithString:[NSString stringWithFormat:@"%@", urlString]];
  }
  else
  {
    url = [NSURL URLWithString:[NSString stringWithFormat:@"%@?%@", urlString, body]];
  }

  __block ASIFormDataRequest *request = [ASIFormDataRequest requestWithURL:url];
  request.delegate = self;
  request.requestMethod = kSKMethodPOST;

  [request addFile:[localPath path] forKey:@"file"];

  //
  // Dropbox uses OAuth to handle its authentication, so we need to pass along the requested
  // tokens and secrets so that we get our stuff back.
  //  
  NSUserDefaults *defaults = [NSUserDefaults standardUserDefaults];
  NSString *token = [defaults objectForKey:@"oauth_token"];
  NSString *secret = [defaults objectForKey:@"oauth_secret"];  

  [request buildPostBody]; 
  NSData *authBody = request.postBody; 

  NSString *header = OAuthorizationHeader(url, request.requestMethod, authBody, kOAuthConsumerKey, kOAuthConsumerSecret, token, secret);  
  [request addRequestHeader:@"Authorization" value:header];  

  [request setCompletionBlock:^{
    NSDictionary *result = (NSDictionary *)[[request responseString] JSONValue];    
    [self.activeUploads removeObjectForKey:remotePath];
    handler(result, nil);
  }];

  [request setFailedBlock:^{
    NSError *error = request.error;
    [self.activeUploads removeObjectForKey:remotePath];
    handler(nil, error);
  }];

  [self.activeUploads setObject:request forKey:remotePath];

  [self.queue addOperation:request];
}

我看到one example那个人正在使用预处理器定义并将OCMock注入到实际的代码库中。这对我来说似乎不对。

测试这样一段代码的最佳策略是什么?

1 个答案:

答案 0 :(得分:1)

这个答案与OCMock无关,所以它可能不是你想要的,但是......

我会做这样的事情:

__block BOOL testPassed = NO;

[engine uploadFileWithFilename:@"FileTest.png" 
                  toRemotePath:@"/" 
                 fromLocalPath:localFilePath 
         withCompletionHandler:^(id response, NSError *error) {
    if (error)
    {
      return;
    }

    testPassed = YES;
  }];

[[NSRunLoop mainRunLoop] runMode:NSDefaultRunLoopMode 
                      beforeDate:[[NSDate date] dateByAddingTimeInterval:10]];

// make sure that testPassed is YES...

这样,你就会阻塞,直到其中一个回调进入主运行循环。