twitch-dl/tests/test_api.py

"""
These tests depend on the channel having some videos and clips published.
"""

import httpx
import m3u8
from twitchdl import twitch
from twitchdl.commands.download import _parse_playlists, get_clip_authenticated_url
from twitchdl.models import Game, VideosPage

TEST_CHANNEL = "bananasaurus_rex"


def test_get_videos():
    page = twitch.get_channel_videos(TEST_CHANNEL, 3, "time")
    assert isinstance(page, VideosPage)
    assert len(page.videos) > 0

    video_id = page.videos[0].id
    video = twitch.get_video(video_id)
    assert video and video.id == video_id

    access_token = twitch.get_access_token(video_id)
    assert "signature" in access_token
    assert "value" in access_token

    playlists = twitch.get_playlists(video_id, access_token)
    assert playlists.startswith("#EXTM3U")

    _, _, url = next(_parse_playlists(playlists))
    playlist = httpx.get(url).text
    assert playlist.startswith("#EXTM3U")

    playlist = m3u8.loads(playlist)
    vod_path = playlist.segments[0].uri
    assert vod_path == "0.ts"


def test_get_clips():
    page = twitch.get_channel_clips(TEST_CHANNEL, "all_time", 3)
    assert len(page.clips) > 0

    slug = page.clips[0].slug
    clip = twitch.get_clip(slug)
    assert clip.slug == slug

    assert get_clip_authenticated_url(slug, "source").startswith("https")


def test_get_game():
    game = twitch.find_game("The Witness")
    assert isinstance(game, Game)
    assert game.id == "17324"
    assert game.name == "The Witness"
    assert game.description

    game = twitch.find_game("Does Not Exist Hopefully")
    assert game is None