From c4de07b2bd7b62e7e598539676ffec8375f62833 Mon Sep 17 00:00:00 2001 From: pared Date: Wed, 21 Aug 2019 14:31:50 +0200 Subject: [PATCH] repo: move: save stage after moving --- dvc/repo/move.py | 1 + tests/func/test_move.py | 8 ++++++++ 2 files changed, 9 insertions(+) diff --git a/dvc/repo/move.py b/dvc/repo/move.py index ea190fe61..6cd8b915c 100644 --- a/dvc/repo/move.py +++ b/dvc/repo/move.py @@ -65,5 +65,6 @@ def move(self, from_path, to_path): with self.state: out.move(to_out) + stage.save() stage.dump() diff --git a/tests/func/test_move.py b/tests/func/test_move.py index 1d2973f99..3004c7b29 100644 --- a/tests/func/test_move.py +++ b/tests/func/test_move.py @@ -192,3 +192,11 @@ class TestMoveFileInsideDirectory(TestDvc): dvc_fullpath = os.path.join(self.DATA_DIR, "data.txt.dvc") self.assertTrue(os.path.exists(data_fullpath)) self.assertTrue(os.path.exists(dvc_fullpath)) + + +def test_move_should_save_stage_info(dvc_repo, repo_dir): + dvc_repo.add(repo_dir.DATA_DIR) + + dvc_repo.move(repo_dir.DATA_DIR, "new_name") + + assert dvc_repo.status() == {}