forked from ansible/ansible
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Start of tests for modify_module, specifically to ensure proper sheba…
…ng replacement on old style modules (ansible#36602) (cherry picked from commit a7062b7)
- Loading branch information
Showing
1 changed file
with
42 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,42 @@ | ||
# Copyright (c) 2018 Matt Martz <matt@sivel.net> | ||
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt) | ||
# -*- coding: utf-8 -*- | ||
|
||
from __future__ import (absolute_import, division, print_function) | ||
__metaclass__ = type | ||
|
||
import pytest | ||
|
||
from ansible.executor.module_common import modify_module | ||
from ansible.module_utils.six import PY2 | ||
|
||
from test_module_common import templar | ||
|
||
|
||
FAKE_OLD_MODULE = b'''#!/usr/bin/python | ||
import sys | ||
print('{"result": "%s"}' % sys.executable) | ||
''' | ||
|
||
|
||
@pytest.fixture | ||
def fake_old_module_open(mocker): | ||
m = mocker.mock_open(read_data=FAKE_OLD_MODULE) | ||
if PY2: | ||
mocker.patch('__builtin__.open', m) | ||
else: | ||
mocker.patch('builtins.open', m) | ||
|
||
|
||
def test_shebang(fake_old_module_open): | ||
(data, style, shebang) = modify_module('fake_module', 'fake_path', {}) | ||
assert shebang == '#!/usr/bin/python' | ||
|
||
|
||
def test_shebang_task_vars(fake_old_module_open, templar): | ||
task_vars = { | ||
'ansible_python_interpreter': '/usr/bin/python3' | ||
} | ||
|
||
(data, style, shebang) = modify_module('fake_module', 'fake_path', {}, task_vars=task_vars, templar=templar) | ||
assert shebang == '#!/usr/bin/python3' |