mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 23:09:44 +01:00
Ruslan Kuprieiev
357ce82f28
This is an extension of our callback stages (no deps, so always considered as changed) for stages with dependencies, so that these stages could be used properly in the middle of the DAG. See attached issue for more info. Related to #2378
117 lines
2.9 KiB
Python
117 lines
2.9 KiB
Python
from dvc.cli import parse_args
|
|
from dvc.command.run import CmdRun
|
|
|
|
|
|
def test_run(mocker, dvc_repo):
|
|
cli_args = parse_args(
|
|
[
|
|
"run",
|
|
"--deps",
|
|
"deps",
|
|
"--outs",
|
|
"outs",
|
|
"--outs-no-cache",
|
|
"outs-no-cache",
|
|
"--metrics",
|
|
"metrics",
|
|
"--metrics-no-cache",
|
|
"metrics-no-cache",
|
|
"--file",
|
|
"file",
|
|
"--cwd",
|
|
"cwd",
|
|
"--wdir",
|
|
"wdir",
|
|
"--no-exec",
|
|
"--yes",
|
|
"--overwrite-dvcfile",
|
|
"--ignore-build-cache",
|
|
"--remove-outs",
|
|
"--no-commit",
|
|
"--outs-persist",
|
|
"outs-persist",
|
|
"--outs-persist-no-cache",
|
|
"outs-persist-no-cache",
|
|
"--always-changed",
|
|
"command",
|
|
]
|
|
)
|
|
assert cli_args.func == CmdRun
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "run", autospec=True)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
deps=["deps"],
|
|
outs=["outs"],
|
|
outs_no_cache=["outs-no-cache"],
|
|
metrics=["metrics"],
|
|
metrics_no_cache=["metrics-no-cache"],
|
|
outs_persist=["outs-persist"],
|
|
outs_persist_no_cache=["outs-persist-no-cache"],
|
|
fname="file",
|
|
cwd="cwd",
|
|
wdir="wdir",
|
|
no_exec=True,
|
|
overwrite=True,
|
|
ignore_build_cache=True,
|
|
remove_outs=True,
|
|
no_commit=True,
|
|
always_changed=True,
|
|
cmd="command",
|
|
)
|
|
|
|
|
|
def test_run_args_from_cli(mocker, dvc_repo):
|
|
args = parse_args(["run", "echo", "foo"])
|
|
cmd = args.func(args)
|
|
m = mocker.patch.object(cmd.repo, "run", autospec=True)
|
|
assert cmd.run() == 0
|
|
m.assert_called_once_with(
|
|
deps=[],
|
|
outs=[],
|
|
outs_no_cache=[],
|
|
metrics=[],
|
|
metrics_no_cache=[],
|
|
outs_persist=[],
|
|
outs_persist_no_cache=[],
|
|
fname=None,
|
|
cwd=None,
|
|
wdir=None,
|
|
no_exec=False,
|
|
overwrite=False,
|
|
ignore_build_cache=False,
|
|
remove_outs=False,
|
|
no_commit=False,
|
|
always_changed=False,
|
|
cmd="echo foo",
|
|
)
|
|
|
|
|
|
def test_run_args_with_spaces(mocker, dvc_repo):
|
|
args = parse_args(["run", "echo", "foo bar"])
|
|
cmd = args.func(args)
|
|
m = mocker.patch.object(cmd.repo, "run", autospec=True)
|
|
assert cmd.run() == 0
|
|
m.assert_called_once_with(
|
|
deps=[],
|
|
outs=[],
|
|
outs_no_cache=[],
|
|
metrics=[],
|
|
metrics_no_cache=[],
|
|
outs_persist=[],
|
|
outs_persist_no_cache=[],
|
|
fname=None,
|
|
cwd=None,
|
|
wdir=None,
|
|
no_exec=False,
|
|
overwrite=False,
|
|
ignore_build_cache=False,
|
|
remove_outs=False,
|
|
no_commit=False,
|
|
always_changed=False,
|
|
cmd='echo "foo bar"',
|
|
)
|