Differential D4920 Diff 24623 python_modules/dagster/dagster_tests/cli_tests/command_tests/test_execute_command.py
Changeset View
Changeset View
Standalone View
Standalone View
python_modules/dagster/dagster_tests/cli_tests/command_tests/test_execute_command.py
from __future__ import print_function | from __future__ import print_function | ||||
import os | import os | ||||
import re | import re | ||||
import click | import click | ||||
import pytest | import pytest | ||||
from click import UsageError | from click import UsageError | ||||
from click.testing import CliRunner | from click.testing import CliRunner | ||||
from dagster.cli.pipeline import execute_execute_command, pipeline_execute_command | from dagster.cli.pipeline import execute_execute_command, pipeline_execute_command | ||||
from dagster.core.errors import DagsterInvariantViolationError | from dagster.core.errors import DagsterInvariantViolationError | ||||
from dagster.core.test_utils import instance_for_test, new_cwd | from dagster.core.test_utils import instance_for_test, new_cwd | ||||
from dagster.utils import file_relative_path, merge_dicts | from dagster.utils import file_relative_path, merge_dicts | ||||
from .test_cli_commands import ( | from .test_cli_commands import ( | ||||
non_existant_python_origin_target_args, | |||||
pipeline_python_origin_contexts, | pipeline_python_origin_contexts, | ||||
valid_pipeline_python_origin_target_cli_args, | valid_pipeline_python_origin_target_cli_args, | ||||
) | ) | ||||
def test_execute_mode_command(): | def test_execute_mode_command(): | ||||
runner = CliRunner() | runner = CliRunner() | ||||
▲ Show 20 Lines • Show All 227 Lines • ▼ Show 20 Lines | with instance_for_test() as instance: | ||||
UsageError, | UsageError, | ||||
match=re.escape("Invalid set of CLI arguments for loading repository/pipeline"), | match=re.escape("Invalid set of CLI arguments for loading repository/pipeline"), | ||||
): | ): | ||||
execute_execute_command( | execute_execute_command( | ||||
kwargs=args, instance=instance, | kwargs=args, instance=instance, | ||||
) | ) | ||||
def test_execute_non_existant_file(): | |||||
with instance_for_test() as instance: | |||||
kwargs = non_existant_python_origin_target_args() | |||||
with pytest.raises(OSError): | |||||
execute_execute_command(kwargs=kwargs, instance=instance) | |||||
def test_attribute_not_found(): | def test_attribute_not_found(): | ||||
with instance_for_test() as instance: | with instance_for_test() as instance: | ||||
with pytest.raises( | with pytest.raises( | ||||
DagsterInvariantViolationError, | DagsterInvariantViolationError, | ||||
match=re.escape("nope not found at module scope in file"), | match=re.escape("nope not found at module scope in file"), | ||||
): | ): | ||||
execute_execute_command( | execute_execute_command( | ||||
kwargs={ | kwargs={ | ||||
▲ Show 20 Lines • Show All 295 Lines • Show Last 20 Lines |