From 0f5216cb14d145c3386139e6cd6a09666e452a9a Mon Sep 17 00:00:00 2001
From: pared <pawelredzynski@gmail.com>
Date: Sun, 31 Mar 2019 21:28:21 +0200
Subject: [PATCH] checkout: reset progress bar on checkout

---
 dvc/progress.py             |  1 +
 tests/unit/test_progress.py | 12 +++++++++++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/dvc/progress.py b/dvc/progress.py
index d101b7bf9..8444b808b 100644
--- a/dvc/progress.py
+++ b/dvc/progress.py
@@ -144,6 +144,7 @@ class ProgressCallback(object):
     def __init__(self, total):
         self.total = total
         self.current = 0
+        progress.reset()
 
     def update(self, name, progress_to_add=1):
         self.current += progress_to_add
diff --git a/tests/unit/test_progress.py b/tests/unit/test_progress.py
index 76e3ce9f8..34de53799 100644
--- a/tests/unit/test_progress.py
+++ b/tests/unit/test_progress.py
@@ -1,7 +1,7 @@
 import mock
 from unittest import TestCase
 
-from dvc.progress import progress, progress_aware
+from dvc.progress import progress, progress_aware, ProgressCallback
 
 
 class TestProgressAware(TestCase):
@@ -25,3 +25,13 @@ class TestProgressAware(TestCase):
         self.assertEqual(4, mock_print.call_count)
 
         progress.finish_target("testing")
+
+
+class TestProgressCallback(TestCase):
+    @mock.patch("dvc.progress.progress")
+    def test_should_init_reset_progress(self, progress_mock):
+        total_files_num = 1
+
+        ProgressCallback(total_files_num)
+
+        self.assertEqual([mock.call.reset()], progress_mock.method_calls)