mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-06 15:28:28 +01:00
Kevin Lloyd Bernal
58bed0e9ba
To be consistent with the same --no-exec param in import-url. Fixes #4567.
65 lines
1.2 KiB
Python
65 lines
1.2 KiB
Python
from dvc.cli import parse_args
|
|
from dvc.command.imp import CmdImport
|
|
|
|
|
|
def test_import(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"import",
|
|
"repo_url",
|
|
"src",
|
|
"--out",
|
|
"out",
|
|
"--file",
|
|
"file",
|
|
"--rev",
|
|
"version",
|
|
]
|
|
)
|
|
assert cli_args.func == CmdImport
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "imp", autospec=True)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
"repo_url",
|
|
path="src",
|
|
out="out",
|
|
fname="file",
|
|
rev="version",
|
|
no_exec=False,
|
|
)
|
|
|
|
|
|
def test_import_no_exec(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"import",
|
|
"repo_url",
|
|
"src",
|
|
"--out",
|
|
"out",
|
|
"--file",
|
|
"file",
|
|
"--rev",
|
|
"version",
|
|
"--no-exec",
|
|
]
|
|
)
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "imp", autospec=True)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
"repo_url",
|
|
path="src",
|
|
out="out",
|
|
fname="file",
|
|
rev="version",
|
|
no_exec=True,
|
|
)
|