diff options
Diffstat (limited to 'example/ios')
-rwxr-xr-x | example/ios/iOS UI Test/iOS UI Test/MCTMsgViewController.mm | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/example/ios/iOS UI Test/iOS UI Test/MCTMsgViewController.mm b/example/ios/iOS UI Test/iOS UI Test/MCTMsgViewController.mm index aeca9384..b60e91c7 100755 --- a/example/ios/iOS UI Test/iOS UI Test/MCTMsgViewController.mm +++ b/example/ios/iOS UI Test/iOS UI Test/MCTMsgViewController.mm @@ -188,8 +188,14 @@ typedef void (^DownloadCallback)(NSError * error); - (NSData *) MCOMessageView:(MCOMessageView *)view dataForPartWithUniqueID:(NSString *)partUniqueID { - NSData * data = [_storage objectForKey:partUniqueID]; - return data; + if ([[NSUserDefaults standardUserDefaults] boolForKey:@"FetchFullMessageEnabled"]) { + MCOAttachment * attachment = (MCOAttachment *) [[_messageView message] partForUniqueID:partUniqueID]; + return [attachment data]; + } + else { + NSData * data = [_storage objectForKey:partUniqueID]; + return data; + } } - (void) MCOMessageView:(MCOMessageView *)view fetchDataForPartWithUniqueID:(NSString *)partUniqueID |