mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-05 15:28:27 +01:00
Ruslan Kuprieiev
251a3269c2
Outs desc will be used by viewer right away. Stage desc is for people to start using now, but we won't be using it it viewer until DAG support is added there.
62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
import logging
|
|
|
|
from dvc.cli import parse_args
|
|
from dvc.command.imp_url import CmdImportUrl
|
|
from dvc.exceptions import DvcException
|
|
|
|
|
|
def test_import_url(mocker):
|
|
cli_args = parse_args(
|
|
["import-url", "src", "out", "--file", "file", "--desc", "description"]
|
|
)
|
|
assert cli_args.func == CmdImportUrl
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "imp_url", autospec=True)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
"src", out="out", fname="file", no_exec=False, desc="description"
|
|
)
|
|
|
|
|
|
def test_failed_import_url(mocker, caplog):
|
|
cli_args = parse_args(["import-url", "http://somesite.com/file_name"])
|
|
assert cli_args.func == CmdImportUrl
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
mocker.patch.object(cmd.repo, "imp_url", side_effect=DvcException("error"))
|
|
with caplog.at_level(logging.ERROR, logger="dvc"):
|
|
assert cmd.run() == 1
|
|
expected_error = (
|
|
"failed to import http://somesite.com/file_name. "
|
|
"You could also try downloading it manually, and "
|
|
"adding it with `dvc add`."
|
|
)
|
|
assert expected_error in caplog.text
|
|
|
|
|
|
def test_import_url_no_exec(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"import-url",
|
|
"--no-exec",
|
|
"src",
|
|
"out",
|
|
"--file",
|
|
"file",
|
|
"--desc",
|
|
"description",
|
|
]
|
|
)
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "imp_url", autospec=True)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
"src", out="out", fname="file", no_exec=True, desc="description"
|
|
)
|