factor out code to handle case where we've already fetched the iamge

1 files changed, 14 insertions, 7 deletions

diff --git a/example/ios/iOS UI Test/iOS UI Test/MCOMessageView.mm b/example/ios/iOS UI Test/iOS UI Test/MCOMessageView.mm
index 97946a96..5e89f33b 100755
--- a/example/ios/iOS UI Test/iOS UI Test/MCOMessageView.mm
+++ b/example/ios/iOS UI Test/iOS UI Test/MCOMessageView.mm
@@ -139,16 +139,23 @@
 		
 		NSString * partUniqueID = [part uniqueID];
 		NSData * data = [[self delegate] MCOMessageView:self dataForPartWithUniqueID:partUniqueID];
+		
+		void (^replaceImages)(NSError *error) = ^(NSError *error) {
+			NSData * downloadedData = [[self delegate] MCOMessageView:self dataForPartWithUniqueID:partUniqueID];
+			NSData * previewData = [[self delegate] MCOMessageView:self previewForData:downloadedData isHTMLInlineImage:[self _isCID:url]];
+			NSString *filename = [NSString stringWithFormat:@"%lu", (unsigned long)urlString.hash];
+			NSURL *cacheURL = [self _cacheJPEGImageData:previewData withFilename:filename];
+			
+			NSString *replaceScript = [NSString stringWithFormat:@"replaceImageSrc(\"%@\", \"%@\")", urlString, cacheURL.absoluteString];
+			[_webView stringByEvaluatingJavaScriptFromString:replaceScript];
+		};
+		
 		if (data == NULL) {
 			[[self delegate] MCOMessageView:self fetchDataForPartWithUniqueID:partUniqueID downloadedFinished:^(NSError * error) {
-				NSData * downloadedData = [[self delegate] MCOMessageView:self dataForPartWithUniqueID:partUniqueID];
-				NSData * previewData = [[self delegate] MCOMessageView:self previewForData:downloadedData isHTMLInlineImage:[self _isCID:url]];
-				NSString *filename = [NSString stringWithFormat:@"%lu", (unsigned long)urlString.hash];
-				NSURL *cacheURL = [self _cacheJPEGImageData:previewData withFilename:filename];
-				
-				NSString *replaceScript = [NSString stringWithFormat:@"replaceImageSrc(\"%@\", \"%@\")", urlString, cacheURL.absoluteString];
-				[_webView stringByEvaluatingJavaScriptFromString:replaceScript];
+				replaceImages(error);
 			}];
+		} else {
+			replaceImages(nil);
 		}
 	}
 }