funkwhale/api/tests/music/test_utils.py

import os

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}),
    ],
)
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