mirror of
https://github.com/ihabunek/twitch-dl
synced 2024-08-30 18:32:25 +00:00
138 lines
4.0 KiB
Python
138 lines
4.0 KiB
Python
import json
|
|
|
|
import pytest
|
|
from click.testing import CliRunner, Result
|
|
|
|
from twitchdl import cli
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def runner():
|
|
return CliRunner(mix_stderr=False)
|
|
|
|
|
|
def assert_ok(result: Result):
|
|
if result.exit_code != 0:
|
|
raise AssertionError(
|
|
f"Command failed with exit code {result.exit_code}\nStderr: {result.stderr}"
|
|
)
|
|
|
|
|
|
def test_info_video(runner: CliRunner):
|
|
result = runner.invoke(cli.info, ["2090131595"])
|
|
assert_ok(result)
|
|
|
|
assert "Frost Fatales 2024 Day 1" in result.stdout
|
|
assert "frozenflygone playing Tomb Raider" in result.stdout
|
|
|
|
|
|
def test_info_video_json(runner: CliRunner):
|
|
result = runner.invoke(cli.info, ["2090131595", "--json"])
|
|
assert_ok(result)
|
|
|
|
video = json.loads(result.stdout)
|
|
assert video["title"] == "Frost Fatales 2024 Day 1"
|
|
assert video["game"] == {"id": "2770", "name": "Tomb Raider"}
|
|
assert video["creator"] == {"login": "frozenflygone", "displayName": "frozenflygone"}
|
|
|
|
|
|
def test_info_clip(runner: CliRunner):
|
|
result = runner.invoke(cli.info, ["PoisedTalentedPuddingChefFrank"])
|
|
assert_ok(result)
|
|
|
|
assert "AGDQ Crashes during Bioshock run" in result.stdout
|
|
assert "GamesDoneQuick playing BioShock" in result.stdout
|
|
|
|
|
|
def test_info_clip_json(runner: CliRunner):
|
|
result = runner.invoke(cli.info, ["PoisedTalentedPuddingChefFrank", "--json"])
|
|
assert_ok(result)
|
|
|
|
clip = json.loads(result.stdout)
|
|
assert clip["slug"] == "PoisedTalentedPuddingChefFrank"
|
|
assert clip["title"] == "AGDQ Crashes during Bioshock run"
|
|
assert clip["game"] == {"id": "15866", "name": "BioShock"}
|
|
assert clip["broadcaster"] == {"displayName": "GamesDoneQuick", "login": "gamesdonequick"}
|
|
|
|
|
|
def test_info_not_found(runner: CliRunner):
|
|
result = runner.invoke(cli.info, ["banana"])
|
|
assert result.exit_code == 1
|
|
assert "Clip banana not found" in result.stderr
|
|
|
|
result = runner.invoke(cli.info, ["12345"])
|
|
assert result.exit_code == 1
|
|
assert "Video 12345 not found" in result.stderr
|
|
|
|
result = runner.invoke(cli.info, [""])
|
|
assert result.exit_code == 1
|
|
assert "Invalid input" in result.stderr
|
|
|
|
|
|
def test_download_clip(runner: CliRunner):
|
|
result = runner.invoke(
|
|
cli.download,
|
|
[
|
|
"PoisedTalentedPuddingChefFrank",
|
|
"-q",
|
|
"source",
|
|
"--dry-run",
|
|
],
|
|
)
|
|
assert_ok(result)
|
|
assert (
|
|
"Found: AGDQ Crashes during Bioshock run by GamesDoneQuick, playing BioShock (30 sec)"
|
|
in result.stdout
|
|
)
|
|
assert (
|
|
"Target: 2020-01-10_3099545841_gamesdonequick_agdq_crashes_during_bioshock_run.mp4"
|
|
in result.stdout
|
|
)
|
|
assert "Dry run, clip not downloaded." in result.stdout
|
|
|
|
|
|
def test_download_video(runner: CliRunner):
|
|
result = runner.invoke(
|
|
cli.download,
|
|
[
|
|
"2090131595",
|
|
"-q",
|
|
"source",
|
|
"--dry-run",
|
|
],
|
|
)
|
|
assert_ok(result)
|
|
assert "Found: Frost Fatales 2024 Day 1 by frozenflygone" in result.stdout
|
|
assert (
|
|
"Output: 2024-03-14_2090131595_frozenflygone_frost_fatales_2024_day_1.mkv" in result.stdout
|
|
)
|
|
assert "Dry run, video not downloaded." in result.stdout
|
|
|
|
|
|
def test_videos(runner: CliRunner):
|
|
result = runner.invoke(cli.videos, ["gamesdonequick", "--json"])
|
|
assert_ok(result)
|
|
videos = json.loads(result.stdout)
|
|
|
|
assert videos["count"] == 10
|
|
assert videos["totalCount"] > 0
|
|
video = videos["videos"][0]
|
|
|
|
result = runner.invoke(cli.videos, "gamesdonequick")
|
|
assert_ok(result)
|
|
|
|
assert f"Video {video['id']}" in result.stdout
|
|
assert video["title"] in result.stdout
|
|
|
|
result = runner.invoke(cli.videos, ["gamesdonequick", "--compact"])
|
|
assert_ok(result)
|
|
|
|
assert video["id"] in result.stdout
|
|
assert video["title"][:60] in result.stdout
|
|
|
|
|
|
def test_videos_channel_not_found(runner: CliRunner):
|
|
result = runner.invoke(cli.videos, ["doesnotexisthopefully"])
|
|
assert result.exit_code == 1
|
|
assert result.stderr.strip() == "Error: Channel doesnotexisthopefully not found"
|