From 5c6f3c5b7cc89841955cbaad674c3d60cb0661ca Mon Sep 17 00:00:00 2001
From: Gaelan Steele <gbs@canishe.com>
Date: Mon, 30 Mar 2020 22:16:54 -0700
Subject: [PATCH] Add test for haokan.baidu.com

---
 src/you_get/extractors/baidu.py | 2 ++
 tests/test.py                   | 9 ++++++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/you_get/extractors/baidu.py b/src/you_get/extractors/baidu.py
index 521d5e99..cbdee2a2 100644
--- a/src/you_get/extractors/baidu.py
+++ b/src/you_get/extractors/baidu.py
@@ -160,6 +160,8 @@ def baidu_download(url, output_dir='.', stream_type=None, merge=True, info_only=
             if not info_only:
                 download_urls(urls, title, ext, size,
                               output_dir=output_dir, merge=False)
+    else:
+        raise NotImplementedError("Unknown baidu URL " + url)
 
 
 def baidu_pan_download(url):
diff --git a/tests/test.py b/tests/test.py
index 6fd3db6c..67315a5b 100644
--- a/tests/test.py
+++ b/tests/test.py
@@ -8,7 +8,8 @@ from you_get.extractors import (
     youtube,
     missevan,
     acfun,
-    bilibili
+    bilibili,
+    baidu
 )
 
 
@@ -45,5 +46,11 @@ class YouGetTests(unittest.TestCase):
         bilibili.download(
             "https://www.bilibili.com/watchlater/#/av74906671/p6", info_only=True
         )
+
+    def test_baidu_haokan(self):
+        baidu.download(
+            'https://haokan.baidu.com/v?vid=6794443542184039159&pd=bjh&fr=bjhauthor&type=video', info_only=True
+        )
+
 if __name__ == '__main__':
     unittest.main()