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\''