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

tests: Extract duplicate code in Windows tests in test_utils

This commit is contained in:
Shreya 2021-06-14 17:59:57 +05:30
parent 6d360ef865
commit c0d9489100
1 changed files with 20 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
monkey/tests/unit_tests/monkey_island/cc/environment/test_utils.py
View File

@ -9,6 +9,13 @@ from monkey_island.cc.environment.utils import (
is_windows_os, is_windows_os,
) )
if is_windows_os():
import win32api
import win32security
FULL_CONTROL = 2032127
ACE_TYPE_ALLOW = 0
@pytest.fixture @pytest.fixture
def test_path_nested(tmpdir): def test_path_nested(tmpdir):
@ -24,6 +31,17 @@ def test_path(tmpdir):
return path return path
def _get_acl_and_sid_from_path(path: str):
create_secure_file(path)
sid, _, _ = win32security.LookupAccountName("", win32api.GetUserName())
security_descriptor = win32security.GetNamedSecurityInfo(
test_path, win32security.SE_FILE_OBJECT, win32security.DACL_SECURITY_INFORMATION
)
acl = security_descriptor.GetSecurityDescriptorDacl()
return acl, sid
def test_create_secure_directory__already_created(test_path): def test_create_secure_directory__already_created(test_path):
os.mkdir(test_path) os.mkdir(test_path)
assert os.path.isdir(test_path) assert os.path.isdir(test_path)
@ -44,19 +62,9 @@ def test_create_secure_directory__perm_linux(test_path):
@pytest.mark.skipif(not is_windows_os(), reason="Tests Windows (not Posix) permissions.") @pytest.mark.skipif(not is_windows_os(), reason="Tests Windows (not Posix) permissions.")
def test_create_secure_directory__perm_windows(test_path): def test_create_secure_directory__perm_windows(test_path):
import win32api
import win32security
FULL_CONTROL = 2032127
ACE_TYPE_ALLOW = 0
create_secure_directory(test_path) create_secure_directory(test_path)
user_sid, _, _ = win32security.LookupAccountName("", win32api.GetUserName()) acl, user_sid = _get_acl_and_sid_from_path()
security_descriptor = win32security.GetNamedSecurityInfo(
test_path, win32security.SE_FILE_OBJECT, win32security.DACL_SECURITY_INFORMATION
)
acl = security_descriptor.GetSecurityDescriptorDacl()
assert acl.GetAceCount() == 1 assert acl.GetAceCount() == 1
@ -89,19 +97,9 @@ def test_create_secure_file__perm_linux(test_path):
@pytest.mark.skipif(not is_windows_os(), reason="Tests Windows (not Posix) permissions.") @pytest.mark.skipif(not is_windows_os(), reason="Tests Windows (not Posix) permissions.")
def test_create_secure_file__perm_windows(test_path): def test_create_secure_file__perm_windows(test_path):
import win32api
import win32security
FULL_CONTROL = 2032127
ACE_TYPE_ALLOW = 0
create_secure_file(test_path) create_secure_file(test_path)
user_sid, _, _ = win32security.LookupAccountName("", win32api.GetUserName()) acl, user_sid = _get_acl_and_sid_from_path()
security_descriptor = win32security.GetNamedSecurityInfo(
test_path, win32security.SE_FILE_OBJECT, win32security.DACL_SECURITY_INFORMATION
)
acl = security_descriptor.GetSecurityDescriptorDacl()
assert acl.GetAceCount() == 1 assert acl.GetAceCount() == 1