253 lines
6.5 KiB
Python
253 lines
6.5 KiB
Python
# Copyright (c) 2015-2018 Cisco Systems, Inc.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
# of this software and associated documentation files (the "Software"), to
|
|
# deal in the Software without restriction, including without limitation the
|
|
# rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
|
|
# sell copies of the Software, and to permit persons to whom the Software is
|
|
# furnished to do so, subject to the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included in
|
|
# all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
# DEALINGS IN THE SOFTWARE.
|
|
|
|
from uuid import uuid4
|
|
import copy
|
|
import functools
|
|
import os
|
|
import shutil
|
|
|
|
try:
|
|
from pathlib import Path
|
|
except ImportError:
|
|
from pathlib2 import Path
|
|
|
|
import pytest
|
|
|
|
from molecule import util
|
|
from molecule import config
|
|
|
|
|
|
@pytest.helpers.register
|
|
def write_molecule_file(filename, data):
|
|
util.write_file(filename, util.safe_dump(data))
|
|
|
|
|
|
@pytest.helpers.register
|
|
def os_split(s):
|
|
rest, tail = os.path.split(s)
|
|
if rest in ("", os.path.sep):
|
|
return (tail,)
|
|
return os_split(rest) + (tail,)
|
|
|
|
|
|
@pytest.fixture
|
|
def _molecule_dependency_galaxy_section_data():
|
|
return {"dependency": {"name": "galaxy"}}
|
|
|
|
|
|
@pytest.fixture
|
|
def _molecule_driver_section_data():
|
|
return {"driver": {"name": "docker", "options": {"managed": True}}}
|
|
|
|
|
|
@pytest.fixture
|
|
def _molecule_lint_section_data():
|
|
return {"lint": {"name": "yamllint"}}
|
|
|
|
|
|
@pytest.fixture
|
|
def _molecule_platforms_section_data():
|
|
return {
|
|
"platforms": [
|
|
{"name": "instance-1", "groups": ["foo", "bar"], "children": ["child1"]},
|
|
{"name": "instance-2", "groups": ["baz", "foo"], "children": ["child2"]},
|
|
]
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def _molecule_provisioner_section_data():
|
|
return {
|
|
"provisioner": {
|
|
"name": "ansible",
|
|
"options": {"become": True},
|
|
"lint": {"name": "ansible-lint"},
|
|
"config_options": {},
|
|
}
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def _molecule_scenario_section_data():
|
|
return {"scenario": {"name": "default"}}
|
|
|
|
|
|
@pytest.fixture
|
|
def _molecule_verifier_section_data():
|
|
return {"verifier": {"name": "testinfra", "lint": {"name": "flake8"}}}
|
|
|
|
|
|
@pytest.fixture
|
|
def molecule_data(
|
|
_molecule_dependency_galaxy_section_data,
|
|
_molecule_driver_section_data,
|
|
_molecule_lint_section_data,
|
|
_molecule_platforms_section_data,
|
|
_molecule_provisioner_section_data,
|
|
_molecule_scenario_section_data,
|
|
_molecule_verifier_section_data,
|
|
):
|
|
|
|
fixtures = [
|
|
_molecule_dependency_galaxy_section_data,
|
|
_molecule_driver_section_data,
|
|
_molecule_lint_section_data,
|
|
_molecule_platforms_section_data,
|
|
_molecule_provisioner_section_data,
|
|
_molecule_scenario_section_data,
|
|
_molecule_verifier_section_data,
|
|
]
|
|
|
|
return functools.reduce(lambda x, y: util.merge_dicts(x, y), fixtures)
|
|
|
|
|
|
@pytest.fixture
|
|
def molecule_directory_fixture(temp_dir):
|
|
return pytest.helpers.molecule_directory()
|
|
|
|
|
|
@pytest.fixture
|
|
def molecule_scenario_directory_fixture(molecule_directory_fixture):
|
|
path = pytest.helpers.molecule_scenario_directory()
|
|
if not os.path.isdir(path):
|
|
os.makedirs(path)
|
|
|
|
return path
|
|
|
|
|
|
@pytest.fixture
|
|
def molecule_ephemeral_directory_fixture(molecule_scenario_directory_fixture):
|
|
path = pytest.helpers.molecule_ephemeral_directory(str(uuid4()))
|
|
if not os.path.isdir(path):
|
|
os.makedirs(path)
|
|
yield
|
|
shutil.rmtree(str(Path(path).parent))
|
|
|
|
|
|
@pytest.fixture
|
|
def molecule_file_fixture(
|
|
molecule_scenario_directory_fixture, molecule_ephemeral_directory_fixture
|
|
):
|
|
return pytest.helpers.molecule_file()
|
|
|
|
|
|
@pytest.fixture
|
|
def config_instance(molecule_file_fixture, molecule_data, request):
|
|
mdc = copy.deepcopy(molecule_data)
|
|
if hasattr(request, "param"):
|
|
util.merge_dicts(mdc, request.getfixturevalue(request.param))
|
|
pytest.helpers.write_molecule_file(molecule_file_fixture, mdc)
|
|
c = config.Config(molecule_file_fixture)
|
|
c.command_args = {"subcommand": "test"}
|
|
|
|
return c
|
|
|
|
|
|
# Mocks
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_print_debug(mocker):
|
|
return mocker.patch("molecule.util.print_debug")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_logger_info(mocker):
|
|
return mocker.patch("logging.Logger.info")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_logger_out(mocker):
|
|
return mocker.patch("molecule.logger.CustomLogger.out")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_logger_warn(mocker):
|
|
return mocker.patch("logging.Logger.warn")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_logger_error(mocker):
|
|
return mocker.patch("logging.Logger.error")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_logger_critical(mocker):
|
|
return mocker.patch("logging.Logger.critical")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_logger_success(mocker):
|
|
return mocker.patch("molecule.logger.CustomLogger.success")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_run_command(mocker):
|
|
m = mocker.patch("molecule.util.run_command")
|
|
m.return_value = mocker.Mock(stdout=b"patched-run-command-stdout")
|
|
|
|
return m
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_ansible_converge(mocker):
|
|
m = mocker.patch("molecule.provisioner.ansible.Ansible.converge")
|
|
m.return_value = "patched-ansible-converge-stdout"
|
|
|
|
return m
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_add_or_update_vars(mocker):
|
|
return mocker.patch("molecule.provisioner.ansible.Ansible._add_or_update_vars")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_yamllint(mocker):
|
|
return mocker.patch("molecule.lint.yamllint.Yamllint.execute")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_flake8(mocker):
|
|
return mocker.patch("molecule.verifier.lint.flake8.Flake8.execute")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_ansible_galaxy(mocker):
|
|
return mocker.patch("molecule.dependency.ansible_galaxy.AnsibleGalaxy.execute")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_testinfra(mocker):
|
|
return mocker.patch("molecule.verifier.testinfra.Testinfra.execute")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_scenario_setup(mocker):
|
|
mocker.patch("molecule.config.Config.env")
|
|
|
|
return mocker.patch("molecule.scenario.Scenario._setup")
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_config_validate(mocker):
|
|
return mocker.patch("molecule.config.Config._validate")
|