import url: test: migrate to dir helpers

This commit is contained in:
pawel
2020-01-08 10:30:33 +01:00
committed by Paweł Redzyński
parent 2966924e47
commit 62d6f4dbdb
+8 -7
View File
@@ -1,4 +1,3 @@
import filecmp
import os
from uuid import uuid4
@@ -6,6 +5,7 @@ import pytest
from mock import patch
import dvc
from dvc.compat import fspath
from dvc.main import main
from dvc.utils import makedirs
from tests.basic_env import TestDvc
@@ -90,15 +90,16 @@ class TestImportFilename(TestDvc):
@pytest.mark.parametrize("dname", [".", "dir", "dir/subdir"])
def test_import_url_to_dir(dname, repo_dir, dvc_repo):
src = repo_dir.DATA
def test_import_url_to_dir(dname, tmp_dir, dvc):
tmp_dir.gen({"data_dir": {"file": "file content"}})
src = os.path.join("data_dir", "file")
makedirs(dname, exist_ok=True)
stage = dvc_repo.imp_url(src, dname)
stage = dvc.imp_url(src, dname)
dst = os.path.join(dname, os.path.basename(src))
dst = tmp_dir / dname / "file"
assert stage.outs[0].fspath == os.path.abspath(dst)
assert stage.outs[0].fspath == fspath(dst)
assert os.path.isdir(dname)
assert filecmp.cmp(repo_dir.DATA, dst, shallow=False)
assert (tmp_dir / dname / "file").read_text() == "file content"