143 lines
4.7 KiB
Python
143 lines
4.7 KiB
Python
import os
|
|
import pathlib
|
|
import tempfile
|
|
|
|
import pytest
|
|
|
|
from funkwhale_api.music import utils
|
|
|
|
DATA_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
|
|
def test_guess_mimetype_try_using_extension(factories, mocker):
|
|
mocker.patch("magic.from_buffer", return_value="audio/mpeg")
|
|
f = factories["music.Upload"].build(audio_file__filename="test.ogg")
|
|
|
|
assert utils.guess_mimetype(f.audio_file) == "audio/mpeg"
|
|
|
|
|
|
@pytest.mark.parametrize("wrong", ["application/octet-stream", "application/x-empty"])
|
|
def test_guess_mimetype_try_using_extension_if_fail(wrong, factories, mocker):
|
|
mocker.patch("magic.from_buffer", return_value=wrong)
|
|
f = factories["music.Upload"].build(audio_file__filename="test.mp3")
|
|
|
|
assert utils.guess_mimetype(f.audio_file) == "audio/mpeg"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"name, expected",
|
|
[
|
|
("sample.flac", {"bitrate": 1608000, "length": 0.001}),
|
|
("test.mp3", {"bitrate": 8000, "length": 267.70285714285717}),
|
|
("test.ogg", {"bitrate": 112000, "length": 1}),
|
|
("test.opus", {"bitrate": 0, "length": 1}), # This Opus file lacks a bitrate
|
|
],
|
|
)
|
|
def test_get_audio_file_data(name, expected):
|
|
path = os.path.join(DATA_DIR, name)
|
|
with open(path, "rb") as f:
|
|
result = utils.get_audio_file_data(f)
|
|
|
|
assert result == expected
|
|
|
|
|
|
def test_guess_mimetype_dont_crash_with_s3(factories, mocker, settings):
|
|
"""See #857"""
|
|
settings.DEFAULT_FILE_STORAGE = "funkwhale_api.common.storage.ASCIIS3Boto3Storage"
|
|
mocker.patch("magic.from_buffer", return_value="none")
|
|
f = factories["music.Upload"].build(audio_file__filename="test.mp3")
|
|
|
|
assert utils.guess_mimetype(f.audio_file) == "audio/mpeg"
|
|
|
|
|
|
def test_increment_downloads_count(factories, mocker, cache, anonymous_user, settings):
|
|
ident = {"type": "anonymous", "id": "noop"}
|
|
get_ident = mocker.patch(
|
|
"funkwhale_api.common.throttling.get_ident", return_value=ident
|
|
)
|
|
cache_set = mocker.spy(utils.cache, "set")
|
|
wsgi_request = mocker.Mock(META={})
|
|
upload = factories["music.Upload"]()
|
|
utils.increment_downloads_count(
|
|
upload=upload, user=anonymous_user, wsgi_request=wsgi_request
|
|
)
|
|
|
|
upload.refresh_from_db()
|
|
get_ident.assert_called_once_with(user=anonymous_user, request=wsgi_request)
|
|
|
|
assert upload.downloads_count == 1
|
|
assert upload.track.downloads_count == 1
|
|
cache_set.assert_called_once_with(
|
|
"downloads_count:upload-{}:{}-{}".format(upload.pk, ident["type"], ident["id"]),
|
|
1,
|
|
settings.MIN_DELAY_BETWEEN_DOWNLOADS_COUNT,
|
|
)
|
|
|
|
|
|
def test_increment_downloads_count_already_tracked(
|
|
factories, mocker, cache, anonymous_user
|
|
):
|
|
ident = {"type": "anonymous", "id": "noop"}
|
|
mocker.patch("funkwhale_api.common.throttling.get_ident", return_value=ident)
|
|
wsgi_request = mocker.Mock(META={})
|
|
upload = factories["music.Upload"]()
|
|
cache.set(
|
|
"downloads_count:upload-{}:{}-{}".format(upload.pk, ident["type"], ident["id"]),
|
|
1,
|
|
)
|
|
|
|
utils.increment_downloads_count(
|
|
upload=upload, user=anonymous_user, wsgi_request=wsgi_request
|
|
)
|
|
|
|
upload.refresh_from_db()
|
|
|
|
assert upload.downloads_count == 0
|
|
assert upload.track.downloads_count == 0
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"path,expected",
|
|
[
|
|
("", [{"name": "Magic", "dir": True}, {"name": "System", "dir": True}]),
|
|
("Magic", [{"name": "file.mp3", "dir": False}]),
|
|
("System", [{"name": "file.ogg", "dir": False}]),
|
|
],
|
|
)
|
|
def test_get_dirs_and_files(path, expected, tmpdir):
|
|
root_path = pathlib.Path(tmpdir)
|
|
(root_path / "Magic").mkdir()
|
|
(root_path / "Magic" / "file.mp3").touch()
|
|
(root_path / "System").mkdir()
|
|
(root_path / "System" / "file.ogg").touch()
|
|
|
|
assert utils.browse_dir(root_path, path) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"name, expected",
|
|
[
|
|
("sample.flac", {"bitrate": 128000, "length": 0}),
|
|
("test.mp3", {"bitrate": 16000, "length": 268}),
|
|
("test.ogg", {"bitrate": 128000, "length": 1}),
|
|
("test.opus", {"bitrate": 128000, "length": 1}),
|
|
],
|
|
)
|
|
def test_transcode_file(name, expected):
|
|
path = pathlib.Path(os.path.join(DATA_DIR, name))
|
|
with tempfile.NamedTemporaryFile() as dest:
|
|
utils.transcode_file(path, pathlib.Path(dest.name))
|
|
with open(dest.name, "rb") as f:
|
|
result = {k: round(v) for k, v in utils.get_audio_file_data(f).items()}
|
|
|
|
assert result == expected
|
|
|
|
|
|
def test_custom_s3_domain(factories, settings):
|
|
"""See #2220"""
|
|
settings.AWS_S3_CUSTOM_DOMAIN = "my.custom.domain.tld"
|
|
settings.DEFAULT_FILE_STORAGE = "funkwhale_api.common.storage.ASCIIS3Boto3Storage"
|
|
f = factories["music.Upload"].build(audio_file__filename="test.mp3")
|
|
|
|
assert f.audio_file.url.startswith("https://")
|