mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 07:18:27 +01:00
Ruslan Kuprieiev
785dcfeef0
This is the most simple (aka dumb) implementation for this functionality that is needed to unblock CICD development.
119 lines
2.6 KiB
Python
119 lines
2.6 KiB
Python
from dvc.cli import parse_args
|
|
from dvc.command.data_sync import CmdDataFetch, CmdDataPull, CmdDataPush
|
|
|
|
|
|
def test_fetch(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"fetch",
|
|
"target1",
|
|
"target2",
|
|
"--jobs",
|
|
"2",
|
|
"--remote",
|
|
"remote",
|
|
"--all-branches",
|
|
"--all-tags",
|
|
"--all-commits",
|
|
"--with-deps",
|
|
"--recursive",
|
|
"--run-cache",
|
|
]
|
|
)
|
|
assert cli_args.func == CmdDataFetch
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "fetch", autospec=True, return_value=0)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
targets=["target1", "target2"],
|
|
jobs=2,
|
|
remote="remote",
|
|
all_branches=True,
|
|
all_tags=True,
|
|
all_commits=True,
|
|
with_deps=True,
|
|
recursive=True,
|
|
run_cache=True,
|
|
)
|
|
|
|
|
|
def test_pull(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"pull",
|
|
"target1",
|
|
"target2",
|
|
"--jobs",
|
|
"2",
|
|
"--remote",
|
|
"remote",
|
|
"--all-branches",
|
|
"--all-tags",
|
|
"--all-commits",
|
|
"--with-deps",
|
|
"--force",
|
|
"--recursive",
|
|
"--run-cache",
|
|
]
|
|
)
|
|
assert cli_args.func == CmdDataPull
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "pull", autospec=True)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
targets=["target1", "target2"],
|
|
jobs=2,
|
|
remote="remote",
|
|
all_branches=True,
|
|
all_tags=True,
|
|
all_commits=True,
|
|
with_deps=True,
|
|
force=True,
|
|
recursive=True,
|
|
run_cache=True,
|
|
)
|
|
|
|
|
|
def test_push(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"push",
|
|
"target1",
|
|
"target2",
|
|
"--jobs",
|
|
"2",
|
|
"--remote",
|
|
"remote",
|
|
"--all-branches",
|
|
"--all-tags",
|
|
"--all-commits",
|
|
"--with-deps",
|
|
"--recursive",
|
|
"--run-cache",
|
|
]
|
|
)
|
|
assert cli_args.func == CmdDataPush
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "push", autospec=True, return_value=0)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
targets=["target1", "target2"],
|
|
jobs=2,
|
|
remote="remote",
|
|
all_branches=True,
|
|
all_tags=True,
|
|
all_commits=True,
|
|
with_deps=True,
|
|
recursive=True,
|
|
run_cache=True,
|
|
)
|