saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Fix pre-commit

Browse source
This commit is contained in:
Twangboy 2023-05-01 15:06:31 -06:00 • committed by Gareth J. Greenaway
parent 8a1a0577aa
commit 0e86853321
1 changed files with 5 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tests/pytests/unit/states/test_win_lgpo_reg.py
View file

@ -44,15 +44,11 @@ def empty_reg_pol_mach():
class_info = salt.utils.win_lgpo_reg.CLASS_INFO
reg_pol_file = pathlib.Path(class_info["Machine"]["policy_path"])
reg_pol_file.parent.mkdir(parents=True, exist_ok=True)
reg_pol_file.write_bytes(
salt.utils.win_lgpo_reg.REG_POL_HEADER.encode("utf-16-le")
)
reg_pol_file.write_bytes(salt.utils.win_lgpo_reg.REG_POL_HEADER.encode("utf-16-le"))
yield
salt.utils.win_reg.delete_key_recursive(hive="HKLM", key="SOFTWARE\\MyKey1")
salt.utils.win_reg.delete_key_recursive(hive="HKLM", key="SOFTWARE\\MyKey2")
reg_pol_file.write_bytes(
salt.utils.win_lgpo_reg.REG_POL_HEADER.encode("utf-16-le")
)
reg_pol_file.write_bytes(salt.utils.win_lgpo_reg.REG_POL_HEADER.encode("utf-16-le"))
@pytest.fixture
@ -60,17 +56,11 @@ def empty_reg_pol_user():
class_info = salt.utils.win_lgpo_reg.CLASS_INFO
reg_pol_file = pathlib.Path(class_info["User"]["policy_path"])
reg_pol_file.parent.mkdir(parents=True, exist_ok=True)
reg_pol_file.write_bytes(
salt.utils.win_lgpo_reg.REG_POL_HEADER.encode("utf-16-le")
)
with salt.utils.files.fopen(reg_pol_file, "wb") as f:
f.write(salt.utils.win_lgpo_reg.REG_POL_HEADER.encode("utf-16-le"))
reg_pol_file.write_bytes(salt.utils.win_lgpo_reg.REG_POL_HEADER.encode("utf-16-le"))
yield
salt.utils.win_reg.delete_key_recursive(hive="HKCU", key="SOFTWARE\\MyKey1")
salt.utils.win_reg.delete_key_recursive(hive="HKCU", key="SOFTWARE\\MyKey2")
reg_pol_file.write_bytes(
salt.utils.win_lgpo_reg.REG_POL_HEADER.encode("utf-16-le")
)
reg_pol_file.write_bytes(salt.utils.win_lgpo_reg.REG_POL_HEADER.encode("utf-16-le"))
@pytest.fixture
@ -600,9 +590,7 @@ def test_user_value_disabled(empty_reg_pol_user):
Test value.disabled in User policy
"""
result = lgpo_reg.value_disabled(
name="MyValue1",
key="SOFTWARE\\MyKey1",
policy_class="User"
name="MyValue1", key="SOFTWARE\\MyKey1", policy_class="User"
)
expected = {
"changes": {