diff --git a/dvc/testing/test_remote.py b/dvc/testing/test_remote.py index ea366a487..aefcc288f 100644 --- a/dvc/testing/test_remote.py +++ b/dvc/testing/test_remote.py @@ -91,3 +91,18 @@ class TestRemote: status_dir = dvc.cloud.status(dir_hashes) _check_status(status_dir, ok=dir_hashes) + + def test_stage_cache_push_pull(self, tmp_dir, dvc, remote): + tmp_dir.gen("foo", "foo") + stage = dvc.stage.add( + deps=["foo"], outs=["bar"], name="copy-foo-bar", cmd="cp foo bar" + ) + dvc.reproduce(stage.addressing) + assert dvc.push(run_cache=True) == 2 + + stage_cache_dir = tmp_dir / dvc.stage_cache.cache_dir + expected = list(stage_cache_dir.rglob("*")) + shutil.rmtree(stage_cache_dir) + + dvc.pull(run_cache=True) + assert list(stage_cache_dir.rglob("*")) == expected diff --git a/dvc/testing/test_run_cache.py b/dvc/testing/test_run_cache.py deleted file mode 100644 index a22ce8554..000000000 --- a/dvc/testing/test_run_cache.py +++ /dev/null @@ -1,19 +0,0 @@ -import shutil - -from .tmp_dir import TmpDir - - -def test_push_pull(tmp_dir, dvc, remote): - tmp_dir.gen("foo", "foo") - stage = dvc.stage.add( - deps=["foo"], outs=["bar"], name="copy-foo-bar", cmd="cp foo bar" - ) - dvc.reproduce(stage.addressing) - assert dvc.push(run_cache=True) == 2 - - stage_cache_dir = TmpDir(dvc.stage_cache.cache_dir) - expected = list(stage_cache_dir.rglob("*")) - shutil.rmtree(stage_cache_dir) - - dvc.pull(run_cache=True) - assert list(stage_cache_dir.rglob("*")) == expected