def test_run_array(self): """Test transactional_update.run with command as array""" salt_mock = { "cmd.run_all": MagicMock(return_value={"stdout": "output", "retcode": 0}) } with patch.dict(tu.__salt__, salt_mock): assert tu.run(["cmd", "--flag", "p1", "p2"]) == "output" salt_mock["cmd.run_all"].assert_called_with( [ "transactional-update", "--non-interactive", "--drop-if-no-change", "--no-selfupdate", "--quiet", "run", "cmd", "--flag", "p1", "p2", ] )
def test_run_error(): """Test transactional_update.run with missing command""" with pytest.raises(CommandExecutionError): tu.run(None)