diff --git a/data_validation/__main__.py b/data_validation/__main__.py index ab8560ea..c342934e 100644 --- a/data_validation/__main__.py +++ b/data_validation/__main__.py @@ -356,6 +356,7 @@ def config_runner(args): setattr(args, "config_dir", None) setattr(args, "config_file", config_file_path) config_managers = build_config_managers_from_yaml(args, config_file_path) + run_validations(args, config_managers) else: if args.kube_completions: logging.warning( @@ -364,7 +365,8 @@ def config_runner(args): config_file_names = cli_tools.list_validations(config_dir=args.config_dir) config_managers = [] for file in config_file_names: - config_managers.extend(build_config_managers_from_yaml(args, file)) + config_managers = build_config_managers_from_yaml(args, file) + run_validations(args, config_managers) else: if args.kube_completions: logging.warning( @@ -372,8 +374,7 @@ def config_runner(args): ) config_file_path = _get_arg_config_file(args) config_managers = build_config_managers_from_yaml(args, config_file_path) - - run_validations(args, config_managers) + run_validations(args, config_managers) def build_config_managers_from_yaml(args, config_file_path): diff --git a/tests/unit/test__main.py b/tests/unit/test__main.py index b9450efe..2336a224 100644 --- a/tests/unit/test__main.py +++ b/tests/unit/test__main.py @@ -162,8 +162,8 @@ def test_config_runner_2(mock_args, mock_build, mock_run, caplog): assert caplog.messages == [ "--kube-completions or -kc specified, however not running in Kubernetes Job completion, check your command line." ] - # assert that 3 config managers are present - assert len(mock_run.call_args.args[1]) == 3 + # assert that validation is called thrice, once for each file + assert mock_run.call_count == 3 @mock.patch("data_validation.__main__.run_validations")