From 76d4e8de92a2745230e20cc2edace542fc58326a Mon Sep 17 00:00:00 2001
From: Lauren Liberda <laura@selfisekai.rocks>
Date: Wed, 28 Apr 2021 11:41:22 +0200
Subject: [PATCH] [wppilot] add tests

---
 haruhi_dl/extractor/wppilot.py | 38 ++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/haruhi_dl/extractor/wppilot.py b/haruhi_dl/extractor/wppilot.py
index 765e1fc16..8b823df62 100644
--- a/haruhi_dl/extractor/wppilot.py
+++ b/haruhi_dl/extractor/wppilot.py
@@ -95,6 +95,32 @@ class WPPilotIE(WPPilotBaseIE):
     _VALID_URL = r'(?:https?://pilot\.wp\.pl/tv/?#|wppilot:)(?P<id>[a-z\d-]+)'
     IE_NAME = 'wppilot'
 
+    _TESTS = [{
+        'url': 'https://pilot.wp.pl/tv/#telewizja-wp-hd',
+        'info_dict': {
+            'id': '158',
+            'ext': 'm3u8',
+            'title': 'Telewizja WP HD',
+        },
+        'params': {
+            'format': 'bestvideo',
+        },
+    }, {
+        # audio only
+        'url': 'https://pilot.wp.pl/tv/#radio-nowy-swiat',
+        'info_dict': {
+            'id': '238',
+            'ext': 'm3u8',
+            'title': 'Radio Nowy Świat',
+        },
+        'params': {
+            'format': 'bestaudio',
+        },
+    }, {
+        'url': 'wppilot:9',
+        'only_matching': True,
+    }]
+
     def _get_channel(self, id_or_slug):
         video_list = self._get_channel_list(cache=True)
         key = 'id' if re.match(r'^\d+$', id_or_slug) else 'slug'
@@ -174,6 +200,18 @@ class WPPilotChannelsIE(WPPilotBaseIE):
     _VALID_URL = r'(?:https?://pilot\.wp\.pl/(?:tv/?)?(?:\?[^#]*)?#?|wppilot:)$'
     IE_NAME = 'wppilot:channels'
 
+    _TESTS = [{
+        'url': 'wppilot:',
+        'info_dict': {
+            'id': 'wppilot',
+            'title': 'WP Pilot',
+        },
+        'playlist_mincount': 100,
+    }, {
+        'url': 'https://pilot.wp.pl/',
+        'only_matching': True,
+    }]
+
     def _real_extract(self, url):
         channel_list = self._get_channel_list()
         categories = channel_list['_meta']['categories']