add test case for check return of audio message cell

dodubaby · Oct 4, 2018 · 09907c1 · 09907c1
1 parent 9ebcd8c
commit 09907c1
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 0 deletions.
diff --git a/Tests/ControllersTest/MessagesViewControllerTests.swift b/Tests/ControllersTest/MessagesViewControllerTests.swift
@@ -185,6 +185,22 @@ class MessagesViewControllerTests: XCTestCase {
         XCTAssertTrue(cell is LocationMessageCell)
     }
 
+    func testCellForItemWithAudioData_returnsAudioMessageCell() {
+        let messagesDataSource = MockMessagesDataSource()
+        sut.messagesCollectionView.messagesDataSource = messagesDataSource
+        messagesDataSource.messages.append(MockMessage(audioURL: URL.init(fileURLWithPath: ""),
+                                                       sender: messagesDataSource.senders[0],
+                                                       messageId: "test_id"))
+
+        sut.messagesCollectionView.reloadData()
+
+        let cell = sut.messagesCollectionView.dataSource?.collectionView(sut.messagesCollectionView,
+                                                                         cellForItemAt: IndexPath(item: 0, section: 0))
+
+        XCTAssertNotNil(cell)
+        XCTAssertTrue(cell is AudioMessageCell)
+    }
+
     // MARK: - Assistants
 
     private func makeMessages(for senders: [Sender]) -> [MessageType] {

diff --git a/Tests/Mocks/MockMessage.swift b/Tests/Mocks/MockMessage.swift
@@ -24,6 +24,7 @@
 
 import Foundation
 import CoreLocation
+import AVFoundation
 @testable import MessageKit
 
 struct MockLocationItem: LocationItem {
@@ -53,6 +54,23 @@ struct MockMediaItem: MediaItem {
 
 }
 
+private struct MockAudiotem: AudioItem {
+
+    var url: URL
+    var size: CGSize
+    var duration: Float
+
+    init(url: URL) {
+        self.url = url
+        self.size = CGSize(width: 160, height: 35)
+        // compute duration
+        let audioAsset = AVURLAsset.init(url: url)
+        self.duration = Float(CMTimeGetSeconds(audioAsset.duration))
+    }
+
+}
+
+
 struct MockMessage: MessageType {
 
     var messageId: String
@@ -94,4 +112,9 @@ struct MockMessage: MessageType {
         self.init(kind: .emoji(emoji), sender: sender, messageId: messageId)
     }
 
+    init(audioURL: URL, sender: Sender, messageId: String) {
+        let audioItem = MockAudiotem(url: audioURL)
+        self.init(kind: .audio(audioItem), sender: sender, messageId: messageId)
+    }
+
 }