funkwhale/api/tests/music/test_lyrics.py

71 wiersze
2.0 KiB
Python

from django.urls import reverse
from funkwhale_api.music import models
from funkwhale_api.music import tasks
from funkwhale_api.music import lyrics as lyrics_utils
def test_works_import_lyrics_if_any(lyricswiki_content, mocker, factories):
mocker.patch(
"funkwhale_api.music.lyrics._get_html", return_value=lyricswiki_content
)
lyrics = factories["music.Lyrics"](
url="http://lyrics.wikia.com/System_Of_A_Down:Chop_Suey!"
)
tasks.fetch_content(lyrics_id=lyrics.pk)
lyrics.refresh_from_db()
self.assertIn("Grab a brush and put on a little makeup", lyrics.content)
def test_clean_content():
c = """<div class="lyricbox">Hello<br /><script>alert('hello');</script>Is it me you're looking for?<br /></div>"""
d = lyrics_utils.extract_content(c)
d = lyrics_utils.clean_content(d)
expected = """Hello
Is it me you're looking for?
"""
assert d == expected
def test_markdown_rendering(factories):
content = """Hello
Is it me you're looking for?"""
l = factories["music.Lyrics"](content=content)
expected = "<p>Hello<br />\nIs it me you're looking for?</p>"
assert expected == l.content_rendered
def test_works_import_lyrics_if_any(
lyricswiki_content, works, tracks, mocker, factories, logged_in_client
):
mocker.patch(
"funkwhale_api.musicbrainz.api.works.get",
return_value=works["get"]["chop_suey"],
)
mocker.patch(
"funkwhale_api.musicbrainz.api.recordings.get",
return_value=tracks["get"]["chop_suey"],
)
mocker.patch(
"funkwhale_api.music.lyrics._get_html", return_value=lyricswiki_content
)
track = factories["music.Track"](
work=None, mbid="07ca77cf-f513-4e9c-b190-d7e24bbad448"
)
url = reverse("api:v1:tracks-lyrics", kwargs={"pk": track.pk})
response = logged_in_client.get(url)
assert response.status_code == 200
track.refresh_from_db()
lyrics = models.Lyrics.objects.latest("id")
work = models.Work.objects.latest("id")
assert track.work == work
assert lyrics.work == work