From 7eb9e38eb65b3c78b9914c14e1f59456eec431f4 Mon Sep 17 00:00:00 2001 From: Danny Aziz Date: Wed, 20 Mar 2019 17:06:05 +0000 Subject: [PATCH] Added more proxies tests --- youtube_transcript_api/test/test_cli.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/youtube_transcript_api/test/test_cli.py b/youtube_transcript_api/test/test_cli.py index 11fdf29..4e1495e 100644 --- a/youtube_transcript_api/test/test_cli.py +++ b/youtube_transcript_api/test/test_cli.py @@ -50,6 +50,15 @@ class TestYouTubeTranscriptCli(TestCase): self.assertEqual(parsed_args.https_proxy, 'https://user:pass@domain:port') self.assertEqual(parsed_args.http_proxy, '') + parsed_args = YouTubeTranscriptCli( + 'v1 v2 --languages de en --json'.split() + )._parse_args() + self.assertEqual(parsed_args.video_ids, ['v1', 'v2']) + self.assertEqual(parsed_args.json, True) + self.assertEqual(parsed_args.languages, ['de', 'en']) + self.assertEqual(parsed_args.http_proxy, '') + self.assertEqual(parsed_args.https_proxy, '') + def test_argument_parsing__only_video_ids(self): parsed_args = YouTubeTranscriptCli('v1 v2'.split())._parse_args() self.assertEqual(parsed_args.video_ids, ['v1', 'v2']) @@ -88,6 +97,18 @@ class TestYouTubeTranscriptCli(TestCase): )._parse_args() self.assertEqual(parsed_args.https_proxy, 'https://user:pass@domain:port') + parsed_args = YouTubeTranscriptCli( + 'v1 v2 --http-proxy http://user:pass@domain:port --https-proxy https://user:pass@domain:port'.split() + )._parse_args() + self.assertEqual(parsed_args.http_proxy, 'http://user:pass@domain:port') + self.assertEqual(parsed_args.https_proxy, 'https://user:pass@domain:port') + + parsed_args = YouTubeTranscriptCli( + 'v1 v2'.split() + )._parse_args() + self.assertEqual(parsed_args.http_proxy, '') + self.assertEqual(parsed_args.https_proxy, '') + def test_run(self): YouTubeTranscriptApi.get_transcripts = MagicMock(return_value=([], [])) YouTubeTranscriptCli('v1 v2 --languages de en'.split()).run()