Fixed the tests

GetStream · Dec 23, 2024 · 076c74a · 076c74a
1 parent d84e738
commit 076c74a
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/StreamVideoTests/Mock/MockLocalMediaAdapter.swift b/StreamVideoTests/Mock/MockLocalMediaAdapter.swift
@@ -86,6 +86,10 @@ final class MockLocalMediaAdapter: LocalMediaAdapting, Mockable {
         stubbedFunctionInput[.trackInfo]?.append(.trackInfo)
         return stubbedFunction[.trackInfo] as? [Stream_Video_Sfu_Models_TrackInfo] ?? []
     }
+
+    func trackInfo(for collectionType: RTCPeerConnectionTrackInfoCollectionType) -> [Stream_Video_Sfu_Models_TrackInfo] {
+        trackInfo()
+    }
 
     func didUpdatePublishOptions(_ publishOptions: PublishOptions) async throws {
         stubbedFunctionInput[.didUpdatePublishOptions]?

diff --git a/StreamVideoTests/Mock/MockRTCPeerConnectionCoordinator.swift b/StreamVideoTests/Mock/MockRTCPeerConnectionCoordinator.swift
@@ -282,7 +282,8 @@ final class MockRTCPeerConnectionCoordinator:
     }
 
     override func trackInfo(
-        for type: TrackType
+        for type: TrackType,
+        collectionType: RTCPeerConnectionTrackInfoCollectionType
     ) -> [Stream_Video_Sfu_Models_TrackInfo] {
         stubbedFunctionInput[.trackInfo]?.append(
             .trackInfo(trackType: type)

diff --git a/...RTC/v2/PeerConnection/MediaAdapters/LocalMediaAdapters/LocalAudioMediaAdapter_Tests.swift b/...RTC/v2/PeerConnection/MediaAdapters/LocalMediaAdapters/LocalAudioMediaAdapter_Tests.swift
@@ -232,7 +232,7 @@ final class LocalAudioMediaAdapter_Tests: XCTestCase, @unchecked Sendable {
     // MARK: - trackInfo
 
     func test_trackInfo_noPublishedTransceivers_returnsEmptyArray() {
-        XCTAssertTrue(subject.trackInfo().isEmpty)
+        XCTAssertTrue(subject.trackInfo(for: .allAvailable).isEmpty)
     }
 
 //    func test_trackInfo_twoPublishedTransceivers_returnsCorrectArray() async throws {

diff --git a/StreamVideoTests/WebRTC/v2/WebRTCJoinRequestFactory_Tests.swift b/StreamVideoTests/WebRTC/v2/WebRTCJoinRequestFactory_Tests.swift
@@ -527,7 +527,7 @@ final class WebRTCJoinRequestFactory_Tests: XCTestCase, @unchecked Sendable {
         )
         mockPublisher?.stubbedTrackInfo = expected
 
-        let result = subject.buildAnnouncedTracks(mockPublisher)
+        let result = subject.buildAnnouncedTracks(mockPublisher, collectionType: .allAvailable)
         let expected: [Stream_Video_Sfu_Models_TrackInfo] = (expected[.audio] ?? [])
             + (expected[.video] ?? [])
             + (expected[.screenshare] ?? [])