mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 08:58:59 +00:00
337 lines
13 KiB
Python
337 lines
13 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: Shane Lee <slee@saltstack.com>
|
|
'''
|
|
# Import Python Libs
|
|
from __future__ import absolute_import, unicode_literals, print_function
|
|
import os
|
|
import sys
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.mock import MagicMock, NO_MOCK, NO_MOCK_REASON, patch
|
|
from tests.support.paths import TMP
|
|
from tests.support.unit import TestCase, skipIf
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.temp as temp
|
|
import salt.modules.win_file as win_file
|
|
import salt.utils.platform
|
|
import salt.utils.win_dacl as win_dacl
|
|
import salt.modules.cmdmod as cmdmod
|
|
from salt.exceptions import CommandExecutionError
|
|
|
|
try:
|
|
WIN_VER = sys.getwindowsversion().major
|
|
except AttributeError:
|
|
WIN_VER = 0
|
|
|
|
|
|
class DummyStat(object):
|
|
st_mode = 33188
|
|
st_ino = 115331251
|
|
st_dev = 44
|
|
st_nlink = 1
|
|
st_uid = 99200001
|
|
st_gid = 99200001
|
|
st_size = 41743
|
|
st_atime = 1552661253
|
|
st_mtime = 1552661253
|
|
st_ctime = 1552661253
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class WinFileTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test cases for salt.modules.win_file
|
|
'''
|
|
FAKE_RET = {'fake': 'ret data'}
|
|
if salt.utils.platform.is_windows():
|
|
FAKE_PATH = os.sep.join(['C:', 'path', 'does', 'not', 'exist'])
|
|
else:
|
|
FAKE_PATH = os.sep.join(['path', 'does', 'not', 'exist'])
|
|
|
|
def setup_loader_modules(self):
|
|
return {
|
|
win_file: {
|
|
'__utils__': {'dacl.set_perms': win_dacl.set_perms},
|
|
'__salt__': {'cmd.run_stdout': cmdmod.run_stdout}
|
|
}
|
|
}
|
|
|
|
def test_issue_43328_stats(self):
|
|
'''
|
|
Make sure that a CommandExecutionError is raised if the file does NOT
|
|
exist
|
|
'''
|
|
with patch('os.path.exists', return_value=False):
|
|
self.assertRaises(CommandExecutionError,
|
|
win_file.stats,
|
|
self.FAKE_PATH)
|
|
|
|
def test_issue_43328_check_perms_no_ret(self):
|
|
'''
|
|
Make sure that a CommandExecutionError is raised if the file does NOT
|
|
exist
|
|
'''
|
|
with patch('os.path.exists', return_value=False):
|
|
self.assertRaises(
|
|
CommandExecutionError, win_file.check_perms, self.FAKE_PATH)
|
|
|
|
@skipIf(not salt.utils.platform.is_windows(), 'Skip on Non-Windows systems')
|
|
@skipIf(WIN_VER < 6, 'Symlinks not supported on Vista an lower')
|
|
def test_issue_52002_check_file_remove_symlink(self):
|
|
'''
|
|
Make sure that directories including symlinks or symlinks can be removed
|
|
'''
|
|
base = temp.dir(prefix='base-', parent=TMP)
|
|
target = os.path.join(base, 'child 1', 'target\\')
|
|
symlink = os.path.join(base, 'child 2', 'link')
|
|
try:
|
|
# Create environment
|
|
self.assertFalse(win_file.directory_exists(target))
|
|
self.assertFalse(win_file.directory_exists(symlink))
|
|
self.assertTrue(win_file.makedirs_(target))
|
|
self.assertTrue(win_file.makedirs_(symlink))
|
|
self.assertTrue(win_file.symlink(target, symlink))
|
|
self.assertTrue(win_file.directory_exists(symlink))
|
|
self.assertTrue(win_file.is_link(symlink))
|
|
# Test removal of directory containing symlink
|
|
self.assertTrue(win_file.remove(base))
|
|
self.assertFalse(win_file.directory_exists(base))
|
|
finally:
|
|
if os.path.exists(base):
|
|
win_file.remove(base)
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
@skipIf(not salt.utils.platform.is_windows(), 'Skip on Non-Windows systems')
|
|
class WinFileCheckPermsTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test cases for the check_perms function in salt.modules.win_file
|
|
'''
|
|
temp_file = ''
|
|
current_user = ''
|
|
|
|
def setup_loader_modules(self):
|
|
self.current_user = salt.utils.win_functions.get_current_user(False)
|
|
return {
|
|
win_file: {
|
|
'__utils__': {'dacl.check_perms': win_dacl.check_perms,
|
|
'dacl.set_perms': win_dacl.set_perms}
|
|
},
|
|
win_dacl: {
|
|
'__opts__': {'test': False},
|
|
}
|
|
}
|
|
|
|
def setUp(self):
|
|
self.temp_file = temp.file(parent=TMP)
|
|
salt.utils.win_dacl.set_owner(obj_name=self.temp_file,
|
|
principal=self.current_user)
|
|
salt.utils.win_dacl.set_inheritance(obj_name=self.temp_file,
|
|
enabled=True)
|
|
self.assertEqual(
|
|
salt.utils.win_dacl.get_owner(obj_name=self.temp_file),
|
|
self.current_user)
|
|
|
|
def tearDown(self):
|
|
os.remove(self.temp_file)
|
|
|
|
def test_check_perms_set_owner_test_true(self):
|
|
'''
|
|
Test setting the owner of a file with test=True
|
|
'''
|
|
expected = {'comment': '',
|
|
'changes': {'owner': 'Administrators'},
|
|
'name': self.temp_file,
|
|
'result': None}
|
|
with patch.dict(win_dacl.__opts__, {'test': True}):
|
|
ret = win_file.check_perms(path=self.temp_file,
|
|
owner='Administrators',
|
|
inheritance=None)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_set_owner(self):
|
|
'''
|
|
Test setting the owner of a file
|
|
'''
|
|
expected = {'comment': '',
|
|
'changes': {'owner': 'Administrators'},
|
|
'name': self.temp_file,
|
|
'result': True}
|
|
ret = win_file.check_perms(path=self.temp_file,
|
|
owner='Administrators',
|
|
inheritance=None)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_deny_test_true(self):
|
|
'''
|
|
Test setting deny perms on a file with test=True
|
|
'''
|
|
expected = {'comment': '',
|
|
'changes': {'perms': {'Users': {'deny': 'read_execute'}}},
|
|
'name': self.temp_file,
|
|
'result': None}
|
|
with patch.dict(win_dacl.__opts__, {'test': True}):
|
|
ret = win_file.check_perms(
|
|
path=self.temp_file,
|
|
deny_perms={'Users': {'perms': 'read_execute'}},
|
|
inheritance=None)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_deny(self):
|
|
'''
|
|
Test setting deny perms on a file
|
|
'''
|
|
expected = {'comment': '',
|
|
'changes': {'perms': {'Users': {'deny': 'read_execute'}}},
|
|
'name': self.temp_file,
|
|
'result': True}
|
|
ret = win_file.check_perms(
|
|
path=self.temp_file,
|
|
deny_perms={'Users': {'perms': 'read_execute'}},
|
|
inheritance=None)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_grant_test_true(self):
|
|
'''
|
|
Test setting grant perms on a file with test=True
|
|
'''
|
|
expected = {'comment': '',
|
|
'changes': {'perms': {'Users': {'grant': 'read_execute'}}},
|
|
'name': self.temp_file,
|
|
'result': None}
|
|
with patch.dict(win_dacl.__opts__, {'test': True}):
|
|
ret = win_file.check_perms(
|
|
path=self.temp_file,
|
|
grant_perms={'Users': {'perms': 'read_execute'}},
|
|
inheritance=None)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_grant(self):
|
|
'''
|
|
Test setting grant perms on a file
|
|
'''
|
|
expected = {'comment': '',
|
|
'changes': {'perms': {'Users': {'grant': 'read_execute'}}},
|
|
'name': self.temp_file,
|
|
'result': True}
|
|
ret = win_file.check_perms(
|
|
path=self.temp_file,
|
|
grant_perms={'Users': {'perms': 'read_execute'}},
|
|
inheritance=None)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_inheritance_false_test_true(self):
|
|
'''
|
|
Test setting inheritance to False with test=True
|
|
'''
|
|
expected = {'comment': '',
|
|
'changes': {'inheritance': False},
|
|
'name': self.temp_file,
|
|
'result': None}
|
|
with patch.dict(win_dacl.__opts__, {'test': True}):
|
|
ret = win_file.check_perms(path=self.temp_file,
|
|
inheritance=False)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_inheritance_false(self):
|
|
'''
|
|
Test setting inheritance to False
|
|
'''
|
|
expected = {'comment': '',
|
|
'changes': {'inheritance': False},
|
|
'name': self.temp_file,
|
|
'result': True}
|
|
ret = win_file.check_perms(path=self.temp_file,
|
|
inheritance=False)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_inheritance_true(self):
|
|
'''
|
|
Test setting inheritance to true when it's already true (default)
|
|
'''
|
|
expected = {'comment': '',
|
|
'changes': {},
|
|
'name': self.temp_file,
|
|
'result': True}
|
|
ret = win_file.check_perms(path=self.temp_file,
|
|
inheritance=True)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_reset_test_true(self):
|
|
'''
|
|
Test resetting perms with test=True. This shows minimal changes
|
|
'''
|
|
# Turn off inheritance
|
|
salt.utils.win_dacl.set_inheritance(obj_name=self.temp_file,
|
|
enabled=False,
|
|
clear=True)
|
|
# Set some permissions
|
|
salt.utils.win_dacl.set_permissions(obj_name=self.temp_file,
|
|
principal='Administrator',
|
|
permissions='full_control')
|
|
expected = {'comment': '',
|
|
'changes': {
|
|
'perms': {
|
|
'Administrators': {'grant': 'full_control'},
|
|
'Users': {'grant': 'read_execute'}},
|
|
'remove_perms': {
|
|
'Administrator': {
|
|
'grant': {'applies to': 'Not Inherited (file)',
|
|
'permissions': 'Full control'}}}},
|
|
'name': self.temp_file,
|
|
'result': None}
|
|
with patch.dict(win_dacl.__opts__, {'test': True}):
|
|
ret = win_file.check_perms(
|
|
path=self.temp_file,
|
|
grant_perms={'Users': {'perms': 'read_execute'},
|
|
'Administrators': {'perms': 'full_control'}},
|
|
inheritance=False,
|
|
reset=True)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_check_perms_reset(self):
|
|
'''
|
|
Test resetting perms on a File
|
|
'''
|
|
# Turn off inheritance
|
|
salt.utils.win_dacl.set_inheritance(obj_name=self.temp_file,
|
|
enabled=False,
|
|
clear=True)
|
|
# Set some permissions
|
|
salt.utils.win_dacl.set_permissions(obj_name=self.temp_file,
|
|
principal='Administrator',
|
|
permissions='full_control')
|
|
expected = {'comment': '',
|
|
'changes': {
|
|
'perms': {
|
|
'Administrators': {'grant': 'full_control'},
|
|
'Users': {'grant': 'read_execute'}},
|
|
'remove_perms': {
|
|
'Administrator': {
|
|
'grant': {'applies to': 'Not Inherited (file)',
|
|
'permissions': 'Full control'}}}},
|
|
'name': self.temp_file,
|
|
'result': True}
|
|
ret = win_file.check_perms(
|
|
path=self.temp_file,
|
|
grant_perms={'Users': {'perms': 'read_execute'},
|
|
'Administrators': {'perms': 'full_control'}},
|
|
inheritance=False,
|
|
reset=True)
|
|
self.assertDictEqual(expected, ret)
|
|
|
|
def test_stat(self):
|
|
with patch('os.path.exists', MagicMock(return_value=True)), \
|
|
patch('salt.modules.win_file._resolve_symlink', MagicMock(side_effect=lambda path: path)), \
|
|
patch('salt.modules.win_file.get_uid', MagicMock(return_value=1)), \
|
|
patch('salt.modules.win_file.uid_to_user', MagicMock(return_value='dummy')), \
|
|
patch('salt.modules.win_file.get_pgid', MagicMock(return_value=1)), \
|
|
patch('salt.modules.win_file.gid_to_group', MagicMock(return_value='dummy')), \
|
|
patch('os.stat', MagicMock(return_value=DummyStat())):
|
|
ret = win_file.stats('dummy', None, True)
|
|
self.assertEqual(ret['mode'], '0644')
|
|
self.assertEqual(ret['type'], 'file')
|