p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Format test_island_config_options_validator.py with black

This commit is contained in:
shreyamalviya 2021-06-09 21:56:44 +05:30
parent 8a321f0290
commit 91a7c42a85
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
monkey/tests/unit_tests/monkey_island/cc/setup/test_island_config_options_validator.py
View File

@ -7,10 +7,10 @@ from common.utils.exceptions import InsecurePermissionsError
from monkey_island.cc.setup.island_config_options import IslandConfigOptions from monkey_island.cc.setup.island_config_options import IslandConfigOptions
from monkey_island.cc.setup.island_config_options_validator import raise_on_invalid_options from monkey_island.cc.setup.island_config_options_validator import raise_on_invalid_options
LINUX_READ_ONLY_BY_USER = 0o400 LINUX_READ_ONLY_BY_USER = 0o400
LINUX_RWX_BY_ALL = 0o777 LINUX_RWX_BY_ALL = 0o777
@pytest.fixture @pytest.fixture
def linux_island_config_options(tmpdir, create_empty_file): def linux_island_config_options(tmpdir, create_empty_file):
crt_file = os.path.join(tmpdir, "test.crt") crt_file = os.path.join(tmpdir, "test.crt")
@ -76,12 +76,12 @@ def test_linux_key_path_insecure_permissions(linux_island_config_options):
def windows_island_config_options(tmpdir, create_empty_file): def windows_island_config_options(tmpdir, create_empty_file):
crt_file = os.path.join(tmpdir, "test.crt") crt_file = os.path.join(tmpdir, "test.crt")
create_empty_file(crt_file) create_empty_file(crt_file)
cmd_to_change_permissions = get_windows_cmd_to_change_permissions(crt_file, 'R') cmd_to_change_permissions = get_windows_cmd_to_change_permissions(crt_file, "R")
subprocess.run(cmd_to_change_permissions, shell=True) subprocess.run(cmd_to_change_permissions, shell=True)
key_file = os.path.join(tmpdir, "test.key") key_file = os.path.join(tmpdir, "test.key")
create_empty_file(key_file) create_empty_file(key_file)
cmd_to_change_permissions = get_windows_cmd_to_change_permissions(key_file, 'R') cmd_to_change_permissions = get_windows_cmd_to_change_permissions(key_file, "R")
subprocess.run(cmd_to_change_permissions, shell=True) subprocess.run(cmd_to_change_permissions, shell=True)
return IslandConfigOptions( return IslandConfigOptions(
@ -121,7 +121,9 @@ def test_windows_crt_path_does_not_exist(windows_island_config_options):
@pytest.mark.skipif(os.name == "posix", reason="Tests Windows (not Posix) permissions.") @pytest.mark.skipif(os.name == "posix", reason="Tests Windows (not Posix) permissions.")
def test_windows_crt_path_insecure_permissions(windows_island_config_options): def test_windows_crt_path_insecure_permissions(windows_island_config_options):
cmd_to_change_permissions = get_windows_cmd_to_change_permissions(windows_island_config_options.crt_path, 'W') cmd_to_change_permissions = get_windows_cmd_to_change_permissions(
windows_island_config_options.crt_path, "W"
)
subprocess.run(cmd_to_change_permissions, shell=True) subprocess.run(cmd_to_change_permissions, shell=True)
with pytest.raises(InsecurePermissionsError): with pytest.raises(InsecurePermissionsError):
@ -138,7 +140,9 @@ def test_windows_key_path_does_not_exist(windows_island_config_options):
@pytest.mark.skipif(os.name == "posix", reason="Tests Windows (not Posix) permissions.") @pytest.mark.skipif(os.name == "posix", reason="Tests Windows (not Posix) permissions.")
def test_windows_key_path_insecure_permissions(windows_island_config_options): def test_windows_key_path_insecure_permissions(windows_island_config_options):
cmd_to_change_permissions = get_windows_cmd_to_change_permissions(windows_island_config_options.key_path, 'W') cmd_to_change_permissions = get_windows_cmd_to_change_permissions(
windows_island_config_options.key_path, "W"
)
subprocess.run(cmd_to_change_permissions, shell=True) subprocess.run(cmd_to_change_permissions, shell=True)
with pytest.raises(InsecurePermissionsError): with pytest.raises(InsecurePermissionsError):