Bladeren bron

FIX randomdir broken by files with unknown mimetype

boyska 6 jaren geleden
bovenliggende
commit
c3d7015d9d
2 gewijzigde bestanden met toevoegingen van 16 en 0 verwijderingen
  1. 2 0
      larigira/fsutils.py
  2. 14 0
      larigira/tests/test_audiogen_randomdir.py

+ 2 - 0
larigira/fsutils.py

@@ -20,6 +20,8 @@ def multi_fnmatch(fname, extensions):
 
 def is_audio(fname):
     mimetype = mimetypes.guess_type(fname)[0]
+    if mimetype is None:
+        return False
     return mimetype.split('/')[0] == 'audio'
 
 

+ 14 - 0
larigira/tests/test_audiogen_randomdir.py

@@ -43,3 +43,17 @@ def test_same_name(tmpdir):
     p.write('')
 
     assert len(candidates([P(tmpdir)])) == 2
+
+
+def test_unknown_mime_ignore(tmpdir):
+    p = tmpdir.join("foo.???")
+    p.write('')
+    assert len(candidates([P(tmpdir)])) == 0
+
+
+def test_unknown_mime_nocrash(tmpdir):
+    p = tmpdir.join("foo.???")
+    p.write('')
+    p = tmpdir.join("foo.ogg")
+    p.write('')
+    assert len(candidates([P(tmpdir)])) == 1