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")