stage: fix --dry-run

Part of iterative#3777
efiop · May 14, 2020 · 0fade54 · 0fade54
1 parent 8c1270a
commit 0fade54
Show file tree

Hide file tree

Showing 2 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"]