mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-05 15:28:27 +01:00
GaoandPeter Rowlands
18d147e2f8
* machine: add `machine list`&`machine modify` 1. add new command `dvc machine list` 2. complete `dvc machine modify` 3. add new unit tests for this two commands to ensure the call. 4. add functionality tests for two commands * Update dvc/config_schema.py Co-authored-by: Peter Rowlands (변기호) <[email protected]> * Update tests/func/machine/test_machine_config.py Co-authored-by: Peter Rowlands (변기호) <[email protected]> * Some reviewed problems 1. capitalize global variables. 2. remove precision restriction. 3. solve windows path seperator problem. Co-authored-by: Peter Rowlands (변기호) <[email protected]>
105 lines
2.8 KiB
Python
105 lines
2.8 KiB
Python
import configobj
|
|
|
|
from dvc.cli import parse_args
|
|
from dvc.command.machine import (
|
|
CmdMachineAdd,
|
|
CmdMachineCreate,
|
|
CmdMachineDestroy,
|
|
CmdMachineList,
|
|
CmdMachineModify,
|
|
CmdMachineRemove,
|
|
CmdMachineSsh,
|
|
)
|
|
|
|
DATA = {
|
|
".dvc": {
|
|
"config": (
|
|
"[feature]\n"
|
|
" machine = true\n"
|
|
"['machine \"foo\"']\n"
|
|
" cloud = aws"
|
|
)
|
|
}
|
|
}
|
|
|
|
|
|
def test_add(tmp_dir):
|
|
tmp_dir.gen({".dvc": {"config": "[feature]\n machine = true"}})
|
|
cli_args = parse_args(["machine", "add", "foo", "aws"])
|
|
assert cli_args.func == CmdMachineAdd
|
|
cmd = cli_args.func(cli_args)
|
|
assert cmd.run() == 0
|
|
config = configobj.ConfigObj(str(tmp_dir / ".dvc" / "config"))
|
|
assert config['machine "foo"']["cloud"] == "aws"
|
|
|
|
|
|
def test_remove(tmp_dir):
|
|
tmp_dir.gen(DATA)
|
|
cli_args = parse_args(["machine", "remove", "foo"])
|
|
assert cli_args.func == CmdMachineRemove
|
|
cmd = cli_args.func(cli_args)
|
|
assert cmd.run() == 0
|
|
config = configobj.ConfigObj(str(tmp_dir / ".dvc" / "config"))
|
|
assert list(config.keys()) == ["feature"]
|
|
|
|
|
|
def test_create(tmp_dir, dvc, mocker):
|
|
cli_args = parse_args(["machine", "create", "foo"])
|
|
assert cli_args.func == CmdMachineCreate
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(
|
|
cmd.repo.machine, "create", autospec=True, return_value=0
|
|
)
|
|
|
|
assert cmd.run() == 0
|
|
m.assert_called_once_with("foo")
|
|
|
|
|
|
def test_destroy(tmp_dir, dvc, mocker):
|
|
cli_args = parse_args(["machine", "destroy", "foo"])
|
|
assert cli_args.func == CmdMachineDestroy
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(
|
|
cmd.repo.machine, "destroy", autospec=True, return_value=0
|
|
)
|
|
|
|
assert cmd.run() == 0
|
|
m.assert_called_once_with("foo")
|
|
|
|
|
|
def test_ssh(tmp_dir, dvc, mocker):
|
|
cli_args = parse_args(["machine", "ssh", "foo"])
|
|
assert cli_args.func == CmdMachineSsh
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(
|
|
cmd.repo.machine, "run_shell", autospec=True, return_value=0
|
|
)
|
|
|
|
assert cmd.run() == 0
|
|
m.assert_called_once_with("foo")
|
|
|
|
|
|
def test_list(tmp_dir, mocker):
|
|
from dvc.ui import ui
|
|
|
|
tmp_dir.gen(DATA)
|
|
cli_args = parse_args(["machine", "list", "foo"])
|
|
assert cli_args.func == CmdMachineList
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(ui, "write", autospec=True)
|
|
assert cmd.run() == 0
|
|
m.assert_called_once_with("cloud=aws")
|
|
|
|
|
|
def test_modified(tmp_dir):
|
|
tmp_dir.gen(DATA)
|
|
cli_args = parse_args(["machine", "modify", "foo", "cloud", "azure"])
|
|
assert cli_args.func == CmdMachineModify
|
|
cmd = cli_args.func(cli_args)
|
|
assert cmd.run() == 0
|
|
config = configobj.ConfigObj(str(tmp_dir / ".dvc" / "config"))
|
|
assert config['machine "foo"']["cloud"] == "azure"
|