Пример #1
0
def test_print_command():
    for cli_args in valid_execute_args():
        execute_print_command(verbose=True,
                              cli_args=cli_args,
                              print_fn=no_print)

    for cli_args in valid_execute_args():
        execute_print_command(verbose=False,
                              cli_args=cli_args,
                              print_fn=no_print)

    runner = CliRunner()

    for cli_args in valid_cli_args():
        result = runner.invoke(pipeline_print_command, cli_args)
        assert result.exit_code == 0

        result = runner.invoke(pipeline_print_command,
                               ['--verbose'] + cli_args)
        assert result.exit_code == 0

    res = runner.invoke(
        pipeline_print_command,
        [
            '--verbose',
            '-f',
            file_relative_path(__file__, 'test_cli_commands.py'),
            '-n',
            'bar',
            'baz',
        ],
    )
    assert res.exit_code == 0
Пример #2
0
def test_job_command_only_selects_job():
    job_kwargs = {
        "workspace": None,
        "pipeline_or_job": "my_job",
        "python_file": file_relative_path(__file__,
                                          "repo_pipeline_and_job.py"),
        "module_name": None,
        "attribute": "my_repo",
    }
    pipeline_kwargs = job_kwargs.copy()
    pipeline_kwargs["pipeline_or_job"] = "my_pipeline"

    with instance_for_test() as instance:
        execute_print_command(
            instance=instance,
            verbose=False,
            cli_args=job_kwargs,
            print_fn=no_print,
            using_job_op_graph_apis=True,
        )

        with pytest.raises(Exception, match="not found in repository"):
            execute_print_command(
                instance,
                verbose=False,
                cli_args=pipeline_kwargs,
                print_fn=no_print,
                using_job_op_graph_apis=True,
            )
Пример #3
0
def test_print_command(gen_pipeline_args):
    with gen_pipeline_args as (cli_args, instance):
        execute_print_command(
            instance=instance,
            verbose=False,
            cli_args=cli_args,
            print_fn=no_print,
        )
Пример #4
0
def test_print_command():
    for cli_args, uses_legacy_repository_yaml_format in valid_execute_args():
        if uses_legacy_repository_yaml_format:
            with pytest.warns(
                UserWarning,
                match=re.escape(
                    'You are using the legacy repository yaml format. Please update your file '
                ),
            ):
                execute_print_command(verbose=True, cli_args=cli_args, print_fn=no_print)
        else:
            execute_print_command(verbose=True, cli_args=cli_args, print_fn=no_print)

    for cli_args, uses_legacy_repository_yaml_format in valid_execute_args():
        if uses_legacy_repository_yaml_format:
            with pytest.warns(
                UserWarning,
                match=re.escape(
                    'You are using the legacy repository yaml format. Please update your file '
                ),
            ):
                execute_print_command(verbose=False, cli_args=cli_args, print_fn=no_print)
        else:
            execute_print_command(verbose=False, cli_args=cli_args, print_fn=no_print)

    runner = CliRunner()

    for cli_args, uses_legacy_repository_yaml_format in valid_cli_args():
        if uses_legacy_repository_yaml_format:
            with pytest.warns(
                UserWarning,
                match=re.escape(
                    'You are using the legacy repository yaml format. Please update your file '
                ),
            ):
                result = runner.invoke(pipeline_print_command, cli_args)
                assert result.exit_code == 0, result.stdout

                result = runner.invoke(pipeline_print_command, ['--verbose'] + cli_args)
                assert result.exit_code == 0, result.stdout
        else:
            result = runner.invoke(pipeline_print_command, cli_args)
            assert result.exit_code == 0, result.stdout

            result = runner.invoke(pipeline_print_command, ['--verbose'] + cli_args)
            assert result.exit_code == 0, result.stdout

    res = runner.invoke(
        pipeline_print_command,
        [
            '--verbose',
            '-f',
            file_relative_path(__file__, 'test_cli_commands.py'),
            '-a',
            'bar',
            '-p',
            'baz',
        ],
    )
    assert res.exit_code == 0, res.stdout
Пример #5
0
def test_print_command():
    for cli_args in valid_execute_args():
        execute_print_command(verbose=True, cli_args=cli_args, print_fn=no_print)

    for cli_args in valid_execute_args():
        execute_print_command(verbose=False, cli_args=cli_args, print_fn=no_print)

    runner = CliRunner()

    for cli_args in valid_cli_args():
        result = runner.invoke(pipeline_print_command, cli_args)
        assert result.exit_code == 0

        result = runner.invoke(pipeline_print_command, ['--verbose'] + cli_args)
        assert result.exit_code == 0
Пример #6
0
def job_print_command(verbose, **cli_args):
    with get_instance_for_service("``dagster job print``") as instance:
        return execute_print_command(instance,
                                     verbose,
                                     cli_args,
                                     click.echo,
                                     using_job_op_graph_apis=True)
Пример #7
0
def test_print_command(gen_pipeline_args):
    with gen_pipeline_args as (cli_args, uses_legacy_repository_yaml_format, instance):
        if uses_legacy_repository_yaml_format:
            with pytest.warns(
                UserWarning,
                match=re.escape(
                    "You are using the legacy repository yaml format. Please update your file "
                ),
            ):
                execute_print_command(
                    verbose=False, cli_args=cli_args, print_fn=no_print, instance=instance
                )
        else:
            execute_print_command(
                verbose=False, cli_args=cli_args, print_fn=no_print, instance=instance
            )
Пример #8
0
def test_print_command_verbose(gen_pipeline_args):
    with gen_pipeline_args as (cli_args, _instance):
        execute_print_command(verbose=True, cli_args=cli_args, print_fn=no_print)
Пример #9
0
def test_print_command():
    for cli_args in valid_cli_args():
        execute_print_command(verbose=True, cli_args=cli_args, print_fn=no_print)

    for cli_args in valid_cli_args():
        execute_print_command(verbose=False, cli_args=cli_args, print_fn=no_print)