Mirrored_Repos/twitch-dl
1
0
Fork 0
mirror of https://github.com/ihabunek/twitch-dl synced 2024-08-30 18:32:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests

Browse Source
This commit is contained in:
Ivan Habunek 2024-04-10 08:04:21 +02:00
parent c1b58e178f
commit 30301c07b9
No known key found for this signature in database
GPG Key ID: F5F0623FF5EBCB3D
2 changed files with 20 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
tests/test_api.py
View File

@ -3,9 +3,10 @@ These tests depend on the channel having some videos and clips published.
""" """
import httpx import httpx
import m3u8
from twitchdl import twitch from twitchdl import twitch
from twitchdl.commands.download import _parse_playlists, get_clip_authenticated_url from twitchdl.commands.download import get_clip_authenticated_url
from twitchdl.playlists import enumerate_vods, load_m3u8, parse_playlists
TEST_CHANNEL = "bananasaurus_rex" TEST_CHANNEL = "bananasaurus_rex"
@ -17,22 +18,25 @@ def test_get_videos():
video_id = videos["edges"][0]["node"]["id"] video_id = videos["edges"][0]["node"]["id"]
video = twitch.get_video(video_id) video = twitch.get_video(video_id)
assert video is not None
assert video["id"] == video_id assert video["id"] == video_id
access_token = twitch.get_access_token(video_id) access_token = twitch.get_access_token(video_id)
assert "signature" in access_token assert "signature" in access_token
assert "value" in access_token assert "value" in access_token
playlists = twitch.get_playlists(video_id, access_token) playlists_txt = twitch.get_playlists(video_id, access_token)
assert playlists.startswith("#EXTM3U") assert playlists_txt.startswith("#EXTM3U")
name, res, url = next(_parse_playlists(playlists)) playlists = parse_playlists(playlists_txt)
playlist = httpx.get(url).text playlist_url = playlists[0].url
assert playlist.startswith("#EXTM3U")
playlist = m3u8.loads(playlist) playlist_txt = httpx.get(playlist_url).text
vod_path = playlist.segments[0].uri assert playlist_txt.startswith("#EXTM3U")
assert vod_path == "0.ts"
playlist_m3u8 = load_m3u8(playlist_txt)
vods = enumerate_vods(playlist_m3u8)
assert vods[0].path == "0.ts"
def test_get_clips(): def test_get_clips():
@ -45,6 +49,7 @@ def test_get_clips():
slug = clips["edges"][0]["node"]["slug"] slug = clips["edges"][0]["node"]["slug"]
clip = twitch.get_clip(slug) clip = twitch.get_clip(slug)
assert clip is not None
assert clip["slug"] == slug assert clip["slug"] == slug
assert get_clip_authenticated_url(slug, "source") assert get_clip_authenticated_url(slug, "source")

6
twitchdl/playlists.py
View File

@ -51,7 +51,11 @@ def load_m3u8(playlist_m3u8: str) -> m3u8.M3U8:
return m3u8.loads(playlist_m3u8) return m3u8.loads(playlist_m3u8)
def enumerate_vods(document: m3u8.M3U8, start: int | None, end: int | None) -> list[Vod]: def enumerate_vods(
document: m3u8.M3U8,
start: int | None = None,
end: int | None = None,
) -> list[Vod]:
"""Extract VODs for download from document.""" """Extract VODs for download from document."""
vods = [] vods = []
vod_start = 0 vod_start = 0