def test_run_chroot_mount_bind(self):
'''
Test cmdmod.run_chroot mount / umount balance with bind mount
'''
mock_mount = MagicMock()
mock_umount = MagicMock()
mock_run_all = MagicMock()
with patch.dict(cmdmod.__salt__, {
'mount.mount': mock_mount,
'mount.umount': mock_umount
}):
with patch('salt.modules.cmdmod.run_all', mock_run_all):
cmdmod.run_chroot('/mnt', 'cmd', binds=['/var'])
self.assertEqual(mock_mount.call_count, 4)
self.assertEqual(mock_umount.call_count, 4)
def test_run_chroot_mount():
"""
Test cmdmod.run_chroot mount / umount balance
"""
mock_mount = MagicMock()
mock_umount = MagicMock()
mock_run_all = MagicMock()
with patch.dict(cmdmod.__salt__, {
"mount.mount": mock_mount,
"mount.umount": mock_umount
}):
with patch("salt.modules.cmdmod.run_all", mock_run_all):
cmdmod.run_chroot("/mnt", "cmd")
assert mock_mount.call_count == 3
assert mock_umount.call_count == 3
def test_run_chroot_mount_bind(self):
"""
Test cmdmod.run_chroot mount / umount balance with bind mount
"""
mock_mount = MagicMock()
mock_umount = MagicMock()
mock_run_all = MagicMock()
with patch.dict(cmdmod.__salt__, {
"mount.mount": mock_mount,
"mount.umount": mock_umount
}):
with patch("salt.modules.cmdmod.run_all", mock_run_all):
cmdmod.run_chroot("/mnt", "cmd", binds=["/var"])
self.assertEqual(mock_mount.call_count, 4)
self.assertEqual(mock_umount.call_count, 4)
def test_run_chroot_runas(self):
"""
Test run_chroot when a runas parameter is provided
"""
with patch.dict(cmdmod.__salt__, {
"mount.mount": MagicMock(),
"mount.umount": MagicMock()
}):
with patch("salt.modules.cmdmod.run_all") as run_all_mock:
cmdmod.run_chroot("/mnt",
"ls",
runas="foobar",
shell="/bin/sh")
run_all_mock.assert_called_with(
"chroot --userspec foobar: /mnt /bin/sh -c ls",
bg=False,
clean_env=False,
cwd=None,
env=None,
ignore_retcode=False,
log_callback=None,
output_encoding=None,
output_loglevel="quiet",
pillar=None,
pillarenv=None,
python_shell=True,
reset_system_locale=True,
rstrip=True,
saltenv="base",
shell="/bin/sh",
stdin=None,
success_retcodes=None,
template=None,
timeout=None,
umask=None,
use_vt=False,
)
