Fixed PlaylistInfosResponse's extraction and improved tests.

b5i · Sep 20, 2024 · e16b297 · e16b297
1 parent d2a2220
commit e16b297
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Sources/YouTubeKit/YouTubeResponseTypes/PlaylistInfos/PlaylistInfosResponse.swift b/Sources/YouTubeKit/YouTubeResponseTypes/PlaylistInfos/PlaylistInfosResponse.swift
@@ -95,7 +95,7 @@ public struct PlaylistInfosResponse: ContinuableResponse {
         for videoTab in videoTabsArray {
             guard videoTab["tabRenderer"]["selected"].bool == true else { continue }
 
-            guard let secondVideoArray = videoTab["tabRenderer"]["content"]["sectionListRenderer"]["contents"].array else { continue }
+            guard let secondVideoArray = videoTab["tabRenderer"]["content"]["sectionListRenderer"]["contents"].array ?? videoTab["tabRenderer"]["content"]["playlistVideoListRenderer"]["contents"].array else { continue }
             for secondVideoArrayPart in secondVideoArray {
                 guard let thirdVideoArray = secondVideoArrayPart["itemSectionRenderer"]["contents"].array else { continue }
 

diff --git a/Tests/YouTubeKitTests/YouTubeKitTests.swift b/Tests/YouTubeKitTests/YouTubeKitTests.swift
@@ -782,7 +782,11 @@ final class YouTubeKitTests: XCTestCase {
 
         var playlistInfosResult = try await playlist.fetchVideosThrowing(youtubeModel: YTM)
 
+        XCTAssertFalse(playlistInfosResult.results.isEmpty)
+
         let playlistContinuation = try await playlistInfosResult.fetchContinuationThrowing(youtubeModel: YTM)
+
+        XCTAssertFalse(playlistContinuation.results.isEmpty)
 
         let videoCount = playlistInfosResult.results.count + playlistContinuation.results.count
         playlistInfosResult.mergeWithContinuation(playlistContinuation)