mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-08 15:28:26 +01:00
32 lines
778 B
Python
32 lines
778 B
Python
from dvc.cli import parse_args
|
|
from dvc.command.get import CmdGet
|
|
|
|
|
|
def test_get(mocker):
|
|
cli_args = parse_args(
|
|
["get", "repo_url", "src", "--out", "out", "--rev", "version"]
|
|
)
|
|
assert cli_args.func == CmdGet
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch("dvc.repo.Repo.get")
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with("repo_url", path="src", out="out", rev="version")
|
|
|
|
|
|
def test_get_url(mocker):
|
|
cli_args = parse_args(
|
|
["get", "repo_url", "src", "--rev", "version", "--show-url"]
|
|
)
|
|
assert cli_args.func == CmdGet
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch("dvc.api.get_url")
|
|
m.return_value = "url"
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with("src", repo="repo_url", rev="version")
|