Porting tests to pytest (#33387)

* Porting tests to pytest

* Achievement Get: No longer need mock/generator.py
  * Now done via pytest's parametrization
  * Port safe_eval to pytest
  * Port text tests to pytest
  * Port test_set_mode_if_different to pytest

* Change conftest AnsibleModule fixtures to be more flexible
  * Move the AnsibleModules fixtures to module_utils/conftest.py for sharing
  * Testing the argspec code requires:
    * injecting both the argspec and the arguments.
    * Patching the arguments into sys.stdin at a different level

* More porting to obsolete mock/procenv.py
  * Port run_command to pytest
  * Port known_hosts tests to pytest
  * Port safe_eval to pytest
  * Port test_distribution_version.py to pytest
  * Port test_log to pytest
  * Port test__log_invocation to pytest
  * Remove unneeded import of procenv in test_postgresql

* Port test_pip to pytest style
  * As part of this, create a pytest ansiblemodule fixture in
    modules/conftest.py.  This is slightly different than the
    approach taken in module_utils because here we need to override the
    AnsibleModule that the modules will inherit from instead of one that
    we're instantiating ourselves.

* Fixup usage of parametrization in test_deprecate_warn

* Check that the pip module failed in our test
pull/33698/head
Toshio Kuratomi 7 years ago committed by GitHub
parent ed376abe42
commit cd36164239
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -1,72 +0,0 @@
# Copyright 2016 Toshio Kuratomi <tkuratomi@ansible.com>
#
# This file is part of Ansible
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# Make coding more python3-ish
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
from collections import Mapping
def make_method(func, args, kwargs):
def test_method(self):
func(self, *args, **kwargs)
# Format the argument string
arg_string = ', '.join(repr(a) for a in args)
kwarg_string = ', '.join('{0}={1}'.format(item[0], repr(item[1])) for item in kwargs.items())
arg_list = []
if arg_string:
arg_list.append(arg_string)
if kwarg_string:
arg_list.append(kwarg_string)
test_method.__name__ = 'test_{0}({1})'.format(func.__name__, ', '.join(arg_list))
return test_method
def add_method(func, *combined_args):
"""
Add a test case via a class decorator.
nose uses generators for this but doesn't work with unittest.TestCase
subclasses. So we have to write our own.
The first argument to this decorator is a test function. All subsequent
arguments are the arguments to create each generated test function with in
the following format:
Each set of arguments is a two-tuple. The first element is an iterable of
positional arguments. the second is a dict representing the kwargs.
"""
def wrapper(cls):
for combined_arg in combined_args:
if len(combined_arg) == 2:
args = combined_arg[0]
kwargs = combined_arg[1]
elif isinstance(combined_arg[0], Mapping):
args = []
kwargs = combined_arg[0]
else:
args = combined_arg[0]
kwargs = {}
test_method = make_method(func, args, kwargs)
setattr(cls, test_method.__name__, test_method)
return cls
return wrapper

@ -1,78 +1,50 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# (c) 2016, James Cammarata <jimi@sngx.net> # (c) 2016, James Cammarata <jimi@sngx.net>
# # (c) 2017, Ansible Project
# This file is part of Ansible # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# Make coding more python3-ish from __future__ import absolute_import, division, print_function
from __future__ import (absolute_import, division)
__metaclass__ = type __metaclass__ = type
import json import pytest
import sys
from units.mock.procenv import swap_stdin_and_argv
from ansible.compat.tests import unittest ARGS = dict(foo=False, bar=[1, 2, 3], bam="bam", baz=u'baz')
from ansible.compat.tests.mock import MagicMock ARGUMENT_SPEC = dict(
class TestModuleUtilsBasic(unittest.TestCase):
def test_module_utils_basic__log_invocation(self):
with swap_stdin_and_argv(stdin_data=json.dumps(dict(
ANSIBLE_MODULE_ARGS=dict(foo=False, bar=[1, 2, 3], bam="bam", baz=u'baz')),
)):
from ansible.module_utils import basic
# test basic log invocation
basic._ANSIBLE_ARGS = None
am = basic.AnsibleModule(
argument_spec=dict(
foo=dict(default=True, type='bool'), foo=dict(default=True, type='bool'),
bar=dict(default=[], type='list'), bar=dict(default=[], type='list'),
bam=dict(default="bam"), bam=dict(default="bam"),
baz=dict(default=u"baz"), baz=dict(default=u"baz"),
password=dict(default=True), password=dict(default=True),
no_log=dict(default="you shouldn't see me", no_log=True), no_log=dict(default="you shouldn't see me", no_log=True),
),
) )
am.log = MagicMock()
@pytest.mark.parametrize('am, stdin', [(ARGUMENT_SPEC, ARGS)], indirect=['am', 'stdin'])
def test_module_utils_basic__log_invocation(am, mocker):
am.log = mocker.MagicMock()
am._log_invocation() am._log_invocation()
# Message is generated from a dict so it will be in an unknown order. # Message is generated from a dict so it will be in an unknown order.
# have to check this manually rather than with assert_called_with() # have to check this manually rather than with assert_called_with()
args = am.log.call_args[0] args = am.log.call_args[0]
self.assertEqual(len(args), 1) assert len(args) == 1
message = args[0] message = args[0]
self.assertEqual( assert len(message) == \
len(message),
len('Invoked with bam=bam bar=[1, 2, 3] foo=False baz=baz no_log=NOT_LOGGING_PARAMETER password=NOT_LOGGING_PASSWORD') len('Invoked with bam=bam bar=[1, 2, 3] foo=False baz=baz no_log=NOT_LOGGING_PARAMETER password=NOT_LOGGING_PASSWORD')
)
self.assertTrue(message.startswith('Invoked with ')) assert message.startswith('Invoked with ')
self.assertIn(' bam=bam', message) assert ' bam=bam' in message
self.assertIn(' bar=[1, 2, 3]', message) assert ' bar=[1, 2, 3]' in message
self.assertIn(' foo=False', message) assert ' foo=False' in message
self.assertIn(' baz=baz', message) assert ' baz=baz' in message
self.assertIn(' no_log=NOT_LOGGING_PARAMETER', message) assert ' no_log=NOT_LOGGING_PARAMETER' in message
self.assertIn(' password=NOT_LOGGING_PASSWORD', message) assert ' password=NOT_LOGGING_PASSWORD' in message
kwargs = am.log.call_args[1] kwargs = am.log.call_args[1]
self.assertEqual( assert kwargs == \
kwargs,
dict(log_args={ dict(log_args={
'foo': 'False', 'foo': 'False',
'bar': '[1, 2, 3]', 'bar': '[1, 2, 3]',
@ -81,4 +53,3 @@ class TestModuleUtilsBasic(unittest.TestCase):
'password': 'NOT_LOGGING_PASSWORD', 'password': 'NOT_LOGGING_PASSWORD',
'no_log': 'NOT_LOGGING_PARAMETER', 'no_log': 'NOT_LOGGING_PARAMETER',
}) })
)

@ -6,44 +6,46 @@ __metaclass__ = type
import json import json
from ansible.compat.tests import unittest import pytest
from ansible.compat.tests.mock import MagicMock from ansible.compat.tests.mock import MagicMock
from units.mock.procenv import swap_stdin_and_argv, swap_stdout
from ansible.module_utils import basic from ansible.module_utils import basic
class TestCallableTypeValidation(unittest.TestCase): MOCK_VALIDATOR_SUCCESS = MagicMock(return_value=42)
def setUp(self): MOCK_VALIDATOR_FAIL = MagicMock(side_effect=TypeError("bad conversion"))
args = json.dumps(dict(ANSIBLE_MODULE_ARGS=dict(arg="42"))) # Data is argspec, argument, expected
self.stdin_swap_ctx = swap_stdin_and_argv(stdin_data=args) VALID_SPECS = (
self.stdin_swap_ctx.__enter__() ({'arg': {'type': int}}, {'arg': 42}, 42),
({'arg': {'type': int}}, {'arg': '42'}, 42),
({'arg': {'type': MOCK_VALIDATOR_SUCCESS}}, {'arg': 42}, 42),
)
# since we can't use context managers and "with" without overriding run(), call them directly INVALID_SPECS = (
self.stdout_swap_ctx = swap_stdout() ({'arg': {'type': int}}, {'arg': "bad"}, "invalid literal for int() with base 10: 'bad'"),
self.fake_stream = self.stdout_swap_ctx.__enter__() ({'arg': {'type': MOCK_VALIDATOR_FAIL}}, {'arg': "bad"}, "bad conversion"),
)
basic._ANSIBLE_ARGS = None
def tearDown(self): @pytest.mark.parametrize('argspec, expected, am, stdin', [(s[0], s[2], s[0], s[1]) for s in VALID_SPECS],
# since we can't use context managers and "with" without overriding run(), call them directly to clean up indirect=['am', 'stdin'])
self.stdin_swap_ctx.__exit__(None, None, None) def test_validator_success(am, mocker, argspec, expected):
self.stdout_swap_ctx.__exit__(None, None, None)
def test_validate_success(self): type_ = argspec['arg']['type']
mock_validator = MagicMock(return_value=42) if isinstance(type_, MagicMock):
m = basic.AnsibleModule(argument_spec=dict( assert type_.called
arg=dict(type=mock_validator) else:
)) assert isinstance(am.params['arg'], type_)
assert am.params['arg'] == expected
self.assertTrue(mock_validator.called)
self.assertEqual(m.params['arg'], 42)
self.assertEqual(type(m.params['arg']), int)
def test_validate_fail(self): @pytest.mark.parametrize('argspec, expected, stdin', [(s[0], s[2], s[1]) for s in INVALID_SPECS],
mock_validator = MagicMock(side_effect=TypeError("bad conversion")) indirect=['stdin'])
with self.assertRaises(SystemExit) as ecm: def test_validator_fail(stdin, capfd, argspec, expected):
m = basic.AnsibleModule(argument_spec=dict( with pytest.raises(SystemExit) as ecm:
arg=dict(type=mock_validator) m = basic.AnsibleModule(argument_spec=argspec)
))
self.assertIn("bad conversion", json.loads(self.fake_stream.getvalue())['msg']) out, err = capfd.readouterr()
assert not err
assert expected in json.loads(out)['msg']
assert json.loads(out)['failed']

@ -1,19 +1,7 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# #
# This file is part of Ansible # Copyright (c) 2017 Ansible Project
# # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
import json import json
import sys import sys
@ -27,49 +15,7 @@ from ansible.module_utils.six import PY3, string_types
from ansible.module_utils._text import to_bytes from ansible.module_utils._text import to_bytes
@pytest.fixture @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
def stdin(mocker, request):
if isinstance(request.param, string_types):
args = request.param
elif isinstance(request.param, MutableMapping):
if 'ANSIBLE_MODULE_ARGS' not in request.param:
request.param = {'ANSIBLE_MODULE_ARGS': request.param}
args = json.dumps(request.param)
else:
raise Exception('Malformed data to the stdin pytest fixture')
real_stdin = sys.stdin
fake_stdin = BytesIO(to_bytes(args, errors='surrogate_or_strict'))
if PY3:
sys.stdin = mocker.MagicMock()
sys.stdin.buffer = fake_stdin
else:
sys.stdin = fake_stdin
yield fake_stdin
sys.stdin = real_stdin
@pytest.fixture
def am(stdin, request):
old_args = ansible.module_utils.basic._ANSIBLE_ARGS
ansible.module_utils.basic._ANSIBLE_ARGS = None
old_argv = sys.argv
sys.argv = ['ansible_unittest']
am = ansible.module_utils.basic.AnsibleModule(
argument_spec=dict(),
)
am._name = 'ansible_unittest'
yield am
ansible.module_utils.basic._ANSIBLE_ARGS = old_args
sys.argv = old_argv
@pytest.mark.parametrize('stdin', [{}], indirect=True)
def test_warn(am, capfd): def test_warn(am, capfd):
am.warn('warning1') am.warn('warning1')
@ -80,7 +26,7 @@ def test_warn(am, capfd):
assert json.loads(out)['warnings'] == ['warning1', 'warning2'] assert json.loads(out)['warnings'] == ['warning1', 'warning2']
@pytest.mark.parametrize('stdin', [{}], indirect=True) @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
def test_deprecate(am, capfd): def test_deprecate(am, capfd):
am.deprecate('deprecation1') am.deprecate('deprecation1')
am.deprecate('deprecation2', '2.3') am.deprecate('deprecation2', '2.3')
@ -99,7 +45,7 @@ def test_deprecate(am, capfd):
] ]
@pytest.mark.parametrize('stdin', [{}], indirect=True) @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
def test_deprecate_without_list(am, capfd): def test_deprecate_without_list(am, capfd):
with pytest.raises(SystemExit): with pytest.raises(SystemExit):
am.exit_json(deprecations='Simple deprecation warning') am.exit_json(deprecations='Simple deprecation warning')

