Fix caplog test case and move call to CmdRemoteAdd

This commit is contained in:
Mayur Kularni
2019-12-10 11:49:19 +05:30
parent b271368845
commit a1ae3578c0
5 changed files with 19 additions and 25 deletions
+4
View File
@@ -19,6 +19,10 @@ class CmdRemoteConfig(CmdConfig):
class CmdRemoteAdd(CmdRemoteConfig):
def run(self):
if self.args.default:
logger.info(
"Setting '{}' as a default remote.".format(self.args.name)
)
self.remote_config.add(
self.args.name,
self.args.url,
+10 -8
View File
@@ -73,15 +73,17 @@ def _external_repo(url=None, rev=None, cache_dir=None):
# add default remote pointing to the original repo's cache location
if os.path.isdir(url):
rconfig = RemoteConfig(repo.config)
if not rconfig.remote_config_exists():
if not rconfig.remote_remote_set():
original_repo = Repo(url)
rconfig.add(
"upstream",
original_repo.cache.local.cache_dir,
default=True,
level=Config.LEVEL_LOCAL,
)
original_repo.close()
try:
rconfig.add(
"aut-generated-upstream",
original_repo.cache.local.cache_dir,
default=True,
level=Config.LEVEL_LOCAL,
)
finally:
original_repo.close()
if cache_dir is not None:
cache_config = CacheConfig(repo.config)
+4 -5
View File
@@ -105,7 +105,6 @@ class RemoteConfig(object):
level=level,
)
if default:
logger.info("Setting '{}' as a default remote.".format(name))
self.config.set(
Config.SECTION_CORE,
Config.SECTION_CORE_REMOTE,
@@ -157,7 +156,7 @@ class RemoteConfig(object):
Config.SECTION_CORE, Config.SECTION_CORE_REMOTE, level=level
)
def remote_config_exists(self):
def remote_remote_set(self):
"""
Checks if default remote config is present.
Args:
@@ -167,7 +166,7 @@ class RemoteConfig(object):
True if the remote config exists, else False
"""
try:
default = self.get_default()
self.get_default()
return True
except ConfigError:
default = None
return bool(default)
return False
-12
View File
@@ -1,9 +1,7 @@
import filecmp
import os
from mock import patch
from dvc.repo import Repo
from dvc.external_repo import external_repo
from dvc.scm.git import Git
from dvc.utils.fs import path_isin
@@ -28,13 +26,3 @@ def test_external_repo(erepo):
assert path_isin(repo.cache.local.cache_dir, repo.root_dir)
assert mock.call_count == 1
def test_external_repo_import_without_remote(erepo, dvc_repo):
src = erepo.FOO
dst = os.path.join(dvc_repo.root_dir, "foo")
Repo.get(erepo.root_dir, src, dst)
assert os.path.exists(dst)
assert os.path.isfile(dst)
assert filecmp.cmp(erepo.FOO, dst, shallow=False)
+1
View File
@@ -101,6 +101,7 @@ def test_get_from_non_dvc_master(erepo, tmp_path, monkeypatch, caplog):
with caplog.at_level(logging.INFO, logger="dvc"):
Repo.get(erepo._root_dir, erepo.FOO, out=imported_file, rev="branch")
assert caplog.text == ""
assert filecmp.cmp(
os.path.join(erepo._root_dir, erepo.FOO), imported_file, shallow=False
)