mirror of
https://github.com/beetbox/beets.git
synced 2025-12-06 16:42:42 +01:00
a6e5201ff3
This is 'the big one', which touches every file so that it all conforms to the given standard.
116 lines
3.3 KiB
Python
116 lines
3.3 KiB
Python
"""Tests for the 'permissions' plugin.
|
|
"""
|
|
|
|
import os
|
|
import platform
|
|
import unittest
|
|
from test._common import touch
|
|
from test.helper import TestHelper
|
|
from unittest.mock import Mock, patch
|
|
|
|
from beets.util import displayable_path
|
|
from beetsplug.permissions import (
|
|
check_permissions,
|
|
convert_perm,
|
|
dirs_in_library,
|
|
)
|
|
|
|
|
|
class PermissionsPluginTest(unittest.TestCase, TestHelper):
|
|
def setUp(self):
|
|
self.setup_beets()
|
|
self.load_plugins("permissions")
|
|
|
|
self.config["permissions"] = {"file": "777", "dir": "777"}
|
|
|
|
def tearDown(self):
|
|
self.teardown_beets()
|
|
self.unload_plugins()
|
|
|
|
def test_permissions_on_album_imported(self):
|
|
self.do_thing(True)
|
|
|
|
def test_permissions_on_item_imported(self):
|
|
self.config["import"]["singletons"] = True
|
|
self.do_thing(True)
|
|
|
|
@patch("os.chmod", Mock())
|
|
def test_failing_to_set_permissions(self):
|
|
self.do_thing(False)
|
|
|
|
def do_thing(self, expect_success):
|
|
if platform.system() == "Windows":
|
|
self.skipTest("permissions not available on Windows")
|
|
|
|
def get_stat(v):
|
|
return (
|
|
os.stat(os.path.join(self.temp_dir, b"import", *v)).st_mode
|
|
& 0o777
|
|
)
|
|
|
|
self.importer = self.create_importer()
|
|
typs = ["file", "dir"]
|
|
|
|
track_file = (b"album 0", b"track 0.mp3")
|
|
self.exp_perms = {
|
|
True: {
|
|
k: convert_perm(self.config["permissions"][k].get())
|
|
for k in typs
|
|
},
|
|
False: {k: get_stat(v) for (k, v) in zip(typs, (track_file, ()))},
|
|
}
|
|
|
|
self.importer.run()
|
|
item = self.lib.items().get()
|
|
|
|
self.assertPerms(item.path, "file", expect_success)
|
|
|
|
for path in dirs_in_library(self.lib.directory, item.path):
|
|
self.assertPerms(path, "dir", expect_success)
|
|
|
|
def assertPerms(self, path, typ, expect_success): # noqa
|
|
for x in [
|
|
(True, self.exp_perms[expect_success][typ], "!="),
|
|
(False, self.exp_perms[not expect_success][typ], "=="),
|
|
]:
|
|
msg = "{} : {} {} {}".format(
|
|
displayable_path(path),
|
|
oct(os.stat(path).st_mode),
|
|
x[2],
|
|
oct(x[1]),
|
|
)
|
|
self.assertEqual(x[0], check_permissions(path, x[1]), msg=msg)
|
|
|
|
def test_convert_perm_from_string(self):
|
|
self.assertEqual(convert_perm("10"), 8)
|
|
|
|
def test_convert_perm_from_int(self):
|
|
self.assertEqual(convert_perm(10), 8)
|
|
|
|
def test_permissions_on_set_art(self):
|
|
self.do_set_art(True)
|
|
|
|
@patch("os.chmod", Mock())
|
|
def test_failing_permissions_on_set_art(self):
|
|
self.do_set_art(False)
|
|
|
|
def do_set_art(self, expect_success):
|
|
if platform.system() == "Windows":
|
|
self.skipTest("permissions not available on Windows")
|
|
self.importer = self.create_importer()
|
|
self.importer.run()
|
|
album = self.lib.albums().get()
|
|
artpath = os.path.join(self.temp_dir, b"cover.jpg")
|
|
touch(artpath)
|
|
album.set_art(artpath)
|
|
self.assertEqual(
|
|
expect_success, check_permissions(album.artpath, 0o777)
|
|
)
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromName(__name__)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(defaultTest="suite")
|