File size: 1,822 Bytes
49627f0 a3600a9 ccd4117 49627f0 328e22b d2c119a 328e22b 82321d6 328e22b d2c119a 328e22b 82321d6 5afc31c 49627f0 ccd4117 328e22b ccd4117 82321d6 328e22b ccd4117 328e22b ccd4117 82321d6 328e22b 82321d6 328e22b 2f4cd16 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
# -*- coding: utf-8 -*-
"""Reusable dependency injected testing components."""
import gzip
import json
import os
import pytest
from pytube import YouTube
def load_playback_file(filename):
"""Load a gzip json playback file."""
cur_fp = os.path.realpath(__file__)
cur_dir = os.path.dirname(cur_fp)
fp = os.path.join(cur_dir, "mocks", filename)
with gzip.open(fp, "rb") as fh:
content = fh.read().decode("utf-8")
return json.loads(content)
def load_and_init_from_playback_file(filename):
"""Load a gzip json playback file and create YouTube instance."""
pb = load_playback_file(filename)
yt = YouTube(pb["url"], defer_prefetch_init=True)
yt.watch_html = pb["watch_html"]
yt.js = pb["js"]
yt.vid_info = pb["video_info"]
yt.descramble()
return yt
@pytest.fixture
def cipher_signature():
"""Youtube instance initialized with video id 9bZkp7q19f0."""
filename = "yt-video-9bZkp7q19f0.json.gz"
return load_and_init_from_playback_file(filename)
@pytest.fixture
def presigned_video():
"""Youtube instance initialized with video id QRS8MkLhQmM."""
filename = "yt-video-QRS8MkLhQmM.json.gz"
return load_and_init_from_playback_file(filename)
@pytest.fixture
def age_restricted():
"""Youtube instance initialized with video id zRbsm3e2ltw."""
filename = "yt-video-zRbsm3e2ltw-1507777044.json.gz"
return load_playback_file(filename)
@pytest.fixture
def playlist_html():
"""Youtube playlist HTML loaded on 2020-01-25 from
https://www.youtube.com/playlist?list=PLzMcBGfZo4-mP7qA9cagf68V06sko5otr"""
file_path = os.path.join(
os.path.dirname(os.path.realpath(__file__)), "mocks", "playlist.html"
)
with open(file_path, encoding="utf-8") as f:
read_data = f.read()
return read_data
|