path: root/datamaps/tests/test_cli.py

import os
import shutil
from pathlib import Path

from click.testing import CliRunner

from datamaps.main import _import


def test_no_expected_sheet_in_batch_import_to_master(mock_config, resource_dir):
    """If there is a batch of spreadsheet files present
    in the input directory, and at least one file is present
    that does not have a sheet as named by the datamap, we want
    the good files to process and the error file to be flagged,
    but the process to continue where it can.

    KeyError thrown by engine.parsing.query_key() needs to be handled.
    """
    runner = CliRunner()
    mock_config.initialise()
    _copy_resources_to_input(mock_config, resource_dir)
    result = runner.invoke(_import, ["templates", "-m"])
    assert "Expected Sheet Missing: sheet Introduction in test_template.xlsm is expected from" \
           " datamap.csv. Not processing that file until fixed." in result.output
    # assert "Imported data from input/dft1_temp.xlsm to output/master.xlsx." in result.output
    # assert "Finished." in result.output


def _copy_resources_to_input(mock_config, resource_dir):
    for fl in os.listdir(resource_dir):
        if os.path.isfile(os.path.join(resource_dir, fl)):
            shutil.copy(
                Path.cwd() / "tests" / "resources" / fl,
                (Path(mock_config.PLATFORM_DOCS_DIR) / "input"),
            )