stage: fix --dry-run (#3799)

* tests: unit: move stage tests into a separate dir * stage: fix --dry-run Part of #3777
iterative · May 14, 2020 · ff4fcc9 · ff4fcc9
1 parent 1676a7a
commit ff4fcc9
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/dvc/stage/run.py b/dvc/stage/run.py
@@ -78,6 +78,7 @@ def cmd_run(stage, *args, **kwargs):
 
 def run_stage(stage, dry=False, force=False, run_cache=False):
     if dry:
+        logger.info("Running command:\n\t{}".format(stage.cmd))
         return
     stage_cache = stage.repo.stage_cache
     stage_cached = (

diff --git a/tests/unit/stage/test_run.py b/tests/unit/stage/test_run.py
@@ -0,0 +1,10 @@
+import logging
+
+from dvc.stage import Stage
+from dvc.stage.run import run_stage
+
+
+def test_run_stage_dry(caplog):
+    with caplog.at_level(level=logging.INFO, logger="dvc"):
+        run_stage(Stage(None, cmd="mycmd arg1 arg2"), dry=True)
+        assert caplog.messages == ["Running command:\n\tmycmd arg1 arg2"]
diff --git a/tests/unit/test_stage.py → tests/unit/stage/test_stage.py b/tests/unit/test_stage.py → tests/unit/stage/test_stage.py