mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-05 23:09:32 +01:00
75 lines
1.4 KiB
Python
75 lines
1.4 KiB
Python
from dvc.cli import parse_args
|
|
from dvc.command.data_sync import CmdDataFetch, CmdDataPull
|
|
|
|
|
|
def test_fetch(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"fetch",
|
|
"target.dvc",
|
|
"-r",
|
|
"remote",
|
|
"-a",
|
|
"-T",
|
|
"-d",
|
|
"-R",
|
|
"--verify",
|
|
"-j",
|
|
"2",
|
|
]
|
|
)
|
|
assert cli_args.func == CmdDataFetch
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch("dvc.repo.Repo.fetch")
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
targets=["target.dvc"],
|
|
remote="remote",
|
|
all_branches=True,
|
|
all_tags=True,
|
|
with_deps=True,
|
|
recursive=True,
|
|
verify=True,
|
|
jobs=2,
|
|
)
|
|
|
|
|
|
def test_pull(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"pull",
|
|
"target.dvc",
|
|
"-r",
|
|
"remote",
|
|
"-a",
|
|
"-T",
|
|
"-d",
|
|
"-R",
|
|
"-f",
|
|
"--verify",
|
|
"-j",
|
|
"2",
|
|
]
|
|
)
|
|
assert cli_args.func == CmdDataPull
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch("dvc.repo.Repo.pull")
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
targets=["target.dvc"],
|
|
remote="remote",
|
|
all_branches=True,
|
|
all_tags=True,
|
|
with_deps=True,
|
|
recursive=True,
|
|
force=True,
|
|
verify=True,
|
|
jobs=2,
|
|
)
|