valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-06 16:45:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #48253 from steveno/add_file_grep_unit_tests

Browse Source 
Add unit tests for file.grep
This commit is contained in:
Nicole Thomas 2018-06-25 10:31:08 -04:00 committed by GitHub
commit 1cee5ec014
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 82 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
tests/unit/modules/test_file.py
View File

@ -28,7 +28,7 @@ import salt.utils.stringutils
import salt.modules.file as filemod
import salt.modules.config as configmod
import salt.modules.cmdmod as cmdmod
from salt.exceptions import CommandExecutionError
from salt.exceptions import CommandExecutionError, SaltInvocationError
from salt.utils.jinja import SaltCacheLoader
SED_CONTENT = '''test
@ -599,6 +599,87 @@ class FileBlockReplaceTestCase(TestCase, LoaderModuleMockMixin):
)
class FileGrepTestCase(TestCase, LoaderModuleMockMixin):
def setup_loader_modules(self):
return {
filemod: {
'__salt__': {
'config.manage_mode': configmod.manage_mode,
'cmd.run': cmdmod.run,
'cmd.run_all': cmdmod.run_all
},
'__opts__': {
'test': False,
'file_roots': {'base': 'tmp'},
'pillar_roots': {'base': 'tmp'},
'cachedir': 'tmp',
'grains': {},
},
'__grains__': {'kernel': 'Linux'},
'__utils__': {'files.is_text': MagicMock(return_value=True)},
}
}
MULTILINE_STRING = textwrap.dedent('''\
Lorem ipsum dolor sit amet, consectetur
adipiscing elit. Nam rhoncus enim ac
bibendum vulputate.
''')
MULTILINE_STRING = os.linesep.join(MULTILINE_STRING.splitlines())
def setUp(self):
self.tfile = tempfile.NamedTemporaryFile(delete=False, mode='w+')
self.tfile.write(self.MULTILINE_STRING)
self.tfile.close()
def tearDown(self):
os.remove(self.tfile.name)
del self.tfile
def test_grep_query_exists(self):
result = filemod.grep(self.tfile.name,
'Lorem ipsum')
self.assertTrue(result, None)
self.assertTrue(result['retcode'] == 0)
self.assertTrue(result['stdout'] == 'Lorem ipsum dolor sit amet, consectetur')
self.assertTrue(result['stderr'] == '')
def test_grep_query_not_exists(self):
result = filemod.grep(self.tfile.name,
'Lorem Lorem')
self.assertTrue(result['retcode'] == 1)
self.assertTrue(result['stdout'] == '')
self.assertTrue(result['stderr'] == '')
def test_grep_query_exists_with_opt(self):
result = filemod.grep(self.tfile.name,
'Lorem ipsum',
'-i')
self.assertTrue(result, None)
self.assertTrue(result['retcode'] == 0)
self.assertTrue(result['stdout'] == 'Lorem ipsum dolor sit amet, consectetur')
self.assertTrue(result['stderr'] == '')
def test_grep_query_not_exists_opt(self):
result = filemod.grep(self.tfile.name,
'Lorem Lorem',
'-v')
self.assertTrue(result['retcode'] == 0)
self.assertTrue(result['stdout'] == FileGrepTestCase.MULTILINE_STRING)
self.assertTrue(result['stderr'] == '')
def test_grep_query_too_many_opts(self):
with self.assertRaisesRegex(SaltInvocationError, '^Passing multiple command line arg') as cm:
result = filemod.grep(self.tfile.name,
'Lorem Lorem',
'-i -b2')
class FileModuleTestCase(TestCase, LoaderModuleMockMixin):
def setup_loader_modules(self):
return {