Fix crash in demo app on simulating incoming audio message (jessesquires#1542)

GianniCarlo · jessesquires · commit 0fef325f6a49 · 2016-04-12T21:59:23.000-07:00
closes jessesquires#1541
diff --git a/JSQMessagesDemo/DemoMessagesViewController.m b/JSQMessagesDemo/DemoMessagesViewController.m
@@ -220,6 +220,18 @@ - (void)receiveMessagePressed:(UIBarButtonItem *)sender
                 
                 newMediaData = videoItemCopy;
             }
+            else if ([copyMediaData isKindOfClass:[JSQAudioMediaItem class]]) {
+                JSQAudioMediaItem *audioItemCopy = [((JSQAudioMediaItem *)copyMediaData) copy];
+                audioItemCopy.appliesMediaViewMaskAsOutgoing = NO;
+                newMediaAttachmentCopy = [audioItemCopy.audioData copy];
+                
+                /**
+                 *  Reset audio item to simulate "downloading" the audio
+                 */
+                audioItemCopy.audioData = nil;
+                
+                newMediaData = audioItemCopy;
+            }
             else {
                 NSLog(@"%s error: unrecognized media item", __PRETTY_FUNCTION__);
             }
@@ -276,6 +288,10 @@ - (void)receiveMessagePressed:(UIBarButtonItem *)sender
                     ((JSQVideoMediaItem *)newMediaData).isReadyToPlay = YES;
                     [self.collectionView reloadData];
                 }
+                else if ([newMediaData isKindOfClass:[JSQAudioMediaItem class]]) {
+                    ((JSQAudioMediaItem *)newMediaData).audioData = newMediaAttachmentCopy;
+                    [self.collectionView reloadData];
+                }
                 else {
                     NSLog(@"%s error: unrecognized media item", __PRETTY_FUNCTION__);
                 }
