Tomar captura de pantalla con Kif

Quiero tomar una captura de pantalla en Kif. Vi en una class privada en el proyecto kif que es posible, pero estoy teniendo dificultades para tratar de convertirlo en un paso. ¿Alguien puede ayudar?

Esto se agregó en KIF 3.0.4. Vea aquí

Después de jugar con esto un poco, agregué esto a mi KIFTestStep.m

  + (id)stepToTakeScreenShotwithName:(NSString *)name; { NSString *description = [NSString stringWithFormat:@"Take a screenshot saved by the name %@", name]; return [self stepWithDescription:description executionBlock:^(KIFTestStep *step, NSError **error) { NSString *outputPath = [NSString stringWithFormat:@"/Users/%@/ScreenShots", NSUserName()]; NSArray *windows = [[UIApplication shanetworkingApplication] windows]; if (windows.count == 0) { return KIFTestStepResultFailure; } UIGraphicsBeginImageContext([[windows objectAtIndex:0] bounds].size); for (UIWindow *window in windows) { [window.layer renderInContext:UIGraphicsGetCurrentContext()]; } UIImage *image = UIGraphicsGetImageFromCurrentImageContext(); UIGraphicsEndImageContext(); NSTimeInterval timeStamp = [[NSDate date] timeIntervalSince1970]; NSNumber *timeStampObj = [NSNumber numberWithDouble: timeStamp]; outputPath = [outputPath stringByExpandingTildeInPath]; outputPath = [outputPath stringByAppendingPathComponent:[name stringByReplacingOccurrencesOfString:@"/" withString:@"_"]]; outputPath = [outputPath stringByAppendingString:[timeStampObj stringValue]]; outputPath = [outputPath stringByAppendingPathExtension:@"png"]; [UIImagePNGRepresentation(image) writeToFile:outputPath atomically:YES]; return KIFTestStepResultSuccess; }]; }