def test_only_plain_files(self):
"""Only plain files will be considered."""
path = "%s%sfile" % (self.test_dir, os.sep)
open(path, "w+").close()
os.symlink(path, "%s%s" % (path, ".mp3"))
os.mkdir("%s%s" % (path, ".ogg"))
self.assertEqual([], get_audio_files(self.test_dir))
def test_only_files_with_audio_extensions(self):
"""Only plain files with an audio file extension will be considered."""
path = "%s%sfile" % (self.test_dir, os.sep)
audio_file1 = "%s%s" % (path, ".mp3")
audio_file2 = "%s%s" % (path, ".ogg")
open(path, "w+").close()
open(audio_file1, "w+").close()
open(audio_file2, "w+").close()
open("%s%s" % (path, ".txt"), "w+").close()
self.assertEqual(
[os.path.basename(audio_file1), os.path.basename(audio_file2)],
get_audio_files(self.test_dir))
