molecule-hetznercloud/molecule_hetznercloud/test/unit/conftest.py
2020-04-24 10:45:20 +01:00

245 lines
6.3 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.
import copy
import functools
import os
import shutil
from uuid import uuid4
import pytest
from molecule import config, util
try:
from pathlib import Path
except ImportError:
from pathlib2 import Path
@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": "hetznercloud", "options": {"managed": True}}}
@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_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_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")