Add test for video where all the streams are playable

WiiliamChik · Feb 17, 2020 · c41011d · c41011d
1 parent 3519369
commit c41011d
Show file tree

Hide file tree

Showing 2 changed files with 530 additions and 0 deletions.
diff --git a/...TubeKit Tests/Cassettes/XCDYouTubeClientTestCase/testVideo2ReturnsAllPlayableStreams.json b/...TubeKit Tests/Cassettes/XCDYouTubeClientTestCase/testVideo2ReturnsAllPlayableStreams.json
diff --git a/XCDYouTubeKit Tests/XCDYouTubeClientTestCase.m b/XCDYouTubeKit Tests/XCDYouTubeClientTestCase.m
@@ -279,6 +279,34 @@ - (void) testVideo1ReturnsNoPlayableStreamsBecauseConnectionError_offline
 	[self waitForExpectationsWithTimeout:5 handler:nil];
 }
 
+- (void) testVideo2ReturnsAllPlayableStreams
+{
+	__weak XCTestExpectation *expectation = [self expectationWithDescription:@""];
+	[[XCDYouTubeClient defaultClient] getVideoWithIdentifier:@"QcIy9NiNbmo" completionHandler:^(XCDYouTubeVideo *video, NSError *error)
+	{
+		XCTAssertNotNil(video);
+		XCTAssertNil(error);
+
+		[[XCDYouTubeClient defaultClient]queryVideo:video cookies:nil completionHandler:^(NSDictionary * _Nonnull streamURLs, NSError * _Nullable queryError, NSDictionary<id, NSError *> *streamErrors) {
+
+			XCTAssertNil(queryError);
+			XCTAssertNil(streamErrors);
+			XCTAssertNotNil(streamURLs);
+
+			for (id key in streamURLs.allKeys)
+			{
+				XCTAssertNotNil(streamURLs[key]);
+			}
+
+			XCTAssertEqual(video.streamURLs.count, streamURLs.count, @"`streamURLs` count should be equal since all the streams are playable.");
+
+			[expectation fulfill];
+		}];
+	}];
+
+	[self waitForExpectationsWithTimeout:90 handler:nil];
+}
+
 - (void) testExpiredLiveVideo
 {
 	__weak XCTestExpectation *expectation = [self expectationWithDescription:@""];