mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-06 15:28:28 +01:00
28 lines
912 B
Python
28 lines
912 B
Python
import mock
|
|
from unittest import TestCase
|
|
|
|
from dvc.progress import progress, progress_aware
|
|
|
|
|
|
class TestProgressAware(TestCase):
|
|
@mock.patch("sys.stdout.isatty", return_value=True)
|
|
@mock.patch("dvc.progress.Progress.print")
|
|
def test(self, mock_print, _):
|
|
# progress is a global object, can be shared between tests when
|
|
# run in multi-threading environment with pytest
|
|
progress.reset()
|
|
function = progress_aware(lambda: None)
|
|
|
|
function()
|
|
# first - called once for progress.clearln()
|
|
mock_print.assert_called_once()
|
|
|
|
progress.update_target("testing", 0, 100)
|
|
function()
|
|
# second - progress.clearln() on refresh on update_target
|
|
# third - progress.print on refresh on update_target
|
|
# fourth - progress.clearln()
|
|
self.assertEqual(4, mock_print.call_count)
|
|
|
|
progress.finish_target("testing")
|