mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
103 lines
2.8 KiB
Python
103 lines
2.8 KiB
Python
import configobj
|
|
|
|
from dvc.main import main
|
|
from dvc.command.config import CmdConfig
|
|
from tests.basic_env import TestDvc
|
|
|
|
|
|
class TestConfigCLI(TestDvc):
|
|
def _contains(self, section, field, value, local=False):
|
|
if local:
|
|
fname = self.dvc.config.config_local_file
|
|
else:
|
|
fname = self.dvc.config.config_file
|
|
|
|
config = configobj.ConfigObj(fname)
|
|
if section not in config.keys():
|
|
return False
|
|
|
|
if field not in config[section].keys():
|
|
return False
|
|
|
|
if config[section][field] != value:
|
|
return False
|
|
|
|
return True
|
|
|
|
def test_root(self):
|
|
ret = main(['root'])
|
|
self.assertEqual(ret, 0)
|
|
|
|
# NOTE: check that `dvc root` is not blocked with dvc lock
|
|
with self.dvc.lock:
|
|
ret = main(['root'])
|
|
self.assertEqual(ret, 0)
|
|
|
|
def _do_test(self, local=False):
|
|
section = 'setsection'
|
|
field = 'setfield'
|
|
section_field = '{}.{}'.format(section, field)
|
|
value = 'setvalue'
|
|
newvalue = 'setnewvalue'
|
|
|
|
base = ['config']
|
|
if local:
|
|
base.append('--local')
|
|
|
|
ret = main(base + [section_field, value])
|
|
self.assertEqual(ret, 0)
|
|
self.assertTrue(self._contains(section, field, value, local))
|
|
|
|
ret = main(base + [section_field])
|
|
self.assertEqual(ret, 0)
|
|
|
|
ret = main(base + [section_field, newvalue])
|
|
self.assertEqual(ret, 0)
|
|
self.assertTrue(self._contains(section, field, newvalue, local))
|
|
self.assertFalse(self._contains(section, field, value, local))
|
|
|
|
ret = main(base + [section_field, '--unset'])
|
|
self.assertEqual(ret, 0)
|
|
self.assertFalse(self._contains(section, field, value, local))
|
|
|
|
def test(self):
|
|
self._do_test(False)
|
|
|
|
def test_local(self):
|
|
self._do_test(True)
|
|
|
|
def test_non_existing(self):
|
|
#FIXME check stdout/err
|
|
|
|
ret = main(['config', 'non_existing_section.field'])
|
|
self.assertEqual(ret, 1)
|
|
|
|
ret = main(['config', 'global.non_existing_field'])
|
|
self.assertEqual(ret, 1)
|
|
|
|
ret = main(['config', 'non_existing_section.field', '-u'])
|
|
self.assertEqual(ret, 1)
|
|
|
|
ret = main(['config', 'global.non_existing_field', '-u'])
|
|
self.assertEqual(ret, 1)
|
|
|
|
ret = main(['config', 'core.remote', 'myremote'])
|
|
self.assertEqual(ret, 0)
|
|
|
|
ret = main(['config', 'core.non_existing_field', '-u'])
|
|
self.assertEqual(ret, 1)
|
|
|
|
def test_failed_write(self):
|
|
class A(object):
|
|
local = False
|
|
name = 'core.remote'
|
|
value = 'myremote'
|
|
unset = False
|
|
|
|
args = A()
|
|
cmd = CmdConfig(args)
|
|
|
|
cmd.configobj.write = None
|
|
ret = cmd.save()
|
|
self.assertNotEqual(ret, 0)
|