@ -1,98 +1,76 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# (c) 2015, Toshio Kuratomi <tkuratomi@ansible.com> # Copyright (c) 2015-2017 Ansible Project
# # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
# This file is part of Ansible
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# Make coding more python3-ish # Make coding more python3-ish
from __future__ import (absolute_import, division) from __future__ import (absolute_import, division)
__metaclass__ = type __metaclass__ = type
import copy
import json import json
import sys import sys
from itertools import chain
from ansible.compat.tests import unittest import pytest
from ansible.module_utils import basic
from units.mock.procenv import swap_stdin_and_argv, swap_stdout
EMPTY_INVOCATION = {u'module_args': {}}
empty_invocation = {u'module_args': {}}
class TestAnsibleModuleExitJson:
"""
class TestAnsibleModuleExitJson(unittest.TestCase): Test that various means of calling exitJson and FailJson return the messages they've been given
def setUp(self): """
args = json.dumps(dict(ANSIBLE_MODULE_ARGS={})) DATA = (
self.stdin_swap_ctx = swap_stdin_and_argv(stdin_data=args) ({}, {'invocation': EMPTY_INVOCATION}),
self.stdin_swap_ctx.__enter__() ({'msg': 'message'}, {'msg': 'message', 'invocation': EMPTY_INVOCATION}),
({'msg': 'success', 'changed': True},
# since we can't use context managers and "with" without overriding run(), call them directly {'msg': 'success', 'changed': True, 'invocation': EMPTY_INVOCATION}),
self.stdout_swap_ctx = swap_stdout() ({'msg': 'nochange', 'changed': False},
self.fake_stream = self.stdout_swap_ctx.__enter__() {'msg': 'nochange', 'changed': False, 'invocation': EMPTY_INVOCATION}),
)
basic._ANSIBLE_ARGS = None
self.module = basic.AnsibleModule(argument_spec=dict()) # pylint bug: https://github.com/PyCQA/pylint/issues/511
# pylint: disable=undefined-variable
def tearDown(self): @pytest.mark.parametrize('args, expected, stdin', ((a, e, {}) for a, e in DATA), indirect=['stdin'])
# since we can't use context managers and "with" without overriding run(), call them directly to clean up def test_exit_json_exits(self, am, capfd, args, expected):
self.stdin_swap_ctx.__exit__(None, None, None) with pytest.raises(SystemExit) as ctx:
self.stdout_swap_ctx.__exit__(None, None, None) am.exit_json(**args)
assert ctx.value.code == 0
def test_exit_json_no_args_exits(self):
with self.assertRaises(SystemExit) as ctx: out, err = capfd.readouterr()
self.module.exit_json() return_val = json.loads(out)
if isinstance(ctx.exception, int): assert return_val == expected
# Python2.6... why does sys.exit behave this way?
self.assertEquals(ctx.exception, 0) # Fail_json is only legal if it's called with a message
else: # pylint bug: https://github.com/PyCQA/pylint/issues/511
self.assertEquals(ctx.exception.code, 0) @pytest.mark.parametrize('args, expected, stdin',
return_val = json.loads(self.fake_stream.getvalue()) ((a, e, {}) for a, e in DATA if 'msg' in a), # pylint: disable=undefined-variable
self.assertEquals(return_val, dict(invocation=empty_invocation)) indirect=['stdin'])
def test_fail_json_exits(self, am, capfd, args, expected):
def test_exit_json_args_exits(self): with pytest.raises(SystemExit) as ctx:
with self.assertRaises(SystemExit) as ctx: am.fail_json(**args)
self.module.exit_json(msg='message') assert ctx.value.code == 1
if isinstance(ctx.exception, int):
# Python2.6... why does sys.exit behave this way? out, err = capfd.readouterr()
self.assertEquals(ctx.exception, 0) return_val = json.loads(out)
else: # Fail_json should add failed=True
self.assertEquals(ctx.exception.code, 0) expected['failed'] = True
return_val = json.loads(self.fake_stream.getvalue()) assert return_val == expected
self.assertEquals(return_val, dict(msg="message", invocation=empty_invocation))
@pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
def test_fail_json_exits(self): def test_fail_json_no_msg(self, am):
with self.assertRaises(SystemExit) as ctx: with pytest.raises(AssertionError) as ctx:
self.module.fail_json(msg='message') am.fail_json()
if isinstance(ctx.exception, int): assert ctx.value.args[0] == "implementation error -- msg to explain the error is required"
# Python2.6... why does sys.exit behave this way?
self.assertEquals(ctx.exception, 1)
else: class TestAnsibleModuleExitValuesRemoved:
self.assertEquals(ctx.exception.code, 1) """
return_val = json.loads(self.fake_stream.getvalue()) Test that ExitJson and FailJson remove password-like values
self.assertEquals(return_val, dict(msg="message", failed=True, invocation=empty_invocation)) """
def test_exit_json_proper_changed(self):
with self.assertRaises(SystemExit) as ctx:
self.module.exit_json(changed=True, msg='success')
return_val = json.loads(self.fake_stream.getvalue())
self.assertEquals(return_val, dict(changed=True, msg='success', invocation=empty_invocation))
class TestAnsibleModuleExitValuesRemoved(unittest.TestCase):
OMIT = 'VALUE_SPECIFIED_IN_NO_LOG_PARAMETER' OMIT = 'VALUE_SPECIFIED_IN_NO_LOG_PARAMETER'
dataset = (
DATA = (
( (
dict(username='person', password='$ecret k3y'), dict(username='person', password='$ecret k3y'),
dict(one=1, pwd='$ecret k3y', url='https://username:password12345@foo.com/login/', dict(one=1, pwd='$ecret k3y', url='https://username:password12345@foo.com/login/',
@ -119,43 +97,27 @@ class TestAnsibleModuleExitValuesRemoved(unittest.TestCase):
), ),
) )
def test_exit_json_removes_values(self): # pylint bug: https://github.com/PyCQA/pylint/issues/511
self.maxDiff = None @pytest.mark.parametrize('am, stdin, return_val, expected',
for args, return_val, expected in self.dataset: (({'username': {}, 'password': {'no_log': True}, 'token': {'no_log': True}}, s, r, e)
params = dict(ANSIBLE_MODULE_ARGS=args) for s, r, e in DATA), # pylint: disable=undefined-variable
params = json.dumps(params) indirect=['am', 'stdin'])
def test_exit_json_removes_values(self, am, capfd, return_val, expected):
with swap_stdin_and_argv(stdin_data=params): with pytest.raises(SystemExit) as ctx:
with swap_stdout(): am.exit_json(**return_val)
basic._ANSIBLE_ARGS = None out, err = capfd.readouterr()
module = basic.AnsibleModule(
argument_spec=dict( assert json.loads(out) == expected
username=dict(),
password=dict(no_log=True), # pylint bug: https://github.com/PyCQA/pylint/issues/511
token=dict(no_log=True), @pytest.mark.parametrize('am, stdin, return_val, expected',
), (({'username': {}, 'password': {'no_log': True}, 'token': {'no_log': True}}, s, r, e)
) for s, r, e in DATA), # pylint: disable=undefined-variable
with self.assertRaises(SystemExit) as ctx: indirect=['am', 'stdin'])
self.assertEquals(module.exit_json(**return_val), expected) def test_fail_json_removes_values(self, am, capfd, return_val, expected):
self.assertEquals(json.loads(sys.stdout.getvalue()), expected)
def test_fail_json_removes_values(self):
self.maxDiff = None
for args, return_val, expected in self.dataset:
expected = copy.deepcopy(expected)
expected['failed'] = True expected['failed'] = True
params = dict(ANSIBLE_MODULE_ARGS=args) with pytest.raises(SystemExit) as ctx:
params = json.dumps(params) am.fail_json(**return_val) == expected
with swap_stdin_and_argv(stdin_data=params): out, err = capfd.readouterr()
with swap_stdout():
basic._ANSIBLE_ARGS = None assert json.loads(out) == expected
module = basic.AnsibleModule(
argument_spec=dict(
username=dict(),
password=dict(no_log=True),
token=dict(no_log=True),
),
)
with self.assertRaises(SystemExit) as ctx:
self.assertEquals(module.fail_json(**return_val), expected)
self.assertEquals(json.loads(sys.stdout.getvalue()), expected)

@ -1,35 +1,20 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# (c) 2012-2014, Michael DeHaan <michael.dehaan@gmail.com> # (c) 2012-2014, Michael DeHaan <michael.dehaan@gmail.com>
# # (c) 2017, Ansible Project
# This file is part of Ansible # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# Make coding more python3-ish from __future__ import absolute_import, division, print_function
from __future__ import (absolute_import, division)
__metaclass__ = type __metaclass__ = type
import sys import sys
import json import json
import syslog import syslog
from itertools import product
from ansible.compat.tests import unittest import pytest
from ansible.compat.tests.mock import patch, MagicMock
from units.mock.procenv import swap_stdin_and_argv
import ansible.module_utils.basic import ansible.module_utils.basic
from ansible.module_utils.six import PY3
try: try:
# Python 3.4+ # Python 3.4+
@ -41,80 +26,47 @@ except ImportError:
pass pass
class TestAnsibleModuleSysLogSmokeTest(unittest.TestCase): class TestAnsibleModuleLogSmokeTest:
def setUp(self): DATA = [u'Text string', u'Toshio くらとみ non-ascii test']
args = json.dumps(dict(ANSIBLE_MODULE_ARGS={})) DATA = DATA + [d.encode('utf-8') for d in DATA]
DATA += [b'non-utf8 :\xff: test']
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually
self.stdin_swap = swap_stdin_and_argv(stdin_data=args)
self.stdin_swap.__enter__()
ansible.module_utils.basic._ANSIBLE_ARGS = None
self.am = ansible.module_utils.basic.AnsibleModule(
argument_spec=dict(),
)
self.am._name = 'unittest'
self.has_journal = ansible.module_utils.basic.has_journal
if self.has_journal:
# Systems with journal can still test syslog
ansible.module_utils.basic.has_journal = False
def tearDown(self):
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually
self.stdin_swap.__exit__(None, None, None)
ansible.module_utils.basic.has_journal = self.has_journal
def test_smoketest_syslog(self): # pylint bug: https://github.com/PyCQA/pylint/issues/511
@pytest.mark.parametrize('msg, stdin', ((m, {}) for m in DATA), indirect=['stdin']) # pylint: disable=undefined-variable
def test_smoketest_syslog(self, am, mocker, msg):
# These talk to the live daemons on the system. Need to do this to # These talk to the live daemons on the system. Need to do this to
# show that what we send doesn't cause an issue once it gets to the # show that what we send doesn't cause an issue once it gets to the
# daemon. These are just smoketests to test that we don't fail. # daemon. These are just smoketests to test that we don't fail.
mocker.patch('ansible.module_utils.basic.has_journal', False)
self.am.log(u'Text string') am.log(u'Text string')
self.am.log(u'Toshio くらとみ non-ascii test') am.log(u'Toshio くらとみ non-ascii test')
self.am.log(b'Byte string')
self.am.log(u'Toshio くらとみ non-ascii test'.encode('utf-8'))
self.am.log(b'non-utf8 :\xff: test')
class TestAnsibleModuleJournaldSmokeTest(unittest.TestCase):
def setUp(self):
args = json.dumps(dict(ANSIBLE_MODULE_ARGS={}))
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually
self.stdin_swap = swap_stdin_and_argv(stdin_data=args)
self.stdin_swap.__enter__()
ansible.module_utils.basic._ANSIBLE_ARGS = None am.log(b'Byte string')
self.am = ansible.module_utils.basic.AnsibleModule( am.log(u'Toshio くらとみ non-ascii test'.encode('utf-8'))
argument_spec=dict(), am.log(b'non-utf8 :\xff: test')
)
self.am._name = 'unittest'
def tearDown(self): @pytest.mark.skipif(not ansible.module_utils.basic.has_journal, reason='python systemd bindings not installed')
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually # pylint bug: https://github.com/PyCQA/pylint/issues/511
self.stdin_swap.__exit__(None, None, None) @pytest.mark.parametrize('msg, stdin', ((m, {}) for m in DATA), indirect=['stdin']) # pylint: disable=undefined-variable
def test_smoketest_journal(self, am, mocker, msg):
@unittest.skipUnless(ansible.module_utils.basic.has_journal, 'python systemd bindings not installed')
def test_smoketest_journal(self):
# These talk to the live daemons on the system. Need to do this to # These talk to the live daemons on the system. Need to do this to
# show that what we send doesn't cause an issue once it gets to the # show that what we send doesn't cause an issue once it gets to the
# daemon. These are just smoketests to test that we don't fail. # daemon. These are just smoketests to test that we don't fail.
mocker.patch('ansible.module_utils.basic.has_journal', True)
self.am.log(u'Text string') am.log(u'Text string')
self.am.log(u'Toshio くらとみ non-ascii test') am.log(u'Toshio くらとみ non-ascii test')
self.am.log(b'Byte string') am.log(b'Byte string')
self.am.log(u'Toshio くらとみ non-ascii test'.encode('utf-8')) am.log(u'Toshio くらとみ non-ascii test'.encode('utf-8'))
self.am.log(b'non-utf8 :\xff: test') am.log(b'non-utf8 :\xff: test')
class TestAnsibleModuleLogSyslog(unittest.TestCase): class TestAnsibleModuleLogSyslog:
"""Test the AnsibleModule Log Method""" """Test the AnsibleModule Log Method"""
py2_output_data = { PY2_OUTPUT_DATA = {
u'Text string': b'Text string', u'Text string': b'Text string',
u'Toshio くらとみ non-ascii test': u'Toshio くらとみ non-ascii test'.encode('utf-8'), u'Toshio くらとみ non-ascii test': u'Toshio くらとみ non-ascii test'.encode('utf-8'),
b'Byte string': b'Byte string', b'Byte string': b'Byte string',
@ -122,7 +74,7 @@ class TestAnsibleModuleLogSyslog(unittest.TestCase):
b'non-utf8 :\xff: test': b'non-utf8 :\xff: test'.decode('utf-8', 'replace').encode('utf-8'), b'non-utf8 :\xff: test': b'non-utf8 :\xff: test'.decode('utf-8', 'replace').encode('utf-8'),
} }
py3_output_data = { PY3_OUTPUT_DATA = {
u'Text string': u'Text string', u'Text string': u'Text string',
u'Toshio くらとみ non-ascii test': u'Toshio くらとみ non-ascii test', u'Toshio くらとみ non-ascii test': u'Toshio くらとみ non-ascii test',
b'Byte string': u'Byte string', b'Byte string': u'Byte string',
@ -130,60 +82,37 @@ class TestAnsibleModuleLogSyslog(unittest.TestCase):
b'non-utf8 :\xff: test': b'non-utf8 :\xff: test'.decode('utf-8', 'replace') b'non-utf8 :\xff: test': b'non-utf8 :\xff: test'.decode('utf-8', 'replace')
} }
def setUp(self): @pytest.mark.parametrize('no_log, stdin', (product((True, False), [{}])), indirect=['stdin'])
args = json.dumps(dict(ANSIBLE_MODULE_ARGS={})) def test_no_log(self, am, mocker, no_log):
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually """Test that when no_log is set, logging does not occur"""
self.stdin_swap = swap_stdin_and_argv(stdin_data=args) mock_syslog = mocker.patch('syslog.syslog', autospec=True)
self.stdin_swap.__enter__() mocker.patch('ansible.module_utils.basic.has_journal', False)
am.no_log = no_log
ansible.module_utils.basic._ANSIBLE_ARGS = None am.log('unittest no_log')
self.am = ansible.module_utils.basic.AnsibleModule( if no_log:
argument_spec=dict(), assert not mock_syslog.called
)
self.am._name = 'unittest'
self.has_journal = ansible.module_utils.basic.has_journal
if self.has_journal:
# Systems with journal can still test syslog
ansible.module_utils.basic.has_journal = False
def tearDown(self):
# teardown/reset
ansible.module_utils.basic.has_journal = self.has_journal
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually
self.stdin_swap.__exit__(None, None, None)
@patch('syslog.syslog', autospec=True)
def test_no_log(self, mock_func):
no_log = self.am.no_log
self.am.no_log = True
self.am.log('unittest no_log')
self.assertFalse(mock_func.called)
self.am.no_log = False
self.am.log('unittest no_log')
mock_func.assert_called_once_with(syslog.LOG_INFO, 'unittest no_log')
self.am.no_log = no_log
def test_output_matches(self):
if sys.version_info >= (3,):
output_data = self.py3_output_data
else: else:
output_data = self.py2_output_data mock_syslog.assert_called_once_with(syslog.LOG_INFO, 'unittest no_log')
# pylint bug: https://github.com/PyCQA/pylint/issues/511
@pytest.mark.parametrize('msg, param, stdin',
((m, p, {}) for m, p in
(PY3_OUTPUT_DATA.items() if PY3 else PY2_OUTPUT_DATA.items())), # pylint: disable=undefined-variable
indirect=['stdin'])
def test_output_matches(self, am, mocker, msg, param):
"""Check that log messages are sent correctly"""
mocker.patch('ansible.module_utils.basic.has_journal', False)
mock_syslog = mocker.patch('syslog.syslog', autospec=True)
for msg, param in output_data.items(): am.log(msg)
with patch('syslog.syslog', autospec=True) as mock_func: mock_syslog.assert_called_once_with(syslog.LOG_INFO, param)
self.am.log(msg)
mock_func.assert_called_once_with(syslog.LOG_INFO, param)
class TestAnsibleModuleLogJournal(unittest.TestCase): @pytest.mark.skipif(not ansible.module_utils.basic.has_journal, reason='python systemd bindings not installed')
class TestAnsibleModuleLogJournal:
"""Test the AnsibleModule Log Method""" """Test the AnsibleModule Log Method"""
output_data = { OUTPUT_DATA = {
u'Text string': u'Text string', u'Text string': u'Text string',
u'Toshio くらとみ non-ascii test': u'Toshio くらとみ non-ascii test', u'Toshio くらとみ non-ascii test': u'Toshio くらとみ non-ascii test',
b'Byte string': u'Byte string', b'Byte string': u'Byte string',
@ -191,82 +120,43 @@ class TestAnsibleModuleLogJournal(unittest.TestCase):
b'non-utf8 :\xff: test': b'non-utf8 :\xff: test'.decode('utf-8', 'replace') b'non-utf8 :\xff: test': b'non-utf8 :\xff: test'.decode('utf-8', 'replace')
} }
# overriding run lets us use context managers for setup/teardown-esque behavior @pytest.mark.parametrize('no_log, stdin', (product((True, False), [{}])), indirect=['stdin'])
def setUp(self): def test_no_log(self, am, mocker, no_log):
args = json.dumps(dict(ANSIBLE_MODULE_ARGS={})) journal_send = mocker.patch('systemd.journal.send')
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually am.no_log = no_log
self.stdin_swap = swap_stdin_and_argv(stdin_data=args) am.log('unittest no_log')
self.stdin_swap.__enter__() if no_log:
assert not journal_send.called
ansible.module_utils.basic._ANSIBLE_ARGS = None else:
self.am = ansible.module_utils.basic.AnsibleModule( assert journal_send.called == 1
argument_spec=dict(),
)
self.am._name = 'unittest'
self.has_journal = ansible.module_utils.basic.has_journal
ansible.module_utils.basic.has_journal = True
self.module_patcher = None
# In case systemd-python is not installed
if not self.has_journal:
self.module_patcher = patch.dict('sys.modules', {'systemd': MagicMock(), 'systemd.journal': MagicMock()})
self.module_patcher.start()
try:
reload(ansible.module_utils.basic)
except NameError:
self._fake_out_reload(ansible.module_utils.basic)
def tearDown(self):
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually
self.stdin_swap.__exit__(None, None, None)
# teardown/reset
ansible.module_utils.basic.has_journal = self.has_journal
if self.module_patcher:
self.module_patcher.stop()
reload(ansible.module_utils.basic)
@patch('systemd.journal.send')
def test_no_log(self, mock_func):
no_log = self.am.no_log
self.am.no_log = True
self.am.log('unittest no_log')
self.assertFalse(mock_func.called)
self.am.no_log = False
self.am.log('unittest no_log')
self.assertEqual(mock_func.called, 1)
# Message # Message
# call_args is a 2-tuple of (arg_list, kwarg_dict) # call_args is a 2-tuple of (arg_list, kwarg_dict)
self.assertTrue(mock_func.call_args[0][0].endswith('unittest no_log'), msg='Message was not sent to log') assert journal_send.call_args[0][0].endswith('unittest no_log'), 'Message was not sent to log'
# log adds this journal field # log adds this journal field
self.assertIn('MODULE', mock_func.call_args[1]) assert 'MODULE' in journal_send.call_args[1]
self.assertIn('basic.py', mock_func.call_args[1]['MODULE']) assert 'basic.py' in journal_send.call_args[1]['MODULE']
self.am.no_log = no_log # pylint bug: https://github.com/PyCQA/pylint/issues/511
@pytest.mark.parametrize('msg, param, stdin',
def test_output_matches(self): ((m, p, {}) for m, p in OUTPUT_DATA.items()), # pylint: disable=undefined-variable
for msg, param in self.output_data.items(): indirect=['stdin'])
with patch('systemd.journal.send', autospec=True) as mock_func: def test_output_matches(self, am, mocker, msg, param):
self.am.log(msg) journal_send = mocker.patch('systemd.journal.send')
self.assertEqual(mock_func.call_count, 1, msg='journal.send not called exactly once') am.log(msg)
self.assertTrue(mock_func.call_args[0][0].endswith(param)) assert journal_send.call_count == 1, 'journal.send not called exactly once'
assert journal_send.call_args[0][0].endswith(param)
@patch('systemd.journal.send')
def test_log_args(self, mock_func): @pytest.mark.parametrize('stdin', ({},), indirect=['stdin'])
self.am.log('unittest log_args', log_args=dict(TEST='log unittest')) def test_log_args(self, am, mocker):
self.assertEqual(mock_func.called, 1) journal_send = mocker.patch('systemd.journal.send')
self.assertTrue(mock_func.call_args[0][0].endswith('unittest log_args'), msg='Message was not sent to log') am.log('unittest log_args', log_args=dict(TEST='log unittest'))
assert journal_send.called == 1
assert journal_send.call_args[0][0].endswith('unittest log_args'), 'Message was not sent to log'
# log adds this journal field # log adds this journal field
self.assertIn('MODULE', mock_func.call_args[1]) assert 'MODULE' in journal_send.call_args[1]
self.assertIn('basic.py', mock_func.call_args[1]['MODULE']) assert 'basic.py' in journal_send.call_args[1]['MODULE']
# We added this journal field # We added this journal field
self.assertIn('TEST', mock_func.call_args[1]) assert 'TEST' in journal_send.call_args[1]
self.assertIn('log unittest', mock_func.call_args[1]['TEST']) assert 'log unittest' in journal_send.call_args[1]['TEST']

@ -1,38 +1,18 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# # Copyright (c) 2017 Ansible Project
# This file is part of Ansible # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# Make coding more python3-ish # Make coding more python3-ish
from __future__ import (absolute_import, division) from __future__ import (absolute_import, division)
__metaclass__ = type __metaclass__ = type
import errno import errno
import json from itertools import product
import sys from io import BytesIO
import time
from io import BytesIO, StringIO
import pytest import pytest
from ansible.compat.tests import unittest from ansible.module_utils._text import to_native
from ansible.compat.tests.mock import call, MagicMock, Mock, patch, sentinel
from ansible.module_utils.six import PY3
import ansible.module_utils.basic
from units.mock.procenv import swap_stdin_and_argv
class OpenBytesIO(BytesIO): class OpenBytesIO(BytesIO):
@ -45,20 +25,10 @@ class OpenBytesIO(BytesIO):
pass pass
class TestAnsibleModuleRunCommand(unittest.TestCase): @pytest.fixture
@pytest.fixture(autouse=True) def mock_os(mocker):
def run_command_mocked_env(self, mocker):
self.cmd_out = {
# os.read() is returning 'bytes', not strings
sentinel.stdout: BytesIO(),
sentinel.stderr: BytesIO(),
}
def mock_os_read(fd, nbytes): def mock_os_read(fd, nbytes):
return self.cmd_out[fd].read(nbytes) return os._cmd_out[fd].read(nbytes)
def mock_select(rlist, wlist, xlist, timeout=1):
return (rlist, [], [])
def mock_os_chdir(path): def mock_os_chdir(path):
if path == '/inaccessible': if path == '/inaccessible':
@ -68,142 +38,169 @@ class TestAnsibleModuleRunCommand(unittest.TestCase):
if path.startswith('/'): if path.startswith('/'):
return path return path
else: else:
return self.os.getcwd.return_value + '/' + path return os.getcwd.return_value + '/' + path
args = json.dumps(dict(ANSIBLE_MODULE_ARGS={})) os = mocker.patch('ansible.module_utils.basic.os')
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually os._cmd_out = {
self.stdin_swap = swap_stdin_and_argv(stdin_data=args) # os.read() is returning 'bytes', not strings
self.stdin_swap.__enter__() mocker.sentinel.stdout: BytesIO(),
mocker.sentinel.stderr: BytesIO(),
ansible.module_utils.basic._ANSIBLE_ARGS = None }
self.module = ansible.module_utils.basic.AnsibleModule(argument_spec=dict())
self.module.fail_json = MagicMock(side_effect=SystemExit) os.path.expandvars.side_effect = lambda x: x
os.path.expanduser.side_effect = lambda x: x
self.os = mocker.patch('ansible.module_utils.basic.os') os.environ = {'PATH': '/bin'}
self.os.path.expandvars.side_effect = lambda x: x os.getcwd.return_value = '/home/foo'
self.os.path.expanduser.side_effect = lambda x: x os.path.isdir.return_value = True
self.os.environ = {'PATH': '/bin'} os.chdir.side_effect = mock_os_chdir
self.os.getcwd.return_value = '/home/foo' os.read.side_effect = mock_os_read
self.os.path.isdir.return_value = True os.path.abspath.side_effect = mock_os_abspath
self.os.chdir.side_effect = mock_os_chdir
self.os.read.side_effect = mock_os_read yield os
self.os.path.abspath.side_effect = mock_os_abspath
self.subprocess = mocker.patch('ansible.module_utils.basic.subprocess') @pytest.fixture
self.cmd = Mock() def mock_subprocess(mocker):
self.cmd.returncode = 0 def mock_select(rlist, wlist, xlist, timeout=1):
self.cmd.stdin = OpenBytesIO() return (rlist, [], [])
self.cmd.stdout.fileno.return_value = sentinel.stdout
self.cmd.stderr.fileno.return_value = sentinel.stderr fake_select = mocker.patch('ansible.module_utils.basic.select')
self.subprocess.Popen.return_value = self.cmd fake_select.select.side_effect = mock_select
self.select = mocker.patch('ansible.module_utils.basic.select') subprocess = mocker.patch('ansible.module_utils.basic.subprocess')
self.select.select.side_effect = mock_select cmd = mocker.MagicMock()
yield cmd.returncode = 0
cmd.stdin = OpenBytesIO()
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually cmd.stdout.fileno.return_value = mocker.sentinel.stdout
self.stdin_swap.__exit__(None, None, None) cmd.stderr.fileno.return_value = mocker.sentinel.stderr
subprocess.Popen.return_value = cmd
def test_list_as_args(self):
self.module.run_command(['/bin/ls', 'a', ' b', 'c ']) yield subprocess
self.assertTrue(self.subprocess.Popen.called)
args, kwargs = self.subprocess.Popen.call_args
self.assertEqual(args, (['/bin/ls', 'a', ' b', 'c '], )) @pytest.fixture()
self.assertEqual(kwargs['shell'], False) def rc_am(mocker, am, mock_os, mock_subprocess):
am.fail_json = mocker.MagicMock(side_effect=SystemExit)
def test_str_as_args(self): am._os = mock_os
self.module.run_command('/bin/ls a " b" "c "') am._subprocess = mock_subprocess
self.assertTrue(self.subprocess.Popen.called) yield am
args, kwargs = self.subprocess.Popen.call_args
self.assertEqual(args, (['/bin/ls', 'a', ' b', 'c '], ))
self.assertEqual(kwargs['shell'], False) class TestRunCommandArgs:
# Format is command as passed to run_command, command to Popen as list, command to Popen as string
def test_tuple_as_args(self): ARGS_DATA = (
self.assertRaises(SystemExit, self.module.run_command, ('ls', '/')) (['/bin/ls', 'a', 'b', 'c'], ['/bin/ls', 'a', 'b', 'c'], '/bin/ls a b c'),
self.assertTrue(self.module.fail_json.called) ('/bin/ls a " b" "c "', ['/bin/ls', 'a', ' b', 'c '], '/bin/ls a " b" "c "'),
)
def test_unsafe_shell(self):
self.module.run_command('ls a " b" "c "', use_unsafe_shell=True) # pylint bug: https://github.com/PyCQA/pylint/issues/511
self.assertTrue(self.subprocess.Popen.called) # pylint: disable=undefined-variable
args, kwargs = self.subprocess.Popen.call_args @pytest.mark.parametrize('cmd, expected, shell, stdin',
self.assertEqual(args, ('ls a " b" "c "', )) ((arg, cmd_str if sh else cmd_lst, sh, {})
self.assertEqual(kwargs['shell'], True) for (arg, cmd_lst, cmd_str), sh in product(ARGS_DATA, (True, False))),
indirect=['stdin'])
def test_cwd(self): def test_args(self, cmd, expected, shell, rc_am):
self.os.getcwd.return_value = '/old' rc_am.run_command(cmd, use_unsafe_shell=shell)
self.module.run_command('/bin/ls', cwd='/new') assert rc_am._subprocess.Popen.called
self.assertEqual(self.os.chdir.mock_calls, args, kwargs = rc_am._subprocess.Popen.call_args
[call('/new'), call('/old'), ]) assert args == (expected, )
assert kwargs['shell'] == shell
def test_cwd_relative_path(self):
self.os.getcwd.return_value = '/old' @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
self.module.run_command('/bin/ls', cwd='sub-dir') def test_tuple_as_args(self, rc_am):
self.assertEqual(self.os.chdir.mock_calls, with pytest.raises(SystemExit):
[call('/old/sub-dir'), call('/old'), ]) rc_am.run_command(('ls', '/'))
assert rc_am.fail_json.called
def test_cwd_not_a_dir(self):
self.os.getcwd.return_value = '/old'
self.os.path.isdir.side_effect = lambda d: d != '/not-a-dir' class TestRunCommandCwd:
self.module.run_command('/bin/ls', cwd='/not-a-dir') @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
self.assertEqual(self.os.chdir.mock_calls, [call('/old'), ]) def test_cwd(self, mocker, rc_am):
rc_am._os.getcwd.return_value = '/old'
def test_cwd_inaccessible(self): rc_am.run_command('/bin/ls', cwd='/new')
self.assertRaises(SystemExit, self.module.run_command, '/bin/ls', cwd='/inaccessible') assert rc_am._os.chdir.mock_calls == [mocker.call('/new'), mocker.call('/old'), ]
self.assertTrue(self.module.fail_json.called)
args, kwargs = self.module.fail_json.call_args @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
self.assertEqual(kwargs['rc'], errno.EPERM) def test_cwd_relative_path(self, mocker, rc_am):
rc_am._os.getcwd.return_value = '/old'
def test_prompt_bad_regex(self): rc_am.run_command('/bin/ls', cwd='sub-dir')
self.assertRaises(SystemExit, self.module.run_command, 'foo', prompt_regex='[pP)assword:') assert rc_am._os.chdir.mock_calls == [mocker.call('/old/sub-dir'), mocker.call('/old'), ]
self.assertTrue(self.module.fail_json.called)
@pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
def test_prompt_no_match(self): def test_cwd_not_a_dir(self, mocker, rc_am):
self.cmd_out[sentinel.stdout] = BytesIO(b'hello') rc_am._os.getcwd.return_value = '/old'
(rc, _, _) = self.module.run_command('foo', prompt_regex='[pP]assword:') rc_am._os.path.isdir.side_effect = lambda d: d != '/not-a-dir'
self.assertEqual(rc, 0) rc_am.run_command('/bin/ls', cwd='/not-a-dir')
assert rc_am._os.chdir.mock_calls == [mocker.call('/old'), ]
def test_prompt_match_wo_data(self):
self.cmd_out[sentinel.stdout] = BytesIO(b'Authentication required!\nEnter password: ') @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
(rc, _, _) = self.module.run_command('foo', prompt_regex=r'[pP]assword:', data=None) def test_cwd_inaccessible(self, rc_am):
self.assertEqual(rc, 257) with pytest.raises(SystemExit):
rc_am.run_command('/bin/ls', cwd='/inaccessible')
def test_check_rc_false(self): assert rc_am.fail_json.called
self.cmd.returncode = 1 args, kwargs = rc_am.fail_json.call_args
(rc, _, _) = self.module.run_command('/bin/false', check_rc=False) assert kwargs['rc'] == errno.EPERM
self.assertEqual(rc, 1)
def test_check_rc_true(self): class TestRunCommandPrompt:
self.cmd.returncode = 1 @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
self.assertRaises(SystemExit, self.module.run_command, '/bin/false', check_rc=True) def test_prompt_bad_regex(self, rc_am):
self.assertTrue(self.module.fail_json.called) with pytest.raises(SystemExit):
args, kwargs = self.module.fail_json.call_args rc_am.run_command('foo', prompt_regex='[pP)assword:')
self.assertEqual(kwargs['rc'], 1) assert rc_am.fail_json.called
def test_text_stdin(self): @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
(rc, stdout, stderr) = self.module.run_command('/bin/foo', data='hello world') def test_prompt_no_match(self, mocker, rc_am):
self.assertEqual(self.cmd.stdin.getvalue(), b'hello world\n') rc_am._os._cmd_out[mocker.sentinel.stdout] = BytesIO(b'hello')
(rc, _, _) = rc_am.run_command('foo', prompt_regex='[pP]assword:')
def test_ascii_stdout(self): assert rc == 0
self.cmd_out[sentinel.stdout] = BytesIO(b'hello')
(rc, stdout, stderr) = self.module.run_command('/bin/cat hello.txt') @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
self.assertEqual(rc, 0) def test_prompt_match_wo_data(self, mocker, rc_am):
rc_am._os._cmd_out[mocker.sentinel.stdout] = BytesIO(b'Authentication required!\nEnter password: ')
(rc, _, _) = rc_am.run_command('foo', prompt_regex=r'[pP]assword:', data=None)
assert rc == 257
class TestRunCommandRc:
@pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
def test_check_rc_false(self, rc_am):
rc_am._subprocess.Popen.return_value.returncode = 1
(rc, _, _) = rc_am.run_command('/bin/false', check_rc=False)
assert rc == 1
@pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
def test_check_rc_true(self, rc_am):
rc_am._subprocess.Popen.return_value.returncode = 1
with pytest.raises(SystemExit):
rc_am.run_command('/bin/false', check_rc=True)
assert rc_am.fail_json.called
args, kwargs = rc_am.fail_json.call_args
assert kwargs['rc'] == 1
class TestRunCommandOutput:
@pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
def test_text_stdin(self, rc_am):
(rc, stdout, stderr) = rc_am.run_command('/bin/foo', data='hello world')
assert rc_am._subprocess.Popen.return_value.stdin.getvalue() == b'hello world\n'
@pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
def test_ascii_stdout(self, mocker, rc_am):
rc_am._os._cmd_out[mocker.sentinel.stdout] = BytesIO(b'hello')
(rc, stdout, stderr) = rc_am.run_command('/bin/cat hello.txt')
assert rc == 0
# module_utils function. On py3 it returns text and py2 it returns # module_utils function. On py3 it returns text and py2 it returns
# bytes because it's returning native strings # bytes because it's returning native strings
if PY3: assert stdout == 'hello'
self.assertEqual(stdout, u'hello')
else: @pytest.mark.parametrize('stdin', [{}], indirect=['stdin'])
self.assertEqual(stdout, b'hello') def test_utf8_output(self, mocker, rc_am):
rc_am._os._cmd_out[mocker.sentinel.stdout] = BytesIO(u'Žarn§'.encode('utf-8'))
def test_utf8_output(self): rc_am._os._cmd_out[mocker.sentinel.stderr] = BytesIO(u'لرئيسية'.encode('utf-8'))
self.cmd_out[sentinel.stdout] = BytesIO(u'Žarn§'.encode('utf-8')) (rc, stdout, stderr) = rc_am.run_command('/bin/something_ugly')
self.cmd_out[sentinel.stderr] = BytesIO(u'لرئيسية'.encode('utf-8')) assert rc == 0
(rc, stdout, stderr) = self.module.run_command('/bin/something_ugly')
self.assertEqual(rc, 0)
# module_utils function. On py3 it returns text and py2 it returns # module_utils function. On py3 it returns text and py2 it returns
# bytes because it's returning native strings # bytes because it's returning native strings
if PY3: assert stdout == to_native(u'Žarn§')
self.assertEqual(stdout, u'Žarn§') assert stderr == to_native(u'لرئيسية')
self.assertEqual(stderr, u'لرئيسية')
else:
self.assertEqual(stdout.decode('utf-8'), u'Žarn§')
self.assertEqual(stderr.decode('utf-8'), u'لرئيسية')

@ -1,99 +1,70 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# (c) 2015-2016, Toshio Kuratomi <tkuratomi@ansible.com> # (c) 2015-2017, Toshio Kuratomi <tkuratomi@ansible.com>
# # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
# This file is part of Ansible
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# Make coding more python3-ish # Make coding more python3-ish
from __future__ import (absolute_import, division) from __future__ import (absolute_import, division)
__metaclass__ = type __metaclass__ = type
import sys from itertools import chain
import json import pytest
from ansible.compat.tests import unittest
from units.mock.procenv import ModuleTestCase
from units.mock.generator import add_method
# Strings that should be converted into a typed value # Strings that should be converted into a typed value
VALID_STRINGS = ( VALID_STRINGS = (
[("'a'", 'a')], ("'a'", 'a'),
[("'1'", '1')], ("'1'", '1'),
[("1", 1)], ("1", 1),
[("True", True)], ("True", True),
[("False", False)], ("False", False),
[("{}", {})], ("{}", {}),
) )
# Passing things that aren't strings should just return the object # Passing things that aren't strings should just return the object
NONSTRINGS = ( NONSTRINGS = (
[({'a': 1}, {'a': 1})], ({'a': 1}, {'a': 1}),
) )
# These strings are not basic types. For security, these should not be # These strings are not basic types. For security, these should not be
# executed. We return the same string and get an exception for some # executed. We return the same string and get an exception for some
INVALID_STRINGS = ( INVALID_STRINGS = (
[("a=1", "a=1", SyntaxError)], ("a=1", "a=1", SyntaxError),
[("a.foo()", "a.foo()", None)], ("a.foo()", "a.foo()", None),
[("import foo", "import foo", None)], ("import foo", "import foo", None),
[("__import__('foo')", "__import__('foo')", ValueError)], ("__import__('foo')", "__import__('foo')", ValueError),
) )
def _check_simple_types(self, code, expected): @pytest.mark.parametrize('code, expected, stdin',
((c, e, {}) for c, e in chain(VALID_STRINGS, NONSTRINGS)),
indirect=['stdin'])
def test_simple_types(am, code, expected):
# test some basic usage for various types # test some basic usage for various types
self.assertEqual(self.am.safe_eval(code), expected) assert am.safe_eval(code) == expected
def _check_simple_types_with_exceptions(self, code, expected): @pytest.mark.parametrize('code, expected, stdin',
((c, e, {}) for c, e in chain(VALID_STRINGS, NONSTRINGS)),
indirect=['stdin'])
def test_simple_types_with_exceptions(am, code, expected):
# Test simple types with exceptions requested # Test simple types with exceptions requested
self.assertEqual(self.am.safe_eval(code, include_exceptions=True), (expected, None)) assert am.safe_eval(code, include_exceptions=True), (expected, None)
def _check_invalid_strings(self, code, expected): @pytest.mark.parametrize('code, expected, stdin',
self.assertEqual(self.am.safe_eval(code), expected) ((c, e, {}) for c, e, dummy in INVALID_STRINGS),
indirect=['stdin'])
def test_invalid_strings(am, code, expected):
def _check_invalid_strings_with_exceptions(self, code, expected, exception): assert am.safe_eval(code) == expected
res = self.am.safe_eval("a=1", include_exceptions=True)
self.assertEqual(res[0], "a=1")
self.assertEqual(type(res[1]), SyntaxError) @pytest.mark.parametrize('code, expected, exception, stdin',
((c, e, ex, {}) for c, e, ex in INVALID_STRINGS),
indirect=['stdin'])
@add_method(_check_simple_types, *VALID_STRINGS) def test_invalid_strings_with_exceptions(am, code, expected, exception):
@add_method(_check_simple_types, *NONSTRINGS) res = am.safe_eval(code, include_exceptions=True)
@add_method(_check_simple_types_with_exceptions, *VALID_STRINGS) assert res[0] == expected
@add_method(_check_simple_types_with_exceptions, *NONSTRINGS) if exception is None:
@add_method(_check_invalid_strings, *[[i[0][0:-1]] for i in INVALID_STRINGS]) assert res[1] == exception
@add_method(_check_invalid_strings_with_exceptions, *INVALID_STRINGS) else:
class TestSafeEval(ModuleTestCase): assert type(res[1]) == exception
def setUp(self):
super(TestSafeEval, self).setUp()
from ansible.module_utils import basic
self.old_ansible_args = basic._ANSIBLE_ARGS
basic._ANSIBLE_ARGS = None
self.am = basic.AnsibleModule(
argument_spec=dict(),
)
def tearDown(self):
super(TestSafeEval, self).tearDown()
from ansible.module_utils import basic
basic._ANSIBLE_ARGS = self.old_ansible_args

@ -1,107 +1,106 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# (c) 2016, Toshio Kuratomi <tkuratomi@ansible.com> # (c) 2016, Toshio Kuratomi <tkuratomi@ansible.com>
# # Copyright (c) 2017 Ansible Project
# This file is part of Ansible # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# Make coding more python3-ish # Make coding more python3-ish
from __future__ import (absolute_import, division, print_function) from __future__ import (absolute_import, division, print_function)
__metaclass__ = type __metaclass__ = type
import os import os
import pytest from itertools import product
try: try:
import builtins import builtins
except ImportError: except ImportError:
import __builtin__ as builtins import __builtin__ as builtins
from ansible.compat.tests import unittest import pytest
from ansible.compat.tests.mock import patch, MagicMock
from ansible.module_utils import known_hosts
from units.mock.procenv import ModuleTestCase
from units.mock.generator import add_method
SYNONYMS_0660 = (
0o660,
'0o660',
'660',
'u+rw-x,g+rw-x,o-rwx',
'u=rw,g=rw,o-rwx',
)
class TestSetModeIfDifferentBase(ModuleTestCase):
def setUp(self): @pytest.fixture
self.mock_stat1 = MagicMock() def mock_stats(mocker):
self.mock_stat1.st_mode = 0o444 mock_stat1 = mocker.MagicMock()
self.mock_stat2 = MagicMock() mock_stat1.st_mode = 0o444
self.mock_stat2.st_mode = 0o660 mock_stat2 = mocker.MagicMock()
mock_stat2.st_mode = 0o660
yield {"before": mock_stat1, "after": mock_stat2}
super(TestSetModeIfDifferentBase, self).setUp()
from ansible.module_utils import basic
self.old_ANSIBLE_ARGS = basic._ANSIBLE_ARGS
basic._ANSIBLE_ARGS = None
self.am = basic.AnsibleModule( @pytest.fixture
argument_spec=dict(), def am_check_mode(am):
) am.check_mode = True
yield am
am.check_mode = False
def tearDown(self):
super(TestSetModeIfDifferentBase, self).tearDown()
from ansible.module_utils import basic
basic._ANSIBLE_ARGS = self.old_ANSIBLE_ARGS
@pytest.fixture
def mock_lchmod(mocker):
m_lchmod = mocker.patch('ansible.module_utils.basic.os.lchmod', return_value=None, create=True)
yield m_lchmod
def _check_no_mode_given_returns_previous_changes(self, previous_changes=True):
with patch('os.lstat', side_effect=[self.mock_stat1]):
self.assertEqual(self.am.set_mode_if_different('/path/to/file', None, previous_changes), previous_changes) @pytest.mark.parametrize('previous_changes, check_mode, stdin',
product((True, False), (True, False), ({},)),
indirect=['stdin'])
def test_no_mode_given_returns_previous_changes(am, mock_stats, mock_lchmod, mocker, previous_changes, check_mode):
am.check_mode = check_mode
mocker.patch('os.lstat', side_effect=[mock_stats['before']])
m_lchmod = mocker.patch('os.lchmod', return_value=None, create=True)
assert am.set_mode_if_different('/path/to/file', None, previous_changes) == previous_changes
assert not m_lchmod.called
def _check_mode_changed_to_0660(self, mode):
@pytest.mark.parametrize('mode, check_mode, stdin',
product(SYNONYMS_0660, (True, False), ({},)),
indirect=['stdin'])
def test_mode_changed_to_0660(am, mock_stats, mocker, mode, check_mode):
# Note: This is for checking that all the different ways of specifying # Note: This is for checking that all the different ways of specifying
# 0660 mode work. It cannot be used to check that setting a mode that is # 0660 mode work. It cannot be used to check that setting a mode that is
# not equivalent to 0660 works. # not equivalent to 0660 works.
with patch('os.lstat', side_effect=[self.mock_stat1, self.mock_stat2, self.mock_stat2]) as m_lstat: am.check_mode = check_mode
with patch('os.lchmod', return_value=None, create=True) as m_lchmod: mocker.patch('os.lstat', side_effect=[mock_stats['before'], mock_stats['after'], mock_stats['after']])
self.assertEqual(self.am.set_mode_if_different('/path/to/file', mode, False), True) m_lchmod = mocker.patch('os.lchmod', return_value=None, create=True)
assert am.set_mode_if_different('/path/to/file', mode, False)
if check_mode:
assert not m_lchmod.called
else:
m_lchmod.assert_called_with(b'/path/to/file', 0o660) m_lchmod.assert_called_with(b'/path/to/file', 0o660)
def _check_mode_unchanged_when_already_0660(self, mode): @pytest.mark.parametrize('mode, check_mode, stdin',
product(SYNONYMS_0660, (True, False), ({},)),
indirect=['stdin'])
def test_mode_unchanged_when_already_0660(am, mock_stats, mocker, mode, check_mode):
# Note: This is for checking that all the different ways of specifying # Note: This is for checking that all the different ways of specifying
# 0660 mode work. It cannot be used to check that setting a mode that is # 0660 mode work. It cannot be used to check that setting a mode that is
# not equivalent to 0660 works. # not equivalent to 0660 works.
with patch('os.lstat', side_effect=[self.mock_stat2, self.mock_stat2, self.mock_stat2]) as m_lstat: am.check_mode = check_mode
self.assertEqual(self.am.set_mode_if_different('/path/to/file', mode, False), False) mocker.patch('os.lstat', side_effect=[mock_stats['after'], mock_stats['after'], mock_stats['after']])
m_lchmod = mocker.patch('os.lchmod', return_value=None, create=True)
assert not am.set_mode_if_different('/path/to/file', mode, False)
assert not m_lchmod.called
SYNONYMS_0660 = (
[[0o660]],
[['0o660']],
[['660']],
)
@pytest.mark.parametrize('check_mode, stdin',
product((True, False), ({},)),
indirect=['stdin'])
def test_missing_lchmod_is_not_link(am, mock_stats, mocker, check_mode):
"""Some platforms have lchmod (*BSD) others do not (Linux)"""
@add_method(_check_no_mode_given_returns_previous_changes, [dict(previous_changes=True)], [dict(previous_changes=False)], ) am.check_mode = check_mode
@add_method(_check_mode_changed_to_0660, *SYNONYMS_0660)
@add_method(_check_mode_unchanged_when_already_0660, *SYNONYMS_0660)
class TestSetModeIfDifferent(TestSetModeIfDifferentBase):
def test_module_utils_basic_ansible_module_set_mode_if_different(self):
with patch('os.lstat') as m:
with patch('os.lchmod', return_value=None, create=True) as m_os:
m.side_effect = [self.mock_stat1, self.mock_stat2, self.mock_stat2]
self.am._symbolic_mode_to_octal = MagicMock(side_effect=Exception)
with pytest.raises(SystemExit):
self.am.set_mode_if_different('/path/to/file', 'o+w,g+w,a-r', False)
original_hasattr = hasattr original_hasattr = hasattr
def _hasattr(obj, name): def _hasattr(obj, name):
@ -109,36 +108,40 @@ class TestSetModeIfDifferent(TestSetModeIfDifferentBase):
return False return False
return original_hasattr(obj, name) return original_hasattr(obj, name)
# FIXME: this isn't working yet mocker.patch('os.lstat', side_effect=[mock_stats['before'], mock_stats['after']])
with patch('os.lstat', side_effect=[self.mock_stat1, self.mock_stat2]): mocker.patch.object(builtins, 'hasattr', side_effect=_hasattr)
with patch.object(builtins, 'hasattr', side_effect=_hasattr): mocker.patch('os.path.islink', return_value=False)
with patch('os.path.islink', return_value=False): m_chmod = mocker.patch('os.chmod', return_value=None)
with patch('os.chmod', return_value=None) as m_chmod:
self.assertEqual(self.am.set_mode_if_different('/path/to/file/no_lchmod', 0o660, False), True) assert am.set_mode_if_different('/path/to/file/no_lchmod', 0o660, False)
with patch('os.lstat', side_effect=[self.mock_stat1, self.mock_stat2]): if check_mode:
with patch.object(builtins, 'hasattr', side_effect=_hasattr): assert not m_chmod.called
with patch('os.path.islink', return_value=True): else:
with patch('os.chmod', return_value=None) as m_chmod: m_chmod.assert_called_with(b'/path/to/file/no_lchmod', 0o660)
with patch('os.stat', return_value=self.mock_stat2):
self.assertEqual(self.am.set_mode_if_different('/path/to/file', 0o660, False), True)
@pytest.mark.parametrize('check_mode, stdin',
product((True, False), ({},)),
def _check_knows_to_change_to_0660_in_check_mode(self, mode): indirect=['stdin'])
# Note: This is for checking that all the different ways of specifying def test_missing_lchmod_is_link(am, mock_stats, mocker, check_mode):
# 0660 mode work. It cannot be used to check that setting a mode that is """Some platforms have lchmod (*BSD) others do not (Linux)"""
# not equivalent to 0660 works.
with patch('os.lstat', side_effect=[self.mock_stat1, self.mock_stat2, self.mock_stat2]) as m_lstat:
self.assertEqual(self.am.set_mode_if_different('/path/to/file', mode, False), True)
am.check_mode = check_mode
original_hasattr = hasattr
@add_method(_check_no_mode_given_returns_previous_changes, [dict(previous_changes=True)], [dict(previous_changes=False)],) def _hasattr(obj, name):
@add_method(_check_knows_to_change_to_0660_in_check_mode, *SYNONYMS_0660) if obj == os and name == 'lchmod':
@add_method(_check_mode_unchanged_when_already_0660, *SYNONYMS_0660) return False
class TestSetModeIfDifferentWithCheckMode(TestSetModeIfDifferentBase): return original_hasattr(obj, name)
def setUp(self):
super(TestSetModeIfDifferentWithCheckMode, self).setUp()
self.am.check_mode = True
def tearDown(self): mocker.patch('os.lstat', side_effect=[mock_stats['before'], mock_stats['after']])
super(TestSetModeIfDifferentWithCheckMode, self).tearDown() mocker.patch.object(builtins, 'hasattr', side_effect=_hasattr)
self.am.check_mode = False mocker.patch('os.path.islink', return_value=True)
m_chmod = mocker.patch('os.chmod', return_value=None)
mocker.patch('os.stat', return_value=mock_stats['after'])
assert am.set_mode_if_different('/path/to/file/no_lchmod', 0o660, False)
if check_mode:
assert not m_chmod.called
else:
m_chmod.assert_called_with(b'/path/to/file/no_lchmod', 0o660)

@ -0,0 +1,65 @@
# Copyright (c) 2017 Ansible Project
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
import json
import sys
from collections import MutableMapping
from io import BytesIO
import pytest
import ansible.module_utils.basic
from ansible.module_utils.six import PY3, string_types
from ansible.module_utils._text import to_bytes
@pytest.fixture
def stdin(mocker, request):
old_args = ansible.module_utils.basic._ANSIBLE_ARGS
ansible.module_utils.basic._ANSIBLE_ARGS = None
old_argv = sys.argv
sys.argv = ['ansible_unittest']
if isinstance(request.param, string_types):
args = request.param
elif isinstance(request.param, MutableMapping):
if 'ANSIBLE_MODULE_ARGS' not in request.param:
request.param = {'ANSIBLE_MODULE_ARGS': request.param}
args = json.dumps(request.param)
else:
raise Exception('Malformed data to the stdin pytest fixture')
fake_stdin = BytesIO(to_bytes(args, errors='surrogate_or_strict'))
if PY3:
mocker.patch('ansible.module_utils.basic.sys.stdin', mocker.MagicMock())
mocker.patch('ansible.module_utils.basic.sys.stdin.buffer', fake_stdin)
else:
mocker.patch('ansible.module_utils.basic.sys.stdin', fake_stdin)
yield fake_stdin
ansible.module_utils.basic._ANSIBLE_ARGS = old_args
sys.argv = old_argv
@pytest.fixture
def am(stdin, request):
old_args = ansible.module_utils.basic._ANSIBLE_ARGS
ansible.module_utils.basic._ANSIBLE_ARGS = None
old_argv = sys.argv
sys.argv = ['ansible_unittest']
argspec = {}
if hasattr(request, 'param'):
if isinstance(request.param, dict):
argspec = request.param
am = ansible.module_utils.basic.AnsibleModule(
argument_spec=argspec,
)
am._name = 'ansible_unittest'
yield am
ansible.module_utils.basic._ANSIBLE_ARGS = old_args
sys.argv = old_argv

@ -1,36 +1,18 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# This file is part of Ansible # Copyright: (c) 2017 Ansible Project
# # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# Make coding more python3-ish from __future__ import absolute_import, division, print_function
from __future__ import (absolute_import, division)
__metaclass__ = type __metaclass__ = type
from itertools import product
# to work around basic.py reading stdin
import json
import pytest import pytest
from units.mock.procenv import swap_stdin_and_argv # the module we are actually testing (sort of)
# for testing
from ansible.compat.tests.mock import patch
# the module we are actually testing (sort of
from ansible.module_utils.facts.system.distribution import DistributionFactCollector from ansible.module_utils.facts.system.distribution import DistributionFactCollector
# to generate the testcase data, you can use the script gen_distribution_version_testcase.py in hacking/tests # to generate the testcase data, you can use the script gen_distribution_version_testcase.py in hacking/tests
TESTSETS = [ TESTSETS = [
{ {
@ -880,8 +862,8 @@ DISTRIB_DESCRIPTION="CoreOS 976.0.0 (Coeur Rouge)"
] ]
@pytest.mark.parametrize("testcase", TESTSETS, ids=lambda x: x['name']) @pytest.mark.parametrize("stdin, testcase", product([{}], TESTSETS), ids=lambda x: x['name'], indirect=['stdin'])
def test_distribution_version(testcase): def test_distribution_version(am, mocker, testcase):
"""tests the distribution parsing code of the Facts class """tests the distribution parsing code of the Facts class
testsets have testsets have
@ -891,26 +873,10 @@ def test_distribution_version(testcase):
* all files that are not listed here are assumed to not exist at all * all files that are not listed here are assumed to not exist at all
* the output of pythons platform.dist() * the output of pythons platform.dist()
* results for the ansible variables distribution* and os_family * results for the ansible variables distribution* and os_family
"""
from ansible.module_utils import basic
args = json.dumps(dict(ANSIBLE_MODULE_ARGS={}))
with swap_stdin_and_argv(stdin_data=args):
basic._ANSIBLE_ARGS = None
module = basic.AnsibleModule(argument_spec=dict())
_test_one_distribution(module, testcase)
def _test_one_distribution(module, testcase):
"""run the test on one distribution testcase
* prepare some mock functions to get the testdata in
* run Facts()
* compare with the expected output
""" """
# prepare some mock functions to get the testdata in
def mock_get_file_content(fname, default=None, strip=True): def mock_get_file_content(fname, default=None, strip=True):
"""give fake content if it exists, otherwise pretend the file is empty""" """give fake content if it exists, otherwise pretend the file is empty"""
data = default data = default
@ -922,7 +888,7 @@ def _test_one_distribution(module, testcase):
data = data.strip() data = data.strip()
return data return data
def mock_get_uname_version(module): def mock_get_uname_version(am):
return testcase.get('uname_v', None) return testcase.get('uname_v', None)
def mock_file_exists(fname, allow_empty=False): def mock_file_exists(fname, allow_empty=False):
@ -942,19 +908,19 @@ def _test_one_distribution(module, testcase):
def mock_platform_version(): def mock_platform_version():
return testcase.get('platform.version', '') return testcase.get('platform.version', '')
@patch('ansible.module_utils.facts.system.distribution.get_file_content', mock_get_file_content) mocker.patch('ansible.module_utils.facts.system.distribution.get_file_content', mock_get_file_content)
@patch('ansible.module_utils.facts.system.distribution.get_uname_version', mock_get_uname_version) mocker.patch('ansible.module_utils.facts.system.distribution.get_uname_version', mock_get_uname_version)
@patch('ansible.module_utils.facts.system.distribution._file_exists', mock_file_exists) mocker.patch('ansible.module_utils.facts.system.distribution._file_exists', mock_file_exists)
@patch('platform.dist', lambda: testcase['platform.dist']) mocker.patch('platform.dist', lambda: testcase['platform.dist'])
@patch('platform.system', mock_platform_system) mocker.patch('platform.system', mock_platform_system)
@patch('platform.release', mock_platform_release) mocker.patch('platform.release', mock_platform_release)
@patch('platform.version', mock_platform_version) mocker.patch('platform.version', mock_platform_version)
def get_facts(testcase):
# run Facts()
distro_collector = DistributionFactCollector() distro_collector = DistributionFactCollector()
res = distro_collector.collect(module) generated_facts = distro_collector.collect(am)
return res
generated_facts = get_facts(testcase) # compare with the expected output
# testcase['result'] has a list of variables and values it expects Facts() to set # testcase['result'] has a list of variables and values it expects Facts() to set
for key, val in testcase['result'].items(): for key, val in testcase['result'].items():

@ -1,33 +1,20 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# (c) 2015, Michael Scherer <mscherer@redhat.com> # (c) 2015, Michael Scherer <mscherer@redhat.com>
# # Copyright (c) 2017 Ansible Project
# This file is part of Ansible # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
#
# Ansible is free software: you can redistribute it and/or modify from __future__ import (absolute_import, division, print_function)
# it under the terms of the GNU General Public License as published by __metaclass__ = type
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
import json import json
import os.path import os.path
import ansible.module_utils.basic import pytest
from ansible.compat.tests import unittest
from ansible.compat.tests.mock import Mock, patch
from ansible.module_utils import known_hosts from ansible.module_utils import known_hosts
from units.mock.procenv import swap_stdin_and_argv
class TestAnsibleModuleKnownHosts(unittest.TestCase): URLS = {
urls = {
'ssh://one.example.org/example.git': { 'ssh://one.example.org/example.git': {
'is_ssh_url': True, 'is_ssh_url': True,
'get_fqdn': 'one.example.org', 'get_fqdn': 'one.example.org',
@ -97,39 +84,36 @@ class TestAnsibleModuleKnownHosts(unittest.TestCase):
}, },
} }
def test_is_ssh_url(self):
for u in self.urls:
self.assertEqual(known_hosts.is_ssh_url(u), self.urls[u]['is_ssh_url'])
def test_get_fqdn_and_port(self): @pytest.mark.parametrize('url, is_ssh_url', ((k, v['is_ssh_url']) for k, v in URLS.items()))
for u in self.urls: def test_is_ssh_url(url, is_ssh_url):
self.assertEqual(known_hosts.get_fqdn_and_port(u), (self.urls[u]['get_fqdn'], self.urls[u]['port'])) assert known_hosts.is_ssh_url(url) == is_ssh_url
def test_add_host_key(self):
# Copied @pytest.mark.parametrize('url, fqdn, port', ((k, v['get_fqdn'], v['port']) for k, v in URLS.items()))
args = json.dumps(dict(ANSIBLE_MODULE_ARGS={})) def test_get_fqdn_and_port(url, fqdn, port):
# unittest doesn't have a clean place to use a context manager, so we have to enter/exit manually assert known_hosts.get_fqdn_and_port(url) == (fqdn, port)
with swap_stdin_and_argv(stdin_data=args):
ansible.module_utils.basic._ANSIBLE_ARGS = None
self.module = ansible.module_utils.basic.AnsibleModule(argument_spec=dict())
get_bin_path = Mock() @pytest.mark.parametrize('fqdn, port, add_host_key_cmd, stdin',
((v['get_fqdn'], v['port'], v['add_host_key_cmd'], {})
for v in URLS.values() if v['is_ssh_url']),
indirect=['stdin'])
def test_add_host_key(am, mocker, fqdn, port, add_host_key_cmd):
get_bin_path = mocker.MagicMock()
get_bin_path.return_value = keyscan_cmd = "/custom/path/ssh-keyscan" get_bin_path.return_value = keyscan_cmd = "/custom/path/ssh-keyscan"
self.module.get_bin_path = get_bin_path am.get_bin_path = get_bin_path
run_command = Mock() run_command = mocker.MagicMock()
run_command.return_value = (0, "Needs output, otherwise thinks ssh-keyscan timed out'", "") run_command.return_value = (0, "Needs output, otherwise thinks ssh-keyscan timed out'", "")
self.module.run_command = run_command am.run_command = run_command
append_to_file = Mock() append_to_file = mocker.MagicMock()
append_to_file.return_value = (None,) append_to_file.return_value = (None,)
self.module.append_to_file = append_to_file am.append_to_file = append_to_file
with patch('os.path.isdir', return_value=True): mocker.patch('os.path.isdir', return_value=True)
with patch('os.path.exists', return_value=True): mocker.patch('os.path.exists', return_value=True)
for u in self.urls:
if self.urls[u]['is_ssh_url']: known_hosts.add_host_key(am, fqdn, port=port)
known_hosts.add_host_key(self.module, self.urls[u]['get_fqdn'], port=self.urls[u]['port']) run_command.assert_called_with(keyscan_cmd + add_host_key_cmd)
run_command.assert_called_with(keyscan_cmd + self.urls[u]['add_host_key_cmd'])

@ -1,13 +1,13 @@
import json # Copyright (c) 2017 Ansible Project
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
import sys import sys
from ansible.compat.tests import unittest from ansible.compat.tests import unittest
from ansible.compat.tests.mock import patch, MagicMock from ansible.compat.tests.mock import patch, MagicMock
from ansible.module_utils.basic import AnsibleModule
from ansible.module_utils.six.moves import builtins from ansible.module_utils.six.moves import builtins
from ansible.module_utils._text import to_native from ansible.module_utils._text import to_native
from units.mock.procenv import swap_stdin_and_argv
import pprint import pprint

@ -1,34 +1,23 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# (c) 2016 Toshio Kuratomi <tkuratomi@ansible.com> # (c) 2016 Toshio Kuratomi <tkuratomi@ansible.com>
# # Copyright (c) 2017 Ansible Project
# This file is part of Ansible # GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
#
# Ansible is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Ansible is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# Make coding more python3-ish # Make coding more python3-ish
from __future__ import (absolute_import, division) from __future__ import (absolute_import, division)
__metaclass__ = type __metaclass__ = type
from ansible.compat.tests import unittest import itertools
from ansible.module_utils.six import PY3
from units.mock.generator import add_method import pytest
from ansible.module_utils.six import PY3
# Internal API while this is still being developed. Eventually move to # Internal API while this is still being developed. Eventually move to
# module_utils.text # module_utils.common.text
from ansible.module_utils._text import to_text, to_bytes, to_native from ansible.module_utils._text import to_text, to_bytes, to_native
# Format: byte representation, text representation, encoding of byte representation # Format: byte representation, text representation, encoding of byte representation
VALID_STRINGS = ( VALID_STRINGS = (
(b'abcde', u'abcde', 'ascii'), (b'abcde', u'abcde', 'ascii'),
@ -40,29 +29,25 @@ VALID_STRINGS = (
) )
def _check_to_text(self, in_string, encoding, expected): @pytest.mark.parametrize('in_string, encoding, expected',
itertools.chain(((d[0], d[2], d[1]) for d in VALID_STRINGS),
((d[1], d[2], d[1]) for d in VALID_STRINGS)))
def test_to_text(in_string, encoding, expected):
"""test happy path of decoding to text""" """test happy path of decoding to text"""
self.assertEqual(to_text(in_string, encoding), expected) assert to_text(in_string, encoding) == expected
def _check_to_bytes(self, in_string, encoding, expected): @pytest.mark.parametrize('in_string, encoding, expected',
itertools.chain(((d[0], d[2], d[0]) for d in VALID_STRINGS),
((d[1], d[2], d[0]) for d in VALID_STRINGS)))
def test_to_bytes(in_string, encoding, expected):
"""test happy path of encoding to bytes""" """test happy path of encoding to bytes"""
self.assertEqual(to_bytes(in_string, encoding), expected) assert to_bytes(in_string, encoding) == expected
def _check_to_native(self, in_string, encoding, py2_expected, py3_expected): @pytest.mark.parametrize('in_string, encoding, expected',
itertools.chain(((d[0], d[2], d[1] if PY3 else d[0]) for d in VALID_STRINGS),
((d[1], d[2], d[1] if PY3 else d[0]) for d in VALID_STRINGS)))
def test_to_native(in_string, encoding, expected):
"""test happy path of encoding to native strings""" """test happy path of encoding to native strings"""
if PY3: assert to_native(in_string, encoding) == expected
self.assertEqual(to_native(in_string, encoding), py3_expected)
else:
self.assertEqual(to_native(in_string, encoding), py2_expected)
@add_method(_check_to_text, [(i[0], i[2], i[1]) for i in VALID_STRINGS])
@add_method(_check_to_text, [(i[1], i[2], i[1]) for i in VALID_STRINGS])
@add_method(_check_to_bytes, [(i[0], i[2], i[0]) for i in VALID_STRINGS])
@add_method(_check_to_bytes, [(i[1], i[2], i[0]) for i in VALID_STRINGS])
@add_method(_check_to_native, [(i[0], i[2], i[0], i[1]) for i in VALID_STRINGS])
@add_method(_check_to_native, [(i[1], i[2], i[0], i[1]) for i in VALID_STRINGS])
class TestModuleUtilsText(unittest.TestCase):
pass

@ -0,0 +1,24 @@
# Copyright (c) 2017 Ansible Project
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
import json
from collections import MutableMapping
import pytest
from ansible.module_utils.six import string_types
from ansible.module_utils._text import to_bytes
@pytest.fixture
def patch_ansible_module(request, mocker):
if isinstance(request.param, string_types):
args = request.param
elif isinstance(request.param, MutableMapping):
if 'ANSIBLE_MODULE_ARGS' not in request.param:
request.param = {'ANSIBLE_MODULE_ARGS': request.param}
args = json.dumps(request.param)
else:
raise Exception('Malformed data to the patch_ansible_module pytest fixture')
mocker.patch('ansible.module_utils.basic._ANSIBLE_ARGS', to_bytes(args))

@ -1,23 +1,24 @@
# Copyright (c) 2017 Ansible Project
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
import json import json
from ansible.compat.tests import unittest import pytest
from ansible.compat.tests.mock import patch
from ansible.module_utils import basic
from ansible.modules.packaging.language import pip from ansible.modules.packaging.language import pip
from units.modules.utils import set_module_args, AnsibleFailJson, ModuleTestCase
pytestmark = pytest.mark.usefixtures('patch_ansible_module')
class TestPip(ModuleTestCase):
def setUp(self):
super(TestPip, self).setUp()
@patch.object(basic.AnsibleModule, 'get_bin_path') @pytest.mark.parametrize('patch_ansible_module', [{'name': 'six'}], indirect=['patch_ansible_module'])
def test_failure_when_pip_absent(self, mock_get_bin_path): def test_failure_when_pip_absent(mocker, capfd):
get_bin_path = mocker.patch('ansible.module_utils.basic.AnsibleModule.get_bin_path')
get_bin_path.return_value = None
mock_get_bin_path.return_value = None with pytest.raises(SystemExit):
with self.assertRaises(AnsibleFailJson):
set_module_args({'name': 'six'})
pip.main() pip.main()
out, err = capfd.readouterr()
results = json.loads(out)
assert results['failed']
assert 'pip needs to be installed' in results['msg']

Loading…
Cancel
Save