def test_cmd_zip(self):
mock = MagicMock(return_value='salt')
with patch.dict(archive.__salt__, {'cmd.run': mock}):
ret = archive.cmd_zip(
'/tmp/salt.{{grains.id}}.zip',
'/tmp/tmpePe8yO,/tmp/tmpLeSw1A',
template='jinja'
)
self.assertEqual(['salt'], ret)
mock.assert_called_once_with(
['zip', '-r', '/tmp/salt.{{grains.id}}.zip',
'/tmp/tmpePe8yO', '/tmp/tmpLeSw1A'],
runas=None, python_shell=False, template='jinja', cwd=None
)
mock = MagicMock(return_value='salt')
with patch.dict(archive.__salt__, {'cmd.run': mock}):
ret = archive.cmd_zip(
'/tmp/salt.{{grains.id}}.zip',
['/tmp/tmpePe8yO', '/tmp/tmpLeSw1A'],
template='jinja'
)
self.assertEqual(['salt'], ret)
mock.assert_called_once_with(
['zip', '-r', '/tmp/salt.{{grains.id}}.zip',
'/tmp/tmpePe8yO', '/tmp/tmpLeSw1A'],
runas=None, python_shell=False, template='jinja', cwd=None
)
def test_cmd_zip(self):
mock = MagicMock(return_value='salt')
with patch.dict(archive.__salt__, {'cmd.run': mock}):
ret = archive.cmd_zip(
'/tmp/salt.{{grains.id}}.zip',
'/tmp/tmpePe8yO,/tmp/tmpLeSw1A',
template='jinja'
)
self.assertEqual(['salt'], ret)
mock.assert_called_once_with(
['zip', '-r', '/tmp/salt.{{grains.id}}.zip',
'/tmp/tmpePe8yO', '/tmp/tmpLeSw1A'],
runas=None, python_shell=False, template='jinja', cwd=None
)
mock = MagicMock(return_value='salt')
with patch.dict(archive.__salt__, {'cmd.run': mock}):
ret = archive.cmd_zip(
'/tmp/salt.{{grains.id}}.zip',
['/tmp/tmpePe8yO', '/tmp/tmpLeSw1A'],
template='jinja'
)
self.assertEqual(['salt'], ret)
mock.assert_called_once_with(
['zip', '-r', '/tmp/salt.{{grains.id}}.zip',
'/tmp/tmpePe8yO', '/tmp/tmpLeSw1A'],
runas=None, python_shell=False, template='jinja', cwd=None
)
def test_zip_raises_exception_if_not_found():
mock = MagicMock(return_value="salt")
with patch.dict(archive.__salt__, {"cmd.run": mock}):
with patch("salt.utils.path.which", lambda exe: None):
with pytest.raises(CommandNotFoundError):
archive.cmd_zip(
"/tmp/salt.{{grains.id}}.zip",
"/tmp/tmpePe8yO,/tmp/tmpLeSw1A",
template="jinja",
)
assert not mock.called
def test_cmd_zip():
with patch("glob.glob", lambda pathname: [pathname]):
with patch("salt.utils.path.which", lambda exe: exe):
mock = MagicMock(return_value="salt")
with patch.dict(archive.__salt__, {"cmd.run": mock}):
ret = archive.cmd_zip(
"/tmp/salt.{{grains.id}}.zip",
"/tmp/tmpePe8yO,/tmp/tmpLeSw1A",
template="jinja",
)
assert ["salt"] == ret
mock.assert_called_once_with(
[
"zip",
"-r",
"/tmp/salt.{{grains.id}}.zip",
"/tmp/tmpePe8yO",
"/tmp/tmpLeSw1A",
],
runas=None,
python_shell=False,
template="jinja",
cwd=None,
)
mock = MagicMock(return_value="salt")
with patch.dict(archive.__salt__, {"cmd.run": mock}):
ret = archive.cmd_zip(
"/tmp/salt.{{grains.id}}.zip",
["/tmp/tmpePe8yO", "/tmp/tmpLeSw1A"],
template="jinja",
)
assert ["salt"] == ret
mock.assert_called_once_with(
[
"zip",
"-r",
"/tmp/salt.{{grains.id}}.zip",
"/tmp/tmpePe8yO",
"/tmp/tmpLeSw1A",
],
runas=None,
python_shell=False,
template="jinja",
cwd=None,
)
def test_cmd_zip(unicode_filename):
"""
Validate using the cmd_zip function
"""
with Archive("zip", unicode_filename=unicode_filename) as arch:
ret = archive.cmd_zip(str(arch.archive), str(arch.src))
assert isinstance(ret, list)
arch.assert_artifacts_in_ret(ret)
