mirror of https://github.com/ansible/ansible.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
# Copyright (c) 2017 Ansible Project
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
|
|
import os
|
|
import json
|
|
import shutil
|
|
import tempfile
|
|
|
|
import pytest
|
|
|
|
from mock import patch, MagicMock
|
|
from ansible.modules import async_wrapper
|
|
|
|
from pprint import pprint
|
|
|
|
|
|
class TestAsyncWrapper:
|
|
|
|
def test_run_module(self, monkeypatch):
|
|
|
|
def mock_get_interpreter(module_path):
|
|
return ['/usr/bin/python']
|
|
|
|
module_result = {'rc': 0}
|
|
module_lines = [
|
|
'#!/usr/bin/python',
|
|
'import sys',
|
|
'sys.stderr.write("stderr stuff")',
|
|
"print('%s')" % json.dumps(module_result)
|
|
]
|
|
module_data = '\n'.join(module_lines) + '\n'
|
|
module_data = module_data.encode('utf-8')
|
|
|
|
workdir = tempfile.mkdtemp()
|
|
fh, fn = tempfile.mkstemp(dir=workdir)
|
|
|
|
with open(fn, 'wb') as f:
|
|
f.write(module_data)
|
|
|
|
command = fn
|
|
jobid = 0
|
|
job_path = os.path.join(os.path.dirname(command), 'job')
|
|
|
|
monkeypatch.setattr(async_wrapper, '_get_interpreter', mock_get_interpreter)
|
|
monkeypatch.setattr(async_wrapper, 'job_path', job_path)
|
|
|
|
res = async_wrapper._run_module(command, jobid)
|
|
|
|
with open(os.path.join(workdir, 'job'), 'r') as f:
|
|
jres = json.loads(f.read())
|
|
|
|
shutil.rmtree(workdir)
|
|
|
|
assert jres.get('rc') == 0
|
|
assert jres.get('stderr') == 'stderr stuff'
|