74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
import json
|
|
from django.urls import reverse
|
|
|
|
from funkwhale_api.music import models
|
|
from funkwhale_api.musicbrainz import api
|
|
from funkwhale_api.music import serializers
|
|
from funkwhale_api.music import lyrics as lyrics_utils
|
|
|
|
from .mocking import lyricswiki
|
|
from . import data as api_data
|
|
|
|
|
|
|
|
def test_works_import_lyrics_if_any(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!')
|
|
|
|
lyrics.fetch_content()
|
|
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(mocker, factories, logged_in_client):
|
|
mocker.patch(
|
|
'funkwhale_api.musicbrainz.api.works.get',
|
|
return_value=api_data.works['get']['chop_suey'])
|
|
mocker.patch(
|
|
'funkwhale_api.musicbrainz.api.recordings.get',
|
|
return_value=api_data.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
|