fediplug/test_fediplug.py

from os import environ

from fediplug.mastodon import extract_links
from fediplug.queue import build_play_command


def test_extract_links():
    toot = {
        "content": '<p><a href="https://cybre.space/tags/nowplaying" class="mention hashtag" rel="tag">#<span>nowplaying</span></a> <a href="https://cybre.space/tags/fediplug" class="mention hashtag" rel="tag">#<span>fediplug</span></a> Grimes ft. Janelle Mon\u00e1e - Venus Fly <a href="https://www.youtube.com/watch?v=eTLTXDHrgtw" rel="nofollow noopener" target="_blank"><span class="invisible">https://www.</span><span class="ellipsis">youtube.com/watch?v=eTLTXDHrgt</span><span class="invisible">w</span></a></p>'
    }
    urls = extract_links(toot)
    assert urls == ["https://www.youtube.com/watch?v=eTLTXDHrgtw"]


def test_build_play_command_default():
    environ.pop("FEDIPLAY_PLAY_COMMAND")
    play_command = build_play_command("Awesome Music.mp3")
    assert (
        play_command
        == "ffplay -v 0 -nostats -hide_banner -autoexit -nodisp 'Awesome Music.mp3'"
    )


def test_build_play_command_specified():
    environ.update(FEDIPLAY_PLAY_COMMAND="afplay {filename}")
    play_command = build_play_command("Awesome Music.mp3")
    assert play_command == "afplay 'Awesome Music.mp3'"