Fix #266: Ensure we do not store bad mimetypes

This commit is contained in:
Eliot Berriot 2018-06-03 18:07:59 +02:00
parent a00a6162c0
commit 5afc3e4637
No known key found for this signature in database
GPG key ID: DD6965E2476E5C27
3 changed files with 9 additions and 4 deletions

View file

@ -15,9 +15,13 @@ def test_guess_mimetype_try_using_extension(factories, mocker):
assert utils.guess_mimetype(f.audio_file) == 'audio/mpeg'
def test_guess_mimetype_try_using_extension_if_fail(factories, mocker):
@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='application/octet-stream')
'magic.from_buffer', return_value=wrong)
f = factories['music.TrackFile'].build(
audio_file__filename='test.mp3')