mirror of
https://github.com/valitydev/salt.git
synced 2024-11-08 01:18:58 +00:00
116 lines
4.0 KiB
Python
116 lines
4.0 KiB
Python
# Import python libs
|
|
import getpass
|
|
import grp
|
|
import os
|
|
import shutil
|
|
import sys
|
|
|
|
# Import salt libs
|
|
from salt.exceptions import CommandExecutionError
|
|
|
|
from saltunittest import skipIf
|
|
import integration
|
|
|
|
|
|
class FileModuleTest(integration.ModuleCase):
|
|
'''
|
|
Validate the file module
|
|
'''
|
|
def setUp(self):
|
|
self.myfile = os.path.join(integration.TMP, 'myfile')
|
|
with open(self.myfile, 'w+') as fp:
|
|
fp.write("Hello\n")
|
|
self.mydir = os.path.join(integration.TMP, 'mydir/isawesome')
|
|
os.makedirs(self.mydir)
|
|
super(FileModuleTest, self).setUp()
|
|
|
|
def tearDown(self):
|
|
os.remove(self.myfile)
|
|
shutil.rmtree(self.mydir, ignore_errors=True)
|
|
super(FileModuleTest, self).tearDown()
|
|
|
|
@skipIf(sys.platform.startswith('win'), 'No chgrp on Windows')
|
|
def test_chown(self):
|
|
user = getpass.getuser()
|
|
if sys.platform == 'darwin':
|
|
group = 'staff'
|
|
elif sys.platform.startswith('linux'):
|
|
group = getpass.getuser()
|
|
ret = self.run_function('file.chown',
|
|
arg=[self.myfile, user, group])
|
|
self.assertIsNone(ret)
|
|
fstat = os.stat(self.myfile)
|
|
self.assertTrue(fstat.st_uid, os.getuid())
|
|
self.assertTrue(fstat.st_gid, grp.getgrnam(group).gr_gid)
|
|
|
|
@skipIf(sys.platform.startswith('win'), 'No chgrp on Windows')
|
|
def test_chown_no_user(self):
|
|
user = 'notanyuseriknow'
|
|
group = getpass.getuser()
|
|
ret = self.run_function('file.chown',
|
|
arg=[self.myfile, user, group])
|
|
self.assertIn('not exist', ret)
|
|
|
|
@skipIf(sys.platform.startswith('win'), 'No chgrp on Windows')
|
|
def test_chown_no_user_no_group(self):
|
|
user = 'notanyuseriknow'
|
|
group = 'notanygroupyoushoulduse'
|
|
ret = self.run_function('file.chown',
|
|
arg=[self.myfile, user, group])
|
|
self.assertIn('Group does not exist', ret)
|
|
self.assertIn('User does not exist', ret)
|
|
|
|
@skipIf(sys.platform.startswith('win'), 'No chgrp on Windows')
|
|
def test_chown_no_path(self):
|
|
user = getpass.getuser()
|
|
if sys.platform == 'darwin':
|
|
group = 'staff'
|
|
elif sys.platform.startswith('linux'):
|
|
group = getpass.getuser()
|
|
ret = self.run_function('file.chown',
|
|
arg=['/tmp/nosuchfile', user, group])
|
|
self.assertIn('File not found', ret)
|
|
|
|
@skipIf(sys.platform.startswith('win'), 'No chgrp on Windows')
|
|
def test_chown_noop(self):
|
|
user = ''
|
|
group = ''
|
|
ret = self.run_function('file.chown',
|
|
arg=[self.myfile, user, group])
|
|
self.assertIsNone(ret)
|
|
fstat = os.stat(self.myfile)
|
|
self.assertTrue(fstat.st_uid, os.getuid())
|
|
self.assertTrue(fstat.st_gid, os.getgid())
|
|
|
|
@skipIf(sys.platform.startswith('win'), 'No chgrp on Windows')
|
|
def test_chgrp(self):
|
|
if sys.platform == 'darwin':
|
|
group = 'everyone'
|
|
elif sys.platform.startswith('linux'):
|
|
group = getpass.getuser()
|
|
ret = self.run_function('file.chgrp',
|
|
arg=[self.myfile, group])
|
|
self.assertIsNone(ret)
|
|
fstat = os.stat(self.myfile)
|
|
self.assertTrue(fstat.st_gid, grp.getgrnam(group).gr_gid)
|
|
|
|
@skipIf(sys.platform.startswith('win'), 'No chgrp on Windows')
|
|
def test_chgrp_failure(self):
|
|
group = 'thisgroupdoesntexist'
|
|
ret = self.run_function('file.chgrp',
|
|
arg=[self.myfile, group])
|
|
self.assertIn('not exist', ret)
|
|
|
|
def test_remove_file(self):
|
|
ret = self.run_function('file.remove', args=[self.myfile])
|
|
self.assertTrue(ret)
|
|
|
|
def test_remove_dir(self):
|
|
ret = self.run_function('file.remove', args=[self.mydir])
|
|
self.assertTrue(ret)
|
|
|
|
def test_cannot_remove(self):
|
|
ret = self.run_function('file.remove', args=['/dev/tty'])
|
|
self.assertEqual(
|
|
'ERROR executing file.remove: File path must be absolute.', ret)
|