hbmartin commited on
Commit
86b6955
·
1 Parent(s): 2a1fbc4

fix test_playlist.test_title

Browse files
Files changed (1) hide show
  1. tests/contrib/test_playlist.py +10 -4
tests/contrib/test_playlist.py CHANGED
@@ -1,10 +1,16 @@
1
  # -*- coding: utf-8 -*-
 
 
2
  from pytube import Playlist
3
 
4
 
5
- def test_title():
6
- list_key = "PLsyeobzWxl7poL9JTVyndKe62ieoN-MZ3"
7
- url = "https://www.youtube.com/playlist?list=" + list_key
 
 
 
 
8
  pl = Playlist(url)
9
  pl_title = pl.title()
10
- assert pl_title == "Python Tutorial for Beginners"
 
1
  # -*- coding: utf-8 -*-
2
+ from unittest import mock
3
+
4
  from pytube import Playlist
5
 
6
 
7
+ @mock.patch("pytube.contrib.playlist.request.get")
8
+ def test_title(request_get):
9
+ request_get.return_value = (
10
+ "<title>(149) Python Tutorial for Beginners "
11
+ "(For Absolute Beginners) - YouTube</title>"
12
+ )
13
+ url = "https://www.fakeurl.com/playlist?list=PLsyeobzWxl7poL9JTVyndKe62ieoN"
14
  pl = Playlist(url)
15
  pl_title = pl.title()
16
+ assert pl_title == "(149) Python Tutorial for Beginners (For Absolute Beginners)"