def test_execute_pipeline_command_missing_args():
runner = CliRunner()
with safe_tempfile_path() as filename:
result = runner.invoke(
execute_pipeline_command,
[
'-y',
file_relative_path(__file__, 'repository_file.yaml'), 'foo',
filename
],
)
assert result.exit_code == 0
with open(filename, 'r') as f:
# Read lines from output file, and strip newline characters
lines = [line.rstrip() for line in f.readlines()]
assert len(lines) == 3
# Check all lines are serialized dagster named tuples
for line in lines:
deserialize_json_to_dagster_namedtuple(line)
assert deserialize_json_to_dagster_namedtuple(
lines[0]) == IPCStartMessage()
assert deserialize_json_to_dagster_namedtuple(
lines[-1]) == IPCEndMessage()
def test_execute_run_command():
runner = CliRunner()
with safe_tempfile_path() as filename:
with seven.TemporaryDirectory() as temp_dir:
instance = DagsterInstance.local_temp(temp_dir)
pipeline_run = create_run_for_test(
instance,
pipeline_name='foo',
environment_dict={},
mode='default',
)
result = runner.invoke(
execute_run_command,
[
'-y',
file_relative_path(__file__, 'repository_file.yaml'),
filename,
"--instance-ref={instance_ref_json}".format(
instance_ref_json=serialize_dagster_namedtuple(
instance.get_ref())),
'--pipeline-run={pipeline_run}'.format(
pipeline_run=serialize_dagster_namedtuple(
pipeline_run)),
],
)
assert result.exit_code == 0
with open(filename, 'r') as f:
# Read lines from output file, and strip newline characters
lines = [line.rstrip() for line in f.readlines()]
assert len(lines) == 13
# Check all lines are serialized dagster named tuples
for line in lines:
deserialize_json_to_dagster_namedtuple(line)
# Check for start ane dnd messages
assert deserialize_json_to_dagster_namedtuple(
lines[0]) == IPCStartMessage()
assert deserialize_json_to_dagster_namedtuple(
lines[-1]) == IPCEndMessage()