add token negotiation tests

1 files changed, 26 insertions, 0 deletions

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