From d75085663ff6638a9b3587c1abba5636fddfbdb8 Mon Sep 17 00:00:00 2001 From: garret Date: Wed, 25 Oct 2023 13:45:45 +0100 Subject: add token negotiation tests --- test_tokens.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100755 test_tokens.py (limited to 'test_tokens.py') diff --git a/test_tokens.py b/test_tokens.py new file mode 100755 index 0000000..ba6475f --- /dev/null +++ b/test_tokens.py @@ -0,0 +1,26 @@ +#!/usr/bin/env python3 +import unittest + +from yt_dlp_plugins.extractor import radiko +from yt_dlp import YoutubeDL + + +class test_tokens(unittest.TestCase): + + def setUp(self): + self.ie = radiko._RadikoBaseIE() + ydl = YoutubeDL(auto_init=False) + self.ie.set_downloader(ydl) + + def test_area(self): + # check areas etc work + for i in range(1, 48): + area = "JP" + str(i) + with self.subTest(f"Negotiating token for {area}", area=area): + token = self.ie._negotiate_token(area) + self.assertEqual(token.get("X-Radiko-AreaId"), area) + + +if __name__ == '__main__': + unittest.main() + # may wish to set failfast=True -- cgit v1.2.3-70-g09d